tvm::tirx::StmtSRef Class Reference

Managed reference to StmtSRefNode. More...

#include <sblock_scope.h>

Inheritance diagram for tvm::tirx::StmtSRef:

Collaboration diagram for tvm::tirx::StmtSRef:

Public Member Functions
	StmtSRef (const StmtNode *stmt, StmtSRefNode *parent, int64_t seq_index)
	The constructor.
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (StmtSRef, ffi::ObjectRef, StmtSRefNode)

Static Public Member Functions
static StmtSRef	InlineMark ()
static StmtSRef	RootMark ()

Detailed Description

Managed reference to StmtSRefNode.

See also

StmtSRefNode

Constructor & Destructor Documentation

StmtSRef()

tvm::tirx::StmtSRef::StmtSRef ( const StmtNode *  stmt, StmtSRefNode *  parent, int64_t  seq_index  )

explicit

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.

Member Function Documentation

InlineMark()

static StmtSRef tvm::tirx::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::tirx::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::tirx::StmtSRef::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	StmtSRef	,
		ffi::ObjectRef	,
		StmtSRefNode
	)

---

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

• include/tvm/s_tir/sblock_scope.h