Allocate a buffer that can be used in body. More...

#include <stmt.h>

Inheritance diagram for tvm::tir::AllocateConstNode:

Collaboration diagram for tvm::tir::AllocateConstNode:

Public Member Functions
int64_t	ConstantAllocationSize () const
	If the buffer size is constant, return the size. Otherwise return 0. More...

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tir.AllocateConst", AllocateConstNode, StmtNode)

Public Member Functions inherited from tvm::tir::StmtNode
	StmtNode ()=default

	StmtNode (Span span)

	TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()

	TVM_FFI_DECLARE_OBJECT_INFO ("tir.Stmt", StmtNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

static int64_t	ConstantAllocationSize (const ffi::Array< PrimExpr > &extents)
	If the buffer size is constant, return the size. Otherwise return 0. More...

Static Public Member Functions inherited from tvm::tir::StmtNode
static void	RegisterReflection ()

Public Attributes
Var	buffer_var
	The buffer variable. More...

ffi::Optional< runtime::Tensor >	data
	The optional data associated to the constant. More...

ffi::Optional< Integer >	irmod_storage_idx
	If the PrimFunc containing the Stmt is added to IRModule, this is an optional index to indicate the index within "constants" attribute, that is a ffi::Array<Tensor> of IRModule. More...

DataType	dtype
	The type of the buffer. More...

ffi::Array< PrimExpr >	extents
	The extents of the buffer. More...

Stmt	body
	The body to be executed. More...

ffi::Map< ffi::String, ffi::Any >	annotations
	Additional annotations about the allocation. More...

Public Attributes inherited from tvm::tir::StmtNode
Span	span
	Span that points to the original source code. Reserved debug information. More...

Additional Inherited Members
Static Public Attributes inherited from tvm::tir::StmtNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

static constexpr const uint32_t	_type_child_slots = 15

Detailed Description

Allocate a buffer that can be used in body.

Member Function Documentation

◆ ConstantAllocationSize() [1/2]

int64_t tvm::tir::AllocateConstNode::ConstantAllocationSize ( ) const

inline

If the buffer size is constant, return the size. Otherwise return 0.

Returns: The result.

◆ ConstantAllocationSize() [2/2]

static int64_t tvm::tir::AllocateConstNode::ConstantAllocationSize ( const ffi::Array< PrimExpr > & extents )

static

If the buffer size is constant, return the size. Otherwise return 0.

Parameters

extents The extents of the buffer.

Returns: The result.

◆ RegisterReflection()

static void tvm::tir::AllocateConstNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tir::AllocateConstNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"tir.AllocateConst"	,
		AllocateConstNode	,
		StmtNode
	)

Member Data Documentation

◆ annotations

ffi::Map<ffi::String, ffi::Any> tvm::tir::AllocateConstNode::annotations

Additional annotations about the allocation.

These annotations can be used as auxiliary hint to future transformations.

◆ body

Stmt tvm::tir::AllocateConstNode::body

The body to be executed.

◆ buffer_var

Var tvm::tir::AllocateConstNode::buffer_var

The buffer variable.

◆ data

ffi::Optional<runtime::Tensor> tvm::tir::AllocateConstNode::data

The optional data associated to the constant.

◆ dtype

DataType tvm::tir::AllocateConstNode::dtype

The type of the buffer.

◆ extents

ffi::Array<PrimExpr> tvm::tir::AllocateConstNode::extents

The extents of the buffer.

◆ irmod_storage_idx

ffi::Optional<Integer> tvm::tir::AllocateConstNode::irmod_storage_idx

If the PrimFunc containing the Stmt is added to IRModule, this is an optional index to indicate the index within "constants" attribute, that is a ffi::Array<Tensor> of IRModule.

The documentation for this class was generated from the following file:

include/tvm/tir/stmt.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members