ir.h File Reference

#include <tvm/s_tir/script/ir_builder/frame.h>
#include <tvm/tirx/script/ir_builder/ir.h>

Include dependency graph for ir.h:

Go to the source code of this file.

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::script
namespace	tvm::script::ir_builder
namespace	tvm::script::ir_builder::s_tir
namespace	tvm::script::ir_builder::s_tir::axis

Functions
FunctionFrame	tvm::script::ir_builder::s_tir::Function (bool is_private=false, bool persistent=false)
FunctionFrame	tvm::script::ir_builder::s_tir::DeclFunction (bool is_private=false, bool persistent=false)
TensorVar	tvm::script::ir_builder::s_tir::MatchTensor (ffi::ObjectRef param, ffi::Array< PrimExpr > shape, PrimType dtype=PrimType::Float(32), ffi::Optional< Expr > data=std::nullopt, ffi::Array< PrimExpr > strides={}, ffi::Optional< PrimExpr > elem_offset=std::nullopt, ffi::String storage_scope="global", int align=-1, int offset_factor=0, ffi::Optional< Layout > layout=std::nullopt, ffi::Array< PrimExpr > allocated_addr={})
	Bind a tensor subregion in an S-TIR block.
SBlockFrame	tvm::script::ir_builder::s_tir::Block (ffi::String name, bool no_realize=false, ffi::String exec_scope="")
	The block declaration statement.
BlockInitFrame	tvm::script::ir_builder::s_tir::Init ()
	The block initialization statement.
void	tvm::script::ir_builder::s_tir::Where (PrimExpr predicate)
	The block predicate statement.
void	tvm::script::ir_builder::s_tir::Reads (ffi::Array< ffi::ObjectRef > tensor_slices)
	The block tensor region reading statement.
void	tvm::script::ir_builder::s_tir::Writes (ffi::Array< ffi::ObjectRef > tensor_slices)
	The block tensor region writing statement.
void	tvm::script::ir_builder::s_tir::BlockAttrs (ffi::Map< ffi::String, ffi::Any > attrs)
	The block annotation statement.
TensorVar	tvm::script::ir_builder::s_tir::SBlockAllocTensor (ffi::Array< PrimExpr > shape, PrimType dtype=PrimType::Float(32), ffi::Optional< Expr > data=std::nullopt, ffi::Array< PrimExpr > strides={}, ffi::Optional< PrimExpr > elem_offset=std::nullopt, ffi::String storage_scope="", int align=-1, int offset_factor=0, ffi::Optional< Layout > layout=std::nullopt, ffi::Array< PrimExpr > allocated_addr={})
	The tensor allocation function.
Var	tvm::script::ir_builder::s_tir::axis::Spatial (Range dom, ffi::Optional< PrimExpr > binding=std::nullopt, PrimType dtype=PrimType::Int(32))
	The spatial block axis defining function.
Var	tvm::script::ir_builder::s_tir::axis::Reduce (Range dom, ffi::Optional< PrimExpr > binding=std::nullopt, PrimType dtype=PrimType::Int(32))
	The reduced block axis defining function.
Var	tvm::script::ir_builder::s_tir::axis::Scan (Range dom, ffi::Optional< PrimExpr > binding=std::nullopt, PrimType dtype=PrimType::Int(32))
	The scanning block axis defining function.
Var	tvm::script::ir_builder::s_tir::axis::Opaque (Range dom, ffi::Optional< PrimExpr > binding=std::nullopt, PrimType dtype=PrimType::Int(32))
	The opaque block axis defining function.
ffi::Array< Var >	tvm::script::ir_builder::s_tir::axis::Remap (ffi::String kinds, ffi::Array< PrimExpr > bindings, PrimType dtype=PrimType::Int(32))
	The block axis remapping function.