Namespaces
	axis

Classes
class	TIRFrameNode
	A base frame that represents the TIR fame with body of statements. More...

class	TIRFrame
	Managed reference to TIRFrameNode. More...

class	PrimFuncFrameNode
	A frame that represents the PrimFunc containing TIR statements. More...

class	PrimFuncFrame
	Managed reference to PrimFuncFrameNode. More...

class	BlockFrameNode
	A frame that represents the block. More...

class	BlockFrame
	Managed reference to BlockFrameNode. More...

class	BlockInitFrameNode
	A frame that represents the block initialization statment. More...

class	BlockInitFrame
	Managed reference to BlockInitFrameNode. More...

class	ForFrameNode
	A frame that represents the for loop. More...

class	ForFrame
	Managed reference to ForFrameNode. More...

class	AssertFrameNode
	A frame that represents the assert statement. Proceeds if the condition is true, otherwise aborts with the message. More...

class	AssertFrame
	Managed reference to AssertFrameNode. More...

class	LetFrameNode
	A frame represents the let binding expression, which binds a var. More...

class	LetFrame
	Managed reference to LetFrameNode. More...

class	LaunchThreadFrameNode
	The LaunchThreadFrameNode. More...

class	LaunchThreadFrame
	Managed reference to LaunchThreadFrameNode. More...

class	RealizeFrameNode
	A frame that represents realization. More...

class	RealizeFrame
	Managed reference to RealizeFrameNode. More...

class	AllocateFrameNode
	A frame represents the allocate. More...

class	AllocateFrame
	Managed reference to AllocateFrameNode. More...

class	AllocateConstFrameNode
	A frame represents the allocate constant. More...

class	AllocateConstFrame
	Managed reference to AllocateConstFrameNode. More...

class	AttrFrameNode
	A frame that represents attribute node. More...

class	AttrFrame
	Managed reference to AttrFrameNode. More...

class	WhileFrameNode
	A frame that represents while loop. More...

class	WhileFrame
	Managed reference to WhileFrameNode. More...

class	IfFrameNode
	A frame that represents if statement. More...

class	IfFrame
	Managed reference to IfFrameNode. More...

class	ThenFrameNode
	A frame that represents then. More...

class	ThenFrame
	Managed reference to ThenFrameNode. More...

class	ElseFrameNode
	A frame that represents else. More...

class	ElseFrame
	Managed reference to ElseFrameNode. More...

class	DeclBufferFrameNode

class	DeclBufferFrame

Functions
Buffer	BufferDecl (Array< PrimExpr > shape, DataType dtype, String buffer_name, Optional< Var > data, Optional< Array< PrimExpr >> strides, Optional< PrimExpr > elem_offset, String storage_scope, int align, int offset_factor, String buffer_type, Optional< Array< IntImm >> axis_separators)
	The buffer declaration function. More...

PrimFuncFrame	PrimFunc (bool is_private)
	The primitive function statement. More...

Var	Arg (String name, Var var)
	The PrimFunc variable arguments adding function. More...

Buffer	Arg (String name, Buffer buffer)
	The PrimFunc buffer arguments adding function. More...

void	FuncName (String name)
	The PrimFunc naming statement. More...

void	FuncAttrs (Map< String, ffi::Any > attrs)
	The PrimFunc annotation statement. More...

Type	FuncRet (Type ret_type)
	The PrimFunc return type statement. More...

Buffer	MatchBuffer (ObjectRef param, Array< PrimExpr > shape, DataType dtype=DataType::Float(32), Optional< Var > data=std::nullopt, Array< PrimExpr > strides={}, PrimExpr elem_offset=PrimExpr(), String storage_scope="global", int align=-1, int offset_factor=0, String buffer_type="default", Optional< Array< IntImm >> axis_separators=std::nullopt)
	The buffer match statement. More...

BlockFrame	Block (String name, bool no_realize=false)
	The block declaration statement. More...

BlockInitFrame	Init ()
	The block initialization statement. More...

void	Where (PrimExpr predicate)
	The block predicate statement. More...

void	Reads (Array< ObjectRef > buffer_slices)
	The block buffer region reading statement. More...

void	Writes (Array< ObjectRef > buffer_slices)
	The block buffer region writing statement. More...

void	BlockAttrs (Map< String, ffi::Any > attrs)
	The block annotation statement. More...

Buffer	AllocBuffer (Array< PrimExpr > shape, DataType dtype=DataType::Float(32), Optional< Var > data=std::nullopt, Array< PrimExpr > strides={}, PrimExpr elem_offset=PrimExpr(), String storage_scope="", int align=-1, int offset_factor=0, String buffer_type="default", Optional< Array< IntImm >> axis_separators=std::nullopt)
	The buffer allocation function. More...

ForFrame	Serial (PrimExpr start, PrimExpr stop, Optional< Map< String, Any >> annotations=std::nullopt)
	The serial For statement. More...

ForFrame	Parallel (PrimExpr start, PrimExpr stop, Optional< Map< String, Any >> annotations=std::nullopt)
	The parallel For statement. More...

ForFrame	Vectorized (PrimExpr start, PrimExpr stop, Optional< Map< String, Any >> annotations=std::nullopt)
	The vectorized For statement. More...

ForFrame	Unroll (PrimExpr start, PrimExpr stop, Optional< Map< String, Any >> annotations=std::nullopt)
	The unrolled For statement. More...

ForFrame	ThreadBinding (PrimExpr start, PrimExpr stop, String thread, Optional< Map< String, Any >> annotations=std::nullopt)
	The thread-binding For statement. More...

ForFrame	Grid (Array< PrimExpr > extents)
	The grid For statement. More...

AssertFrame	Assert (PrimExpr condition, String message)
	The assertion statement. More...

LetFrame	LetStmt (PrimExpr value, Optional< Type > type_annotation=std::nullopt, Optional< Var > var=std::nullopt)
	The let binding. More...

RealizeFrame	Realize (tvm::tir::BufferRegion buffer_slice, String storage_scope, PrimExpr condition)
	The realization. More...

AllocateFrame	Allocate (Array< PrimExpr > extents, DataType dtype, String storage_scope="", Optional< PrimExpr > condition=std::nullopt, Optional< Map< String, Any >> annotations=std::nullopt)
	The allocate node. More...

AllocateConstFrame	AllocateConst (NDArray data, DataType dtype, Array< PrimExpr > extents, Optional< Map< String, Any >> annotations=std::nullopt)
	The allocate constant node. More...

AttrFrame	Attr (ffi::Any node, String attr_key, PrimExpr value)
	Create an attribute. More...

WhileFrame	While (PrimExpr condition)
	Create a while loop. More...

IfFrame	If (PrimExpr condition)
	Create an if statement. More...

ThenFrame	Then ()
	Create a then. More...

ElseFrame	Else ()
	Create an else. More...

DeclBufferFrame	DeclBuffer (Array< PrimExpr > shape, DataType dtype, String buffer_name, Optional< Var > data, Optional< Array< PrimExpr >> strides, Optional< PrimExpr > elem_offset, String storage_scope, int align, int offset_factor, String buffer_type, Optional< Array< IntImm >> axis_separators)
	The buffer declaration frame. More...

LaunchThreadFrame	LaunchThread (Var var, PrimExpr extent)
	Launch a thread. More...

LaunchThreadFrame	LaunchThread (String thread_tag, PrimExpr extent)
	Launch a new thread. More...

Var	EnvThread (String thread_tag, DataType dtype=DataType::Int(32))
	Bind a var to thread env. More...

void	BufferStore (Buffer buffer, PrimExpr value, Array< PrimExpr > indices, Optional< PrimExpr > predicate)
	Store data in a buffer. More...

void	Evaluate (PrimExpr value)
	Evaluate the input expression. More...

Var	Handle (runtime::DataType dtype=runtime::DataType::Void(), String storage_scope="global", bool is_size_var=false, bool is_unknown_type=false)
	Create a TIR var that represents a pointer. More...

Var	TensormapHandle ()

PrimExpr	BFloat8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat8x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat8x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat8x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat8x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat8x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat16x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat32x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	BFloat64x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float16x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float32x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float64x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt8x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt16x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt32x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	UInt64x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int8x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int16x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int32x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Int64x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E3M4x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZ (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3B11FNUZx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FN (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZ (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E4M3FNUZx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2x4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2x8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2x16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2x32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2x64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZ (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E5M2FNUZx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNU (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNUx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNUx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNUx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNUx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float8E8M0FNUx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FN (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FNx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FNx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FNx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FNx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E2M3FNx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FN (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FNx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FNx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FNx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FNx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float6E3M2FNx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FN (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FNx4 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FNx8 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FNx16 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FNx32 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Float4E2M1FNx64 (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Boolean (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

PrimExpr	Void (Optional< PrimExpr > expr=std::nullopt, bool is_size_var=false)

Function Documentation

◆ Allocate()

AllocateFrame tvm::script::ir_builder::tir::Allocate	(	Array< PrimExpr >	extents,
		DataType	dtype,
		String	storage_scope = `""`,
		Optional< PrimExpr >	condition = `std::nullopt`,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The allocate node.

Parameters

extents	The extents of the allocate.
dtype	The data type of the buffer.
storage_scope	The storage scope.
condition	The condition.
annotations	Additional annotation hints.

Returns: The created AllocateFrame.

◆ AllocateConst()

AllocateConstFrame tvm::script::ir_builder::tir::AllocateConst	(	NDArray	data,
		DataType	dtype,
		Array< PrimExpr >	extents,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The allocate constant node.

Parameters

data	The data associated with the constant.
dtype	The data type of the buffer.
extents	The extents of the allocate.
annotations	Additional annotation hints.

Returns: The created AllocateConstFrame.

◆ AllocBuffer()

Buffer tvm::script::ir_builder::tir::AllocBuffer	(	Array< PrimExpr >	shape,
		DataType	dtype = `DataType::Float(32)`,
		Optional< Var >	data = `std::nullopt`,
		Array< PrimExpr >	strides = `{}`,
		PrimExpr	elem_offset = `PrimExpr()`,
		String	storage_scope = `""`,
		int	align = `-1`,
		int	offset_factor = `0`,
		String	buffer_type = `"default"`,
		Optional< Array< IntImm >>	axis_separators = `std::nullopt`
	)

The buffer allocation function.

Parameters

shape	The type of the buffer prior to flattening.
dtype	The data type in the content of the buffer.
data	The pointer to the head of the data.
strides	The strides of each dimension.
elem_offset	The offset in terms of number of dtype elements (including lanes).
storage_scope	The optional storage scope of buffer data pointer.
align	The alignment requirement of data pointer in bytes.
offset_factor	The factor of elem_offset field.
buffer_type	The buffer type.
axis_separators	The separators between input axes when generating flattened output axes.

Returns: The allocated buffer.

◆ Arg() [1/2]

Buffer tvm::script::ir_builder::tir::Arg	(	String	name,
		Buffer	buffer
	)

The PrimFunc buffer arguments adding function.

Parameters

name	The name of the buffer.
buffer	The buffer argument.

Returns: The buffer.

◆ Arg() [2/2]

Var tvm::script::ir_builder::tir::Arg	(	String	name,
		Var	var
	)

The PrimFunc variable arguments adding function.

Parameters

name	The name of the variable.
var	The variable argument.

Returns: The variable.

◆ Assert()

AssertFrame tvm::script::ir_builder::tir::Assert	(	PrimExpr	condition,
		String	message
	)

The assertion statement.

Parameters

condition	The assertion condition.
message	The error message when the assertion fails.

Returns: The AssertFrame.

◆ Attr()

AttrFrame tvm::script::ir_builder::tir::Attr	(	ffi::Any	node,
		String	attr_key,
		PrimExpr	value
	)

Create an attribute.

Parameters

node	The node to annotate the attribute.
attr_key	Attribute type key.
value	The value of the attribute.

Returns: The result AttrFrame.

◆ BFloat16()

PrimExpr tvm::script::ir_builder::tir::BFloat16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat16x16()

PrimExpr tvm::script::ir_builder::tir::BFloat16x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat16x32()

PrimExpr tvm::script::ir_builder::tir::BFloat16x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat16x4()

PrimExpr tvm::script::ir_builder::tir::BFloat16x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat16x64()

PrimExpr tvm::script::ir_builder::tir::BFloat16x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat16x8()

PrimExpr tvm::script::ir_builder::tir::BFloat16x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32()

PrimExpr tvm::script::ir_builder::tir::BFloat32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32x16()

PrimExpr tvm::script::ir_builder::tir::BFloat32x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32x32()

PrimExpr tvm::script::ir_builder::tir::BFloat32x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32x4()

PrimExpr tvm::script::ir_builder::tir::BFloat32x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32x64()

PrimExpr tvm::script::ir_builder::tir::BFloat32x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat32x8()

PrimExpr tvm::script::ir_builder::tir::BFloat32x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64()

PrimExpr tvm::script::ir_builder::tir::BFloat64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64x16()

PrimExpr tvm::script::ir_builder::tir::BFloat64x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64x32()

PrimExpr tvm::script::ir_builder::tir::BFloat64x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64x4()

PrimExpr tvm::script::ir_builder::tir::BFloat64x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64x64()

PrimExpr tvm::script::ir_builder::tir::BFloat64x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat64x8()

PrimExpr tvm::script::ir_builder::tir::BFloat64x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8()

PrimExpr tvm::script::ir_builder::tir::BFloat8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8x16()

PrimExpr tvm::script::ir_builder::tir::BFloat8x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8x32()

PrimExpr tvm::script::ir_builder::tir::BFloat8x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8x4()

PrimExpr tvm::script::ir_builder::tir::BFloat8x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8x64()

PrimExpr tvm::script::ir_builder::tir::BFloat8x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BFloat8x8()

PrimExpr tvm::script::ir_builder::tir::BFloat8x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Block()

BlockFrame tvm::script::ir_builder::tir::Block	(	String	name,
		bool	no_realize = `false`
	)

The block declaration statement.

Parameters

name	The name of the block.
no_realize	The flag whether to construct BlockRealize or Block.

Returns: The BlockFrame.

◆ BlockAttrs()

void tvm::script::ir_builder::tir::BlockAttrs ( Map< String, ffi::Any > attrs )

The block annotation statement.

Parameters

attrs The annotation of the block.

◆ Boolean()

PrimExpr tvm::script::ir_builder::tir::Boolean	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ BufferDecl()

Buffer tvm::script::ir_builder::tir::BufferDecl	(	Array< PrimExpr >	shape,
		DataType	dtype,
		String	buffer_name,
		Optional< Var >	data,
		Optional< Array< PrimExpr >>	strides,
		Optional< PrimExpr >	elem_offset,
		String	storage_scope,
		int	align,
		int	offset_factor,
		String	buffer_type,
		Optional< Array< IntImm >>	axis_separators
	)

The buffer declaration function.

Parameters

shape	The type of the buffer prior to flattening.
dtype	The data type in the content of the buffer.
buffer_name	The name of the buffer.
data	The pointer to the head of the data.
strides	The strides of each dimension.
elem_offset	The offset in terms of number of dtype elements (including lanes).
storage_scope	The optional storage scope of buffer data pointer.
align	The alignment requirement of data pointer in bytes.
offset_factor	The factor of elem_offset field.
buffer_type	The buffer type.
axis_separators	The separators between input axes when generating flattened output axes.

Returns: The declared buffer.

◆ BufferStore()

void tvm::script::ir_builder::tir::BufferStore	(	Buffer	buffer,
		PrimExpr	value,
		Array< PrimExpr >	indices,
		Optional< PrimExpr >	predicate
	)

Store data in a buffer.

Parameters

buffer	The buffer.
value	The value to be stored.
indices	The indices location to be stored.
predicate	A vector mask of boolean values indicating which lanes of a vector are to be stored. The number lanes of the mask must be equal to the number of lanes in value.

◆ DeclBuffer()

DeclBufferFrame tvm::script::ir_builder::tir::DeclBuffer	(	Array< PrimExpr >	shape,
		DataType	dtype,
		String	buffer_name,
		Optional< Var >	data,
		Optional< Array< PrimExpr >>	strides,
		Optional< PrimExpr >	elem_offset,
		String	storage_scope,
		int	align,
		int	offset_factor,
		String	buffer_type,
		Optional< Array< IntImm >>	axis_separators
	)

The buffer declaration frame.

Parameters

shape	The type of the buffer prior to flattening.
dtype	The data type in the content of the buffer.
buffer_name	The name of the buffer.
data	The pointer to the head of the data.
strides	The strides of each dimension.
elem_offset	The offset in terms of number of dtype elements (including lanes).
storage_scope	The optional storage scope of buffer data pointer.
align	The alignment requirement of data pointer in bytes.
offset_factor	The factor of elem_offset field.
buffer_type	The buffer type.
axis_separators	The separators between input axes when generating flattened output axes.

Returns: The declared buffer.

◆ Else()

ElseFrame tvm::script::ir_builder::tir::Else ( )

Create an else.

Returns: The result ElseFrame.

◆ EnvThread()

Var tvm::script::ir_builder::tir::EnvThread	(	String	thread_tag,
		DataType	dtype = `DataType::Int(32)`
	)

Bind a var to thread env.

Parameters

thread_tag	The thread type tag.
dtype	The data type of the variable.

Returns: The result variable which gets bound to the thread env.

◆ Evaluate()

void tvm::script::ir_builder::tir::Evaluate ( PrimExpr value )

Evaluate the input expression.

Parameters

value The input expression to evaluate.

◆ Float16()

PrimExpr tvm::script::ir_builder::tir::Float16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float16x16()

PrimExpr tvm::script::ir_builder::tir::Float16x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float16x32()

PrimExpr tvm::script::ir_builder::tir::Float16x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float16x4()

PrimExpr tvm::script::ir_builder::tir::Float16x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float16x64()

PrimExpr tvm::script::ir_builder::tir::Float16x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float16x8()

PrimExpr tvm::script::ir_builder::tir::Float16x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32()

PrimExpr tvm::script::ir_builder::tir::Float32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32x16()

PrimExpr tvm::script::ir_builder::tir::Float32x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32x32()

PrimExpr tvm::script::ir_builder::tir::Float32x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32x4()

PrimExpr tvm::script::ir_builder::tir::Float32x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32x64()

PrimExpr tvm::script::ir_builder::tir::Float32x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float32x8()

PrimExpr tvm::script::ir_builder::tir::Float32x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FN()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FN	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FNx16()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FNx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FNx32()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FNx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FNx4()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FNx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FNx64()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FNx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float4E2M1FNx8()

PrimExpr tvm::script::ir_builder::tir::Float4E2M1FNx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64()

PrimExpr tvm::script::ir_builder::tir::Float64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64x16()

PrimExpr tvm::script::ir_builder::tir::Float64x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64x32()

PrimExpr tvm::script::ir_builder::tir::Float64x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64x4()

PrimExpr tvm::script::ir_builder::tir::Float64x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64x64()

PrimExpr tvm::script::ir_builder::tir::Float64x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float64x8()

PrimExpr tvm::script::ir_builder::tir::Float64x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FN()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FN	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FNx16()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FNx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FNx32()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FNx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FNx4()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FNx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FNx64()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FNx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E2M3FNx8()

PrimExpr tvm::script::ir_builder::tir::Float6E2M3FNx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FN()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FN	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FNx16()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FNx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FNx32()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FNx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FNx4()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FNx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FNx64()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FNx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float6E3M2FNx8()

PrimExpr tvm::script::ir_builder::tir::Float6E3M2FNx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8()

PrimExpr tvm::script::ir_builder::tir::Float8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4x16()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4x32()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4x4()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4x64()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E3M4x8()

PrimExpr tvm::script::ir_builder::tir::Float8E3M4x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZ()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZ	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZx16()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZx32()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZx4()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZx64()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3B11FNUZx8()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3B11FNUZx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FN()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FN	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZ()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZ	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZx16()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZx32()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZx4()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZx64()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNUZx8()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNUZx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNx16()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNx32()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNx4()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNx64()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3FNx8()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3FNx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3x16()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3x32()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3x4()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3x64()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E4M3x8()

PrimExpr tvm::script::ir_builder::tir::Float8E4M3x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZ()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZ	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZx16()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZx32()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZx4()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZx64()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2FNUZx8()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2FNUZx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2x16()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2x32()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2x4()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2x64()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E5M2x8()

PrimExpr tvm::script::ir_builder::tir::Float8E5M2x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNU()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNU	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNUx16()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNUx16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNUx32()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNUx32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNUx4()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNUx4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNUx64()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNUx64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8E8M0FNUx8()

PrimExpr tvm::script::ir_builder::tir::Float8E8M0FNUx8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8x16()

PrimExpr tvm::script::ir_builder::tir::Float8x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8x32()

PrimExpr tvm::script::ir_builder::tir::Float8x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8x4()

PrimExpr tvm::script::ir_builder::tir::Float8x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8x64()

PrimExpr tvm::script::ir_builder::tir::Float8x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Float8x8()

PrimExpr tvm::script::ir_builder::tir::Float8x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ FuncAttrs()

void tvm::script::ir_builder::tir::FuncAttrs ( Map< String, ffi::Any > attrs )

The PrimFunc annotation statement.

Parameters

attrs The annotations of the PrimFunc.

◆ FuncName()

void tvm::script::ir_builder::tir::FuncName ( String name )

The PrimFunc naming statement.

Parameters

name	The name of the PrimFunc.

◆ FuncRet()

Type tvm::script::ir_builder::tir::FuncRet ( Type ret_type )

The PrimFunc return type statement.

Parameters

ret_type The return type of the PrimFunc.

Returns: The return type.

◆ Grid()

ForFrame tvm::script::ir_builder::tir::Grid ( Array< PrimExpr > extents )

The grid For statement.

Parameters

extents The extents of the iteration.

Returns: The ForFrame.

◆ Handle()

Var tvm::script::ir_builder::tir::Handle	(	runtime::DataType	dtype = `runtime::DataType::Void()`,
		String	storage_scope = `"global"`,
		bool	is_size_var = `false`,
		bool	is_unknown_type = `false`
	)

inline

Create a TIR var that represents a pointer.

Parameters

dtype	The data type of the pointer.
storage_scope	The storage scope of the pointer.
is_size_var	Whether the pointer is a size var.
is_unknown_type	Used to distinguish between `PrimType(DataType::Handle())` and `PointerType(PrimType(DataType::Void()))`. If true, resolve dtype of `Void()` as `PrimType`, and if false resolve dtype of `Void()` as a `PointerType`.

Returns: The pointer.

◆ If()

IfFrame tvm::script::ir_builder::tir::If ( PrimExpr condition )

Create an if statement.

Parameters

condition The condition of if statement.

Returns: The result IfFrame.

◆ Init()

BlockInitFrame tvm::script::ir_builder::tir::Init ( )

The block initialization statement.

Returns: The BlockInitFrame.

◆ Int16()

PrimExpr tvm::script::ir_builder::tir::Int16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int16x16()

PrimExpr tvm::script::ir_builder::tir::Int16x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int16x32()

PrimExpr tvm::script::ir_builder::tir::Int16x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int16x4()

PrimExpr tvm::script::ir_builder::tir::Int16x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int16x64()

PrimExpr tvm::script::ir_builder::tir::Int16x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int16x8()

PrimExpr tvm::script::ir_builder::tir::Int16x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32()

PrimExpr tvm::script::ir_builder::tir::Int32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32x16()

PrimExpr tvm::script::ir_builder::tir::Int32x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32x32()

PrimExpr tvm::script::ir_builder::tir::Int32x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32x4()

PrimExpr tvm::script::ir_builder::tir::Int32x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32x64()

PrimExpr tvm::script::ir_builder::tir::Int32x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int32x8()

PrimExpr tvm::script::ir_builder::tir::Int32x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64()

PrimExpr tvm::script::ir_builder::tir::Int64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64x16()

PrimExpr tvm::script::ir_builder::tir::Int64x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64x32()

PrimExpr tvm::script::ir_builder::tir::Int64x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64x4()

PrimExpr tvm::script::ir_builder::tir::Int64x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64x64()

PrimExpr tvm::script::ir_builder::tir::Int64x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int64x8()

PrimExpr tvm::script::ir_builder::tir::Int64x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8()

PrimExpr tvm::script::ir_builder::tir::Int8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8x16()

PrimExpr tvm::script::ir_builder::tir::Int8x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8x32()

PrimExpr tvm::script::ir_builder::tir::Int8x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8x4()

PrimExpr tvm::script::ir_builder::tir::Int8x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8x64()

PrimExpr tvm::script::ir_builder::tir::Int8x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Int8x8()

PrimExpr tvm::script::ir_builder::tir::Int8x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ LaunchThread() [1/2]

LaunchThreadFrame tvm::script::ir_builder::tir::LaunchThread	(	String	thread_tag,
		PrimExpr	extent
	)

Launch a new thread.

Parameters

thread_tag	The thread type tag.
extent	The extent of environment thread.

Returns: The result LaunchThreadFrame.

◆ LaunchThread() [2/2]

LaunchThreadFrame tvm::script::ir_builder::tir::LaunchThread	(	Var	var,
		PrimExpr	extent
	)

Launch a thread.

Parameters

var	The iteration variable.
extent	The extent of environment thread.

Returns: The result LaunchThreadFrame.

◆ LetStmt()

LetFrame tvm::script::ir_builder::tir::LetStmt	(	PrimExpr	value,
		Optional< Type >	type_annotation = `std::nullopt`,
		Optional< Var >	var = `std::nullopt`
	)

The let binding.

Parameters

value	The value to be bound.
type_annotation	The type annotation of the let binding. Usually it is used for fine-grained var typing, particularly, PointerType.
var	The variable to be bound. If not specified, a new variable will be created.

Returns: The created LetFrame.

◆ MatchBuffer()

Buffer tvm::script::ir_builder::tir::MatchBuffer	(	ObjectRef	param,
		Array< PrimExpr >	shape,
		DataType	dtype = `DataType::Float(32)`,
		Optional< Var >	data = `std::nullopt`,
		Array< PrimExpr >	strides = `{}`,
		PrimExpr	elem_offset = `PrimExpr()`,
		String	storage_scope = `"global"`,
		int	align = `-1`,
		int	offset_factor = `0`,
		String	buffer_type = `"default"`,
		Optional< Array< IntImm >>	axis_separators = `std::nullopt`
	)

The buffer match statement.

Parameters

param	The parameter of the PrimFunc to match.
shape	The type of the buffer prior to flattening.
dtype	The data type in the content of the buffer.
data	The pointer to the head of the data.
strides	The strides of each dimension.
elem_offset	The offset in terms of number of dtype elements (including lanes).
storage_scope	The optional storage scope of buffer data pointer.
align	The alignment requirement of data pointer in bytes.
offset_factor	The factor of elem_offset field.
buffer_type	The buffer type.
axis_separators	The separators between input axes when generating flattened output axes.

Returns: The matched buffer.

Examples: /workspace/include/tvm/script/ir_builder/base.h.

◆ Parallel()

ForFrame tvm::script::ir_builder::tir::Parallel	(	PrimExpr	start,
		PrimExpr	stop,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The parallel For statement.

Parameters

start	The minimum value of iteration.
stop	The maximum value of iteration.
annotations	The optional annotations of the For statement.

Returns: The ForFrame.

◆ PrimFunc()

PrimFuncFrame tvm::script::ir_builder::tir::PrimFunc ( bool is_private )

The primitive function statement.

Returns: The PrimFuncFrame.

◆ Reads()

void tvm::script::ir_builder::tir::Reads ( Array< ObjectRef > buffer_slices )

The block buffer region reading statement.

Parameters

buffer_slices The array of buffer regions to read.

◆ Realize()

RealizeFrame tvm::script::ir_builder::tir::Realize	(	tvm::tir::BufferRegion	buffer_slice,
		String	storage_scope,
		PrimExpr	condition
	)

The realization.

Parameters

buffer_slice	The region of buffer access.
storage_scope	The storage scope associated with this realization.
condition	The condition expression.

Returns: The result RealizeFrame.

◆ Serial()

ForFrame tvm::script::ir_builder::tir::Serial	(	PrimExpr	start,
		PrimExpr	stop,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The serial For statement.

Parameters

start	The minimum value of iteration.
stop	The maximum value of iteration.
annotations	The optional annotations of the For statement.

Returns: The ForFrame.

◆ TensormapHandle()

Var tvm::script::ir_builder::tir::TensormapHandle ( )

inline

◆ Then()

ThenFrame tvm::script::ir_builder::tir::Then ( )

Create a then.

Returns: The result ThenFrame.

◆ ThreadBinding()

ForFrame tvm::script::ir_builder::tir::ThreadBinding	(	PrimExpr	start,
		PrimExpr	stop,
		String	thread,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The thread-binding For statement.

Parameters

start	The minimum value of iteration.
stop	The maximum value of iteration.
thread	The thread for loop variable to bind.
annotations	The optional annotations of the For statement.

Returns: The ForFrame.

◆ UInt16()

PrimExpr tvm::script::ir_builder::tir::UInt16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt16x16()

PrimExpr tvm::script::ir_builder::tir::UInt16x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt16x32()

PrimExpr tvm::script::ir_builder::tir::UInt16x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt16x4()

PrimExpr tvm::script::ir_builder::tir::UInt16x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt16x64()

PrimExpr tvm::script::ir_builder::tir::UInt16x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt16x8()

PrimExpr tvm::script::ir_builder::tir::UInt16x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32()

PrimExpr tvm::script::ir_builder::tir::UInt32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32x16()

PrimExpr tvm::script::ir_builder::tir::UInt32x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32x32()

PrimExpr tvm::script::ir_builder::tir::UInt32x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32x4()

PrimExpr tvm::script::ir_builder::tir::UInt32x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32x64()

PrimExpr tvm::script::ir_builder::tir::UInt32x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt32x8()

PrimExpr tvm::script::ir_builder::tir::UInt32x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64()

PrimExpr tvm::script::ir_builder::tir::UInt64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64x16()

PrimExpr tvm::script::ir_builder::tir::UInt64x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64x32()

PrimExpr tvm::script::ir_builder::tir::UInt64x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64x4()

PrimExpr tvm::script::ir_builder::tir::UInt64x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64x64()

PrimExpr tvm::script::ir_builder::tir::UInt64x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt64x8()

PrimExpr tvm::script::ir_builder::tir::UInt64x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8()

PrimExpr tvm::script::ir_builder::tir::UInt8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8x16()

PrimExpr tvm::script::ir_builder::tir::UInt8x16	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8x32()

PrimExpr tvm::script::ir_builder::tir::UInt8x32	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8x4()

PrimExpr tvm::script::ir_builder::tir::UInt8x4	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8x64()

PrimExpr tvm::script::ir_builder::tir::UInt8x64	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ UInt8x8()

PrimExpr tvm::script::ir_builder::tir::UInt8x8	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Unroll()

ForFrame tvm::script::ir_builder::tir::Unroll	(	PrimExpr	start,
		PrimExpr	stop,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The unrolled For statement.

Parameters

start	The minimum value of iteration.
stop	The maximum value of iteration.
annotations	The optional annotations of the For statement.

Returns: The ForFrame.

◆ Vectorized()

ForFrame tvm::script::ir_builder::tir::Vectorized	(	PrimExpr	start,
		PrimExpr	stop,
		Optional< Map< String, Any >>	annotations = `std::nullopt`
	)

The vectorized For statement.

Parameters

start	The minimum value of iteration.
stop	The maximum value of iteration.
annotations	The optional annotations of the For statement.

Returns: The ForFrame.

◆ Void()

PrimExpr tvm::script::ir_builder::tir::Void	(	Optional< PrimExpr >	expr = `std::nullopt`,
		bool	is_size_var = `false`
	)

inline

◆ Where()

void tvm::script::ir_builder::tir::Where ( PrimExpr predicate )

The block predicate statement.

Parameters

predicate The predicate condition.

◆ While()

WhileFrame tvm::script::ir_builder::tir::While ( PrimExpr condition )

Create a while loop.

Parameters

condition The termination condition of the loop.

Returns: The result WhileFrame.

◆ Writes()

void tvm::script::ir_builder::tir::Writes ( Array< ObjectRef > buffer_slices )

The block buffer region writing statement.

Parameters

buffer_slices The array of buffer regions to write.

Namespaces

Classes

Functions

Function Documentation

◆ Allocate()

◆ AllocateConst()

◆ AllocBuffer()

◆ Arg() [1/2]

◆ Arg() [2/2]

◆ Assert()

◆ Attr()

◆ BFloat16()

◆ BFloat16x16()

◆ BFloat16x32()

◆ BFloat16x4()

◆ BFloat16x64()

◆ BFloat16x8()

◆ BFloat32()

◆ BFloat32x16()

◆ BFloat32x32()

◆ BFloat32x4()

◆ BFloat32x64()

◆ BFloat32x8()

◆ BFloat64()

◆ BFloat64x16()

◆ BFloat64x32()

◆ BFloat64x4()

◆ BFloat64x64()

◆ BFloat64x8()

◆ BFloat8()

◆ BFloat8x16()

◆ BFloat8x32()

◆ BFloat8x4()

◆ BFloat8x64()

◆ BFloat8x8()

◆ Block()

◆ BlockAttrs()

◆ Boolean()

◆ BufferDecl()

◆ BufferStore()

◆ DeclBuffer()

◆ Else()

◆ EnvThread()

◆ Evaluate()

◆ Float16()

◆ Float16x16()

◆ Float16x32()

◆ Float16x4()

◆ Float16x64()

◆ Float16x8()

◆ Float32()

◆ Float32x16()

◆ Float32x32()

◆ Float32x4()

◆ Float32x64()

◆ Float32x8()

◆ Float4E2M1FN()

◆ Float4E2M1FNx16()

◆ Float4E2M1FNx32()

◆ Float4E2M1FNx4()

◆ Float4E2M1FNx64()

◆ Float4E2M1FNx8()

◆ Float64()

◆ Float64x16()

◆ Float64x32()

◆ Float64x4()

◆ Float64x64()

◆ Float64x8()

◆ Float6E2M3FN()

◆ Float6E2M3FNx16()

◆ Float6E2M3FNx32()

◆ Float6E2M3FNx4()

◆ Float6E2M3FNx64()

◆ Float6E2M3FNx8()

◆ Float6E3M2FN()

◆ Float6E3M2FNx16()

◆ Float6E3M2FNx32()

◆ Float6E3M2FNx4()

◆ Float6E3M2FNx64()

◆ Float6E3M2FNx8()