tile_primitive.h File Reference

TIRX tile primitive statements, operators, and reified lambda expressions. More...

#include <tvm/ffi/object.h>
#include <tvm/ir/op.h>
#include <tvm/target/target.h>
#include <tvm/tirx/exec_scope.h>
#include <tvm/tirx/stmt.h>
#include <tvm/tirx/var.h>

Include dependency graph for tile_primitive.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::tirx::LambdaExprNode
	A reified Python lambda: a list of bound variables and a body over them. More...
class	tvm::tirx::LambdaExpr
	Managed reference to LambdaExprNode. More...
class	tvm::tirx::DispatchContextNode
	The context information of the kernel required by op dispatch. More...
class	tvm::tirx::DispatchContext
	Managed reference to DispatchContextNode. More...
class	tvm::tirx::TilePrimitiveCallNode
	TIRX TilePrimitiveCall stmt. More...
class	tvm::tirx::TilePrimitiveCall
	Managed reference to TilePrimitiveCallNode. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::tirx
namespace	tvm::tirx::callback

Typedefs
using	tvm::tirx::FArgSanitizer = ffi::TypedFunction< void(tvm::Op, ffi::Array< ffi::ObjectRef >)>
	The type of the function that sanitizes the arguments of a TIRX operator.

Functions
const Op &	tvm::tirx::cast ()
	See pesudo code below:
const Op &	tvm::tirx::copy ()
	See pesudo code below:
const Op &	tvm::tirx::copy_async ()
	See pesudo code below:
const Op &	tvm::tirx::fill ()
	See pesudo code below:
const Op &	tvm::tirx::gemm ()
	See pesudo code below:
const Op &	tvm::tirx::gemm_async ()
	See pesudo code below:
const Op &	tvm::tirx::zero ()
const Op &	tvm::tirx::sqrt ()
const Op &	tvm::tirx::exp ()
const Op &	tvm::tirx::exp2 ()
const Op &	tvm::tirx::add ()
const Op &	tvm::tirx::sub ()
const Op &	tvm::tirx::mul ()
const Op &	tvm::tirx::fdiv ()
const Op &	tvm::tirx::minimum ()
const Op &	tvm::tirx::maximum ()
const Op &	tvm::tirx::reciprocal ()
const Op &	tvm::tirx::sum ()
const Op &	tvm::tirx::max ()
const Op &	tvm::tirx::min ()
const Op &	tvm::tirx::memset ()
const Op &	tvm::tirx::reduce_negate ()
const Op &	tvm::tirx::binary_reduce ()
const Op &	tvm::tirx::unary_reduce ()
const Op &	tvm::tirx::binary_chain ()
const Op &	tvm::tirx::select ()
const Op &	tvm::tirx::fma ()
const Op &	tvm::tirx::silu ()
const Op &	tvm::tirx::compose_op ()
const Op &	tvm::tirx::permute_layout ()

Variables
constexpr const char *	tvm::tirx::callback::kPrivateAlloc = "private_alloc"
	The buffers allocated by the operator.
constexpr const char *	tvm::tirx::callback::kDeviceInitStmt = "device_init_stmt"
	The initialization statement of the operator. which will be inserted at the beginning of the kernel.
constexpr const char *	tvm::tirx::callback::kHostInitStmt = "host_init_stmt"
	The initialization statement of the operator. which will be inserted at the beginning of the kernel.
constexpr const char *	tvm::tirx::callback::kPostBufferDefStmt = "post_buffer_def_stmt"
	Statements to be inserted after a specific buffer's definition (DeclBuffer/AllocBuffer). Stored as Map<BufferVar, ffi::Array<Stmt>>.

Detailed Description

TIRX tile primitive statements, operators, and reified lambda expressions.