tvm
|
Allocate a buffer that can be used in body. More...
#include <stmt.h>
Public Member Functions | |
void | VisitAttrs (AttrVisitor *v) |
bool | SEqualReduce (const AllocateConstNode *other, SEqualReducer equal) const |
void | SHashReduce (SHashReducer hash_reduce) const |
int64_t | ConstantAllocationSize () const |
If the buffer size is constant, return the size. Otherwise return 0. More... | |
TVM_DECLARE_FINAL_OBJECT_INFO (AllocateConstNode, StmtNode) | |
![]() | |
StmtNode ()=default | |
StmtNode (Span span) | |
TVM_OBJECT_ENABLE_SCRIPT_PRINTER () | |
TVM_DECLARE_BASE_OBJECT_INFO (StmtNode, Object) | |
![]() | |
uint32_t | type_index () const |
std::string | GetTypeKey () const |
size_t | GetTypeKeyHash () const |
template<typename TargetType > | |
bool | IsInstance () const |
bool | unique () const |
Object () | |
Object (const Object &other) | |
Object (Object &&other) | |
Object & | operator= (const Object &other) |
Object & | operator= (Object &&other) |
Static Public Member Functions | |
static int64_t | ConstantAllocationSize (const Array< PrimExpr > &extents) |
If the buffer size is constant, return the size. Otherwise return 0. More... | |
![]() | |
static std::string | TypeIndex2Key (uint32_t tindex) |
Get the type key of the corresponding index from runtime. More... | |
static size_t | TypeIndex2KeyHash (uint32_t tindex) |
Get the type key hash of the corresponding index from runtime. More... | |
static uint32_t | TypeKey2Index (const std::string &key) |
Get the type index of the corresponding key from runtime. More... | |
static uint32_t | _GetOrAllocRuntimeTypeIndex () |
static uint32_t | RuntimeTypeIndex () |
Public Attributes | |
Var | buffer_var |
The buffer variable. More... | |
Optional< runtime::NDArray > | data |
The optional data associated to the constant. More... | |
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 Array<NDArray> of IRModule. More... | |
DataType | dtype |
The type of the buffer. More... | |
Array< PrimExpr > | extents |
The extents of the buffer. More... | |
Stmt | body |
The body to be executed. More... | |
Map< String, ObjectRef > | annotations |
Additional annotations about the allocation. More... | |
![]() | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "tir.AllocateConst" |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
![]() | |
static constexpr const char * | _type_key = "tir.Stmt" |
static constexpr const bool | _type_has_method_sequal_reduce = true |
static constexpr const bool | _type_has_method_shash_reduce = true |
static constexpr const uint32_t | _type_child_slots = 15 |
![]() | |
static constexpr const char * | _type_key = "runtime.Object" |
static constexpr bool | _type_final = false |
static constexpr uint32_t | _type_child_slots = 0 |
static constexpr bool | _type_child_slots_can_overflow = true |
static constexpr bool | _type_has_method_visit_attrs = true |
static constexpr bool | _type_has_method_sequal_reduce = false |
static constexpr bool | _type_has_method_shash_reduce = false |
static constexpr uint32_t | _type_index = TypeIndex::kDynamic |
Additional Inherited Members | |
![]() | |
typedef void(* | FDeleter) (Object *self) |
Object deleter. More... | |
using | RefCounterType = std::atomic< int32_t > |
![]() | |
void | IncRef () |
developer function, increases reference counter. More... | |
void | DecRef () |
developer function, decrease reference counter. More... | |
![]() | |
static uint32_t | GetOrAllocRuntimeTypeIndex (const std::string &key, uint32_t static_tindex, uint32_t parent_tindex, uint32_t type_child_slots, bool type_child_slots_can_overflow) |
Get the type index using type key. More... | |
![]() | |
uint32_t | type_index_ {0} |
Type index(tag) that indicates the type of the object. More... | |
RefCounterType | ref_counter_ {0} |
The internal reference counter. More... | |
FDeleter | deleter_ = nullptr |
deleter of this object to enable customized allocation. If the deleter is nullptr, no deletion will be performed. The creator of the object must always set the deleter field properly. More... | |
Allocate a buffer that can be used in body.
|
inline |
If the buffer size is constant, return the size. Otherwise return 0.
|
static |
If the buffer size is constant, return the size. Otherwise return 0.
extents | The extents of the buffer. |
|
inline |
|
inline |
tvm::tir::AllocateConstNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | AllocateConstNode | , |
StmtNode | |||
) |
|
inline |
|
staticconstexpr |
|
staticconstexpr |
|
staticconstexpr |
Additional annotations about the allocation.
These annotations can be used as auxiliary hint to future transformations.
Stmt tvm::tir::AllocateConstNode::body |
The body to be executed.
Var tvm::tir::AllocateConstNode::buffer_var |
The buffer variable.
Optional<runtime::NDArray> tvm::tir::AllocateConstNode::data |
The optional data associated to the constant.
DataType tvm::tir::AllocateConstNode::dtype |
The type of the buffer.