Match introduces a constraint that the source tensor region can be remapped to the data layout specified by the tensor field. The constraint can be checked in later part of lowering (or optionally during runtime). More...

#include <stmt.h>

Inheritance diagram for tvm::s_tir::MatchTensorRegionNode:

Collaboration diagram for tvm::s_tir::MatchTensorRegionNode:

Public Member Functions
	MatchTensorRegionNode (ffi::UnsafeInit tag)

	MatchTensorRegionNode (tirx::TensorVar tensor, TensorRegion source)

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.MatchTensorRegion", MatchTensorRegionNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
tirx::TensorVar	tensor
	The target tensor.

TensorRegion	source
	The source tensor region.

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

Detailed Description

Match introduces a constraint that the source tensor region can be remapped to the data layout specified by the tensor field. The constraint can be checked in later part of lowering (or optionally during runtime).

MatchTensorRegion provides a mechanism to represent data layout and compactness constraints in low-level hardware primitives in the IR and defer the check after the sequence of transformations.

Constructor & Destructor Documentation

◆ MatchTensorRegionNode() [1/2]

tvm::s_tir::MatchTensorRegionNode::MatchTensorRegionNode ( ffi::UnsafeInit tag )

inlineexplicit

◆ MatchTensorRegionNode() [2/2]

tvm::s_tir::MatchTensorRegionNode::MatchTensorRegionNode	(	tirx::TensorVar	tensor,
		TensorRegion	source
	)

inline

Member Function Documentation

◆ RegisterReflection()

static void tvm::s_tir::MatchTensorRegionNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::MatchTensorRegionNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.MatchTensorRegion"	,
		MatchTensorRegionNode	,
		ffi::Object
	)

Member Data Documentation

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::s_tir::MatchTensorRegionNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

staticconstexpr

◆ source

TensorRegion tvm::s_tir::MatchTensorRegionNode::source

The source tensor region.

◆ tensor

tirx::TensorVar tvm::s_tir::MatchTensorRegionNode::tensor

The target tensor.

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

include/tvm/s_tir/stmt.h

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes