Managed reference to StmtSRefNode.
More...
#include <block_scope.h>
Managed reference to StmtSRefNode.
- See also
- StmtSRefNode
◆ StmtSRef()
The constructor.
- Parameters
-
stmt | The corresponding stmt node, can be either block or for loop. |
parent | The parent sref. |
seq_index | The location in an array if the parent of the stmt contains multiple children. -1 if the parent does not contain multiple children. |
◆ InlineMark()
static StmtSRef tvm::tir::StmtSRef::InlineMark |
( |
| ) |
|
|
static |
- Returns
- A special StmtSRef, which doesn't point to any stmt in the AST, only serving as a "mark" to hint compute-at to do the work of compute-inline
- Note
- This is only as a faked loop sref for compute-at and reverse-compute-at, i.e.
compute-at(block, loop_sref): compute-inline(block) if loop_sref.same_as(InlineMark()) no-op if loop_sref.same_as(RootMark()) compute-at-impl(block, loop_sref) otherwise
◆ RootMark()
static StmtSRef tvm::tir::StmtSRef::RootMark |
( |
| ) |
|
|
static |
- Returns
- A special StmtSRef, which doesn't point to any stmt in the AST, only serving as a "mark" to hint compute-at to do nothing
- Note
- This is only as a faked loop sref for compute-at and reverse-compute-at, i.e.
compute-at(block, loop_sref): compute-inline(block) if loop_sref.same_as(InlineMark()) no-op if loop_sref.same_as(RootMark()) compute-at-impl(block, loop_sref) otherwise
◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()
tvm::tir::StmtSRef::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE |
( |
StmtSRef |
, |
|
|
ObjectRef |
, |
|
|
StmtSRefNode |
|
|
) |
| |
The documentation for this class was generated from the following file: