Namespaces
	axis

Classes
class	AllocateConstFrame
	Managed reference to AllocateConstFrameNode. More...

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

class	AllocateFrame
	Managed reference to AllocateFrameNode. More...

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

class	AssertFrame
	Managed reference to AssertFrameNode. More...

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

class	AttrFrame
	Managed reference to AttrFrameNode. More...

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

class	BlockFrame
	Managed reference to BlockFrameNode. More...

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

class	BlockInitFrame
	Managed reference to BlockInitFrameNode. More...

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

class	DeclBufferFrame

class	DeclBufferFrameNode

class	ElseFrame
	Managed reference to ElseFrameNode. More...

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

class	ForFrame
	Managed reference to ForFrameNode. More...

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

class	IfFrame
	Managed reference to IfFrameNode. More...

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

class	LaunchThreadFrame
	Managed reference to LaunchThreadFrameNode. More...

class	LaunchThreadFrameNode
	The LaunchThreadFrameNode. More...

class	LetFrame
	Managed reference to LetFrameNode. More...

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

class	PrimFuncFrame
	Managed reference to PrimFuncFrameNode. More...

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

class	RealizeFrame
	Managed reference to RealizeFrameNode. More...

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

class	ThenFrame
	Managed reference to ThenFrameNode. More...

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

class	TIRFrame
	Managed reference to TIRFrameNode. More...

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

class	WhileFrame
	Managed reference to WhileFrameNode. More...

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

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 ()
	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, ObjectRef > 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=NullOpt, Array< PrimExpr > strides={}, PrimExpr elem_offset=PrimExpr(), String storage_scope="global", int align=-1, int offset_factor=0, String buffer_type="default", Array< IntImm > axis_separators={})
	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, ObjectRef > attrs)
	The block annotation statement. More...

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

ForFrame	Serial (PrimExpr start, PrimExpr stop, Optional< Map< String, ObjectRef >> annotations=NullOpt)
	The serial For statement. More...

ForFrame	Parallel (PrimExpr start, PrimExpr stop, Optional< Map< String, ObjectRef >> annotations=NullOpt)
	The parallel For statement. More...

ForFrame	Vectorized (PrimExpr start, PrimExpr stop, Optional< Map< String, ObjectRef >> annotations=NullOpt)
	The vectorized For statement. More...

ForFrame	Unroll (PrimExpr start, PrimExpr stop, Optional< Map< String, ObjectRef >> annotations=NullOpt)
	The unrolled For statement. More...

ForFrame	ThreadBinding (PrimExpr start, PrimExpr stop, String thread, Optional< Map< String, ObjectRef >> annotations=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	Let (Var var, PrimExpr value)
	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=NullOpt, Optional< Map< String, ObjectRef >> annotations=NullOpt)
	The allocate node. More...

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

AttrFrame	Attr (ObjectRef 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...

Var	EnvThread (String thread_tag)
	Bind a var to thread env. More...

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

void	Prefetch (Buffer buffer, Array< Range > bounds)
	The prefetch hint for a buffer. More...

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

PrimExpr	Ptr (runtime::DataType dtype, String storage_scope="global")
	The pointer declaration function. More...

PrimExpr	Float8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float8x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float8x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float8x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float8x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float8x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float16x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float32x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Float64x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt8x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt16x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt32x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	UInt64x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int8x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int16x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int32x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64x4 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64x8 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64x16 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64x32 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Int64x64 (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Boolean (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Handle (Optional< PrimExpr > expr=NullOpt)

PrimExpr	Void (Optional< PrimExpr > expr=NullOpt)

Function Documentation

◆ Allocate()

AllocateFrame tvm::script::ir_builder::tir::Allocate	(	Array< PrimExpr >	extents,
		DataType	dtype,
		String	storage_scope = `""`,
		Optional< PrimExpr >	condition = `NullOpt`,
		Optional< Map< String, ObjectRef >>	annotations = `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, ObjectRef >>	annotations = `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 = `NullOpt`,
		Array< PrimExpr >	strides = `{}`,
		PrimExpr	elem_offset = `PrimExpr()`,
		String	storage_scope = `""`,
		int	align = `-1`,
		int	offset_factor = `0`,
		String	buffer_type = `"default"`,
		Array< IntImm >	axis_separators = `{}`
	)

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]

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.

◆ Arg() [2/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.

◆ 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	(	ObjectRef	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.

◆ 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, ObjectRef > attrs )

The block annotation statement.

Parameters

attrs The annotation of the block.

◆ Boolean()

PrimExpr tvm::script::ir_builder::tir::Boolean ( Optional< PrimExpr > expr = NullOpt )

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
	)

Store data in a buffer.

Parameters

buffer	The buffer.
value	The value to be stored.
indices	The indices location to be stored.

◆ 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 )

Bind a var to thread env.

Parameters

thread_tag The thread type tag.

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 = NullOpt )

inline

◆ Float16x16()

PrimExpr tvm::script::ir_builder::tir::Float16x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float16x32()

PrimExpr tvm::script::ir_builder::tir::Float16x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float16x4()

PrimExpr tvm::script::ir_builder::tir::Float16x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float16x64()

PrimExpr tvm::script::ir_builder::tir::Float16x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float16x8()

PrimExpr tvm::script::ir_builder::tir::Float16x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32()

PrimExpr tvm::script::ir_builder::tir::Float32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32x16()

PrimExpr tvm::script::ir_builder::tir::Float32x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32x32()

PrimExpr tvm::script::ir_builder::tir::Float32x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32x4()

PrimExpr tvm::script::ir_builder::tir::Float32x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32x64()

PrimExpr tvm::script::ir_builder::tir::Float32x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float32x8()

PrimExpr tvm::script::ir_builder::tir::Float32x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64()

PrimExpr tvm::script::ir_builder::tir::Float64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64x16()

PrimExpr tvm::script::ir_builder::tir::Float64x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64x32()

PrimExpr tvm::script::ir_builder::tir::Float64x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64x4()

PrimExpr tvm::script::ir_builder::tir::Float64x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64x64()

PrimExpr tvm::script::ir_builder::tir::Float64x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float64x8()

PrimExpr tvm::script::ir_builder::tir::Float64x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8()

PrimExpr tvm::script::ir_builder::tir::Float8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8x16()

PrimExpr tvm::script::ir_builder::tir::Float8x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8x32()

PrimExpr tvm::script::ir_builder::tir::Float8x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8x4()

PrimExpr tvm::script::ir_builder::tir::Float8x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8x64()

PrimExpr tvm::script::ir_builder::tir::Float8x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Float8x8()

PrimExpr tvm::script::ir_builder::tir::Float8x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ FuncAttrs()

void tvm::script::ir_builder::tir::FuncAttrs ( Map< String, ObjectRef > 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()

PrimExpr tvm::script::ir_builder::tir::Handle ( Optional< PrimExpr > expr = NullOpt )

inline

◆ 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 = NullOpt )

inline

◆ Int16x16()

PrimExpr tvm::script::ir_builder::tir::Int16x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int16x32()

PrimExpr tvm::script::ir_builder::tir::Int16x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int16x4()

PrimExpr tvm::script::ir_builder::tir::Int16x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int16x64()

PrimExpr tvm::script::ir_builder::tir::Int16x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int16x8()

PrimExpr tvm::script::ir_builder::tir::Int16x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32()

PrimExpr tvm::script::ir_builder::tir::Int32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32x16()

PrimExpr tvm::script::ir_builder::tir::Int32x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32x32()

PrimExpr tvm::script::ir_builder::tir::Int32x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32x4()

PrimExpr tvm::script::ir_builder::tir::Int32x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32x64()

PrimExpr tvm::script::ir_builder::tir::Int32x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int32x8()

PrimExpr tvm::script::ir_builder::tir::Int32x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64()

PrimExpr tvm::script::ir_builder::tir::Int64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64x16()

PrimExpr tvm::script::ir_builder::tir::Int64x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64x32()

PrimExpr tvm::script::ir_builder::tir::Int64x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64x4()

PrimExpr tvm::script::ir_builder::tir::Int64x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64x64()

PrimExpr tvm::script::ir_builder::tir::Int64x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int64x8()

PrimExpr tvm::script::ir_builder::tir::Int64x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8()

PrimExpr tvm::script::ir_builder::tir::Int8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8x16()

PrimExpr tvm::script::ir_builder::tir::Int8x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8x32()

PrimExpr tvm::script::ir_builder::tir::Int8x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8x4()

PrimExpr tvm::script::ir_builder::tir::Int8x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8x64()

PrimExpr tvm::script::ir_builder::tir::Int8x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Int8x8()

PrimExpr tvm::script::ir_builder::tir::Int8x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ LaunchThread()

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.

◆ Let()

LetFrame tvm::script::ir_builder::tir::Let	(	Var	var,
		PrimExpr	value
	)

The let binding.

Parameters

var	The variable to bind.
value	The value to be bound.

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 = `NullOpt`,
		Array< PrimExpr >	strides = `{}`,
		PrimExpr	elem_offset = `PrimExpr()`,
		String	storage_scope = `"global"`,
		int	align = `-1`,
		int	offset_factor = `0`,
		String	buffer_type = `"default"`,
		Array< IntImm >	axis_separators = `{}`
	)

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, ObjectRef >>	annotations = `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.

◆ Prefetch()

void tvm::script::ir_builder::tir::Prefetch	(	Buffer	buffer,
		Array< Range >	bounds
	)

The prefetch hint for a buffer.

Parameters

buffer	The buffer to be prefetched.
bounds	The bounds to be prefetched.

◆ PrimFunc()

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

The primitive function statement.

Returns: The PrimFuncFrame.

◆ Ptr()

PrimExpr tvm::script::ir_builder::tir::Ptr	(	runtime::DataType	dtype,
		String	storage_scope = `"global"`
	)

The pointer declaration function.

Parameters

dtype	The data type of the pointer.
storage_scope	The storage scope of the pointer.

Returns: The pointer.

◆ 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, ObjectRef >>	annotations = `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.

◆ 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, ObjectRef >>	annotations = `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 = NullOpt )

inline

◆ UInt16x16()

PrimExpr tvm::script::ir_builder::tir::UInt16x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt16x32()

PrimExpr tvm::script::ir_builder::tir::UInt16x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt16x4()

PrimExpr tvm::script::ir_builder::tir::UInt16x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt16x64()

PrimExpr tvm::script::ir_builder::tir::UInt16x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt16x8()

PrimExpr tvm::script::ir_builder::tir::UInt16x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32()

PrimExpr tvm::script::ir_builder::tir::UInt32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32x16()

PrimExpr tvm::script::ir_builder::tir::UInt32x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32x32()

PrimExpr tvm::script::ir_builder::tir::UInt32x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32x4()

PrimExpr tvm::script::ir_builder::tir::UInt32x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32x64()

PrimExpr tvm::script::ir_builder::tir::UInt32x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt32x8()

PrimExpr tvm::script::ir_builder::tir::UInt32x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64()

PrimExpr tvm::script::ir_builder::tir::UInt64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64x16()

PrimExpr tvm::script::ir_builder::tir::UInt64x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64x32()

PrimExpr tvm::script::ir_builder::tir::UInt64x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64x4()

PrimExpr tvm::script::ir_builder::tir::UInt64x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64x64()

PrimExpr tvm::script::ir_builder::tir::UInt64x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt64x8()

PrimExpr tvm::script::ir_builder::tir::UInt64x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8()

PrimExpr tvm::script::ir_builder::tir::UInt8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8x16()

PrimExpr tvm::script::ir_builder::tir::UInt8x16 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8x32()

PrimExpr tvm::script::ir_builder::tir::UInt8x32 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8x4()

PrimExpr tvm::script::ir_builder::tir::UInt8x4 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8x64()

PrimExpr tvm::script::ir_builder::tir::UInt8x64 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ UInt8x8()

PrimExpr tvm::script::ir_builder::tir::UInt8x8 ( Optional< PrimExpr > expr = NullOpt )

inline

◆ Unroll()

ForFrame tvm::script::ir_builder::tir::Unroll	(	PrimExpr	start,
		PrimExpr	stop,
		Optional< Map< String, ObjectRef >>	annotations = `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, ObjectRef >>	annotations = `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 = NullOpt )

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()

◆ Block()

◆ BlockAttrs()

◆ Boolean()

◆ BufferDecl()

◆ BufferStore()

◆ DeclBuffer()

◆ Else()

◆ EnvThread()

◆ Evaluate()

◆ Float16()

◆ Float16x16()

◆ Float16x32()

◆ Float16x4()

◆ Float16x64()

◆ Float16x8()

◆ Float32()

◆ Float32x16()

◆ Float32x32()

◆ Float32x4()

◆ Float32x64()

◆ Float32x8()

◆ Float64()

◆ Float64x16()

◆ Float64x32()

◆ Float64x4()

◆ Float64x64()

◆ Float64x8()

◆ Float8()

◆ Float8x16()

◆ Float8x32()

◆ Float8x4()

◆ Float8x64()

◆ Float8x8()

◆ FuncAttrs()

◆ FuncName()

◆ FuncRet()

◆ Grid()

◆ Handle()

◆ If()

◆ Init()

◆ Int16()

◆ Int16x16()

◆ Int16x32()

◆ Int16x4()

◆ Int16x64()

◆ Int16x8()

◆ Int32()

◆ Int32x16()

◆ Int32x32()

◆ Int32x4()

◆ Int32x64()

◆ Int32x8()

◆ Int64()

◆ Int64x16()

◆ Int64x32()

◆ Int64x4()

◆ Int64x64()

◆ Int64x8()

◆ Int8()

◆ Int8x16()

◆ Int8x32()

◆ Int8x4()

◆ Int8x64()

◆ Int8x8()

◆ LaunchThread()

◆ Let()

◆ MatchBuffer()

◆ Parallel()

◆ Prefetch()