buffer.h File Reference

Symbolic n-dimensional array, to represent a memory buffer. More...

#include <tvm/ffi/container/array.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ffi/string.h>
#include <tvm/ir/expr.h>
#include <tvm/tirx/layout.h>
#include <tvm/tirx/var.h>
#include <string>

Include dependency graph for buffer.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::tirx::BufferTypeNode
	Structural type of a TIRx buffer variable. More...
class	tvm::tirx::BufferType
	Managed reference to BufferTypeNode. More...
class	tvm::tirx::BufferVar
	Checked zero-state view over an ordinary VarNode with BufferType. More...
struct	tvm::ffi::TypeTraits< tirx::BufferVar >

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::tirx
namespace	tvm::ffi

Macros
#define	TVM_INDEX_DEFAULT_I64   1

Functions
PrimType	tvm::tirx::DefaultIndexPrimType ()
	if TVM_INDEX_DEFAULT_I64 is set, return int64, otherwise return int32
DLDataType	tvm::tirx::DefaultIndexType ()
bool	tvm::tirx::operator== (const BufferVar &lhs, const BufferVar &rhs)
bool	tvm::tirx::operator!= (const BufferVar &lhs, const BufferVar &rhs)
BufferVar	tvm::tirx::GetBufferVar (const VarNode *var)
	Recover a checked buffer view from an ordinary VarNode pointer.
ffi::ObjectPtr< BufferTypeNode >	tvm::tirx::CopyBufferType (const BufferVar &var)
BufferVar	tvm::tirx::RebuildBufferVar (const BufferVar &var, ffi::ObjectPtr< BufferTypeNode > type, ffi::Optional< ffi::String > name=std::nullopt)
BufferVar	tvm::tirx::decl_buffer (ffi::Array< PrimExpr > shape, PrimType dtype=PrimType::Float(32), ffi::String name="buffer", ffi::String storage_scope="", Span span=Span())
	Construct a new buffer given shape, and dtype.
tirx::BufferVar	tvm::tirx::BufferWithOffsetAlignment (ffi::Array< PrimExpr > shape, PrimType dtype, std::string name, int data_alignment, int offset_factor, std::string memory_scope="")
	Creates a TIR buffer for the provided parameters.
TensorLoad	tvm::tirx::BufferLoad (BufferVar buffer, ffi::Array< PrimExpr > indices, Span span=Span())
	Construct a TensorLoad from a BufferVar.

Detailed Description

Symbolic n-dimensional array, to represent a memory buffer.

Macro Definition Documentation

TVM_INDEX_DEFAULT_I64

#define TVM_INDEX_DEFAULT_I64   1