tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::tir::BufferRealizeNode Class Reference

Annotate the region where the buffer need to be read and write in the body. We only need to allocate the space for the corresponding region. More...

#include <stmt.h>

Inheritance diagram for tvm::tir::BufferRealizeNode:
Collaboration diagram for tvm::tir::BufferRealizeNode:

Public Member Functions

 BufferRealizeNode ()=default
 
 BufferRealizeNode (Buffer buffer, Array< Range > bounds, PrimExpr condition, Stmt body, Span span=Span())
 
 TVM_DECLARE_FINAL_OBJECT_INFO (BufferRealizeNode, StmtNode)
 
- Public Member Functions inherited from tvm::tir::StmtNode
 StmtNode ()=default
 
 StmtNode (Span span)
 
 TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()
 
 TVM_DECLARE_BASE_OBJECT_INFO (StmtNode, Object)
 

Static Public Member Functions

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

Public Attributes

Buffer buffer
 The buffer variable. More...
 
Array< Rangebounds
 Bounds to be realized. More...
 
PrimExpr condition
 Only realize if condition holds. More...
 
Stmt body
 The body of realization. More...
 
- Public Attributes inherited from tvm::tir::StmtNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Static Public Attributes

static constexpr const char * _type_key = "tir.BufferRealize"
 
- Static Public Attributes inherited from tvm::tir::StmtNode
static constexpr const char * _type_key = "tir.Stmt"
 
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 15
 

Detailed Description

Annotate the region where the buffer need to be read and write in the body. We only need to allocate the space for the corresponding region.

Note
There should be at most one BufferRealize for each buffer. BufferRealize is not necessary for external buffers, since they are assumed to be fully allocated.
See also
BufferLoad, BufferStore

Constructor & Destructor Documentation

◆ BufferRealizeNode() [1/2]

tvm::tir::BufferRealizeNode::BufferRealizeNode ( )
default

◆ BufferRealizeNode() [2/2]

tvm::tir::BufferRealizeNode::BufferRealizeNode ( Buffer  buffer,
Array< Range bounds,
PrimExpr  condition,
Stmt  body,
Span  span = Span() 
)
inline

Member Function Documentation

◆ RegisterReflection()

static void tvm::tir::BufferRealizeNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::tir::BufferRealizeNode::TVM_DECLARE_FINAL_OBJECT_INFO ( BufferRealizeNode  ,
StmtNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::tir::BufferRealizeNode::_type_key = "tir.BufferRealize"
staticconstexpr

◆ body

Stmt tvm::tir::BufferRealizeNode::body

The body of realization.

◆ bounds

Array<Range> tvm::tir::BufferRealizeNode::bounds

Bounds to be realized.

◆ buffer

Buffer tvm::tir::BufferRealizeNode::buffer

The buffer variable.

◆ condition

PrimExpr tvm::tir::BufferRealizeNode::condition

Only realize if condition holds.


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