tvm::tirx::SBlockNode Class Reference

A block is a basic schedule unit in TIR. More...

#include <stmt.h>

Inheritance diagram for tvm::tirx::SBlockNode:

Collaboration diagram for tvm::tirx::SBlockNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tirx.SBlock", SBlockNode, StmtNode)
Public Member Functions inherited from tvm::tirx::StmtNode
	StmtNode ()=default
	StmtNode (Span span)
	TVM_FFI_DECLARE_OBJECT_INFO ("tirx.Stmt", StmtNode, ffi::Object)

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

Public Attributes
ffi::Array< IterVar >	iter_vars
	The variables of the block.
ffi::Array< BufferRegion >	reads
	The read buffer regions of the block.
ffi::Array< BufferRegion >	writes
	The write buffer regions of the block.
ffi::String	name_hint
	The name_hint of the block.
ffi::Array< BufferVar >	alloc_buffers
	The buffer allocated in the block.
ffi::Array< MatchBufferRegion >	match_buffers
	The match buffer regions.
ffi::Map< ffi::String, ffi::Any >	annotations
	The annotation of the block.
ffi::Optional< Stmt >	init
	The init statement is executed during the first iteration of reduction loops in a reduction block. The optional init field allows us to represent initialization and reduction update in a single block and transform them collectively. We also provide primitives to decompose the init into a separate block during scheduling. Init field is std::nullopt if there is no reduction iter_vars.
Stmt	body
	The body of the block.
Public Attributes inherited from tvm::tirx::StmtNode
Span	span
	Span that points to the original source code. Reserved debug information.

Additional Inherited Members
Static Public Attributes inherited from tvm::tirx::StmtNode
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
static constexpr const uint32_t	_type_child_slots = 15

Detailed Description

A block is a basic schedule unit in TIR.

Note

SBlock's body is parameterized by iter vars.

with T.sblock(name):
    v0 = T.axis.S(domain, value0)
    v1 = T.axis.R(domain, value1)
    ...
    T.reads([buffer0[start:end, ...], ...])
    T.writes([buffer1[start:end, ...], ...])
    T.where(predicate)
    buffer2 = T.alloc_buffer(shape, dtype)
    buffer3 = T.match_buffer(source_buffer[start:end, ...])
    T.attr({attr_key: attr_value, ...})
    with T.init():
        // init body
    // body

Member Function Documentation

RegisterReflection()

static void tvm::tirx::SBlockNode::RegisterReflection ( )

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tirx::SBlockNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"tirx.SBlock"	,
		SBlockNode	,
		StmtNode
	)

Member Data Documentation

alloc_buffers

ffi::Array<BufferVar> tvm::tirx::SBlockNode::alloc_buffers

The buffer allocated in the block.

annotations

ffi::Map<ffi::String, ffi::Any> tvm::tirx::SBlockNode::annotations

The annotation of the block.

body

Stmt tvm::tirx::SBlockNode::body

The body of the block.

init

ffi::Optional<Stmt> tvm::tirx::SBlockNode::init

The init statement is executed during the first iteration of reduction loops in a reduction block. The optional init field allows us to represent initialization and reduction update in a single block and transform them collectively. We also provide primitives to decompose the init into a separate block during scheduling. Init field is std::nullopt if there is no reduction iter_vars.

iter_vars

ffi::Array<IterVar> tvm::tirx::SBlockNode::iter_vars

The variables of the block.

match_buffers

ffi::Array<MatchBufferRegion> tvm::tirx::SBlockNode::match_buffers

The match buffer regions.

name_hint

ffi::String tvm::tirx::SBlockNode::name_hint

The name_hint of the block.

reads

ffi::Array<BufferRegion> tvm::tirx::SBlockNode::reads

The read buffer regions of the block.

writes

ffi::Array<BufferRegion> tvm::tirx::SBlockNode::writes

The write buffer regions of the block.

---

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

• include/tvm/tirx/stmt.h