tirx_op.h File Reference

TIRX built-in operators. More...

#include <tvm/ir/op.h>
#include <tvm/target/target.h>
#include <tvm/tirx/exec_scope.h>
#include <tvm/tirx/stmt.h>
#include <tvm/tirx/tirx_stmt.h>

Include dependency graph for tirx_op.h:

Go to the source code of this file.

Classes
class	tvm::tirx::ScheduleContextNode
	The context information of the kernel required by op schedule. More...
class	tvm::tirx::ScheduleContext
	Managed reference to ScheduleContextNode. 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...

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	tvm::tirx
	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. More...
using	tvm::tirx::FOpScheduler = ffi::TypedFunction< Stmt(tvm::Op, ffi::Array< ffi::ObjectRef >, ScheduleContext)>
	The type of the function that schedules a TIRX operator. More...

Functions
const Op &	tvm::tirx::cast ()
	See pesudo code below: More...
const Op &	tvm::tirx::permute_dims ()
	See pesudo code below: More...
const Op &	tvm::tirx::copy ()
	See pesudo code below: More...
const Op &	tvm::tirx::copy_async ()
	See pesudo code below: More...
const Op &	tvm::tirx::fill ()
	See pesudo code below: More...
const Op &	tvm::tirx::gemm ()
	See pesudo code below: More...
const Op &	tvm::tirx::gemm_async ()
	See pesudo code below: More...
const Op &	tvm::tirx::zero ()
const Op &	tvm::tirx::sqrt ()
const Op &	tvm::tirx::exp ()
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::tvm_kernel_replace_point ()
	See pesudo code below: More...

Variables
constexpr const char *	tvm::tirx::callback::kPrivateAlloc = "private_alloc"
	The buffers allocated by the operator. More...
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. More...
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. More...
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<Buffer, ffi::Array<Stmt>>. More...

Detailed Description

TIRX built-in operators.

TIRX statements.