Allocate a buffer that can be used in body.
More...
#include <stmt.h>
Allocate a buffer that can be used in body.
◆ ConstantAllocationSize() [1/2]
int64_t tvm::tir::AllocateNode::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::AllocateNode::ConstantAllocationSize |
( |
const 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::AllocateNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
◆ _type_key
constexpr const char* tvm::tir::AllocateNode::_type_key = "tir.Allocate" |
|
staticconstexpr |
◆ annotations
Map<String, ffi::Any> tvm::tir::AllocateNode::annotations |
Additional annotations about the allocation.
These annotations can be used as auxiliary hint to future transformations.
◆ body
Stmt tvm::tir::AllocateNode::body |
◆ buffer_var
Var tvm::tir::AllocateNode::buffer_var |
◆ condition
PrimExpr tvm::tir::AllocateNode::condition |
Only allocate buffer when condition is satisfied.
◆ dtype
◆ extents
Array<PrimExpr> tvm::tir::AllocateNode::extents |
The extents of the buffer.
The documentation for this class was generated from the following file: