tvm::tirx::tile Namespace Reference

Dispatch interface for lowering tile instruction Calls. More...

Namespaces
namespace	callback
namespace	op_attr

Classes
class	DispatchContext
	Managed reference to DispatchContextObj. More...
class	DispatchContextObj
	The context information of the kernel required by op dispatch. More...

Typedefs
using	FTileOpDispatch = ffi::TypedFunction< Function(const Call &, const DispatchContext &)>
	Lower a tile instruction Call using its execution context.

Functions
TVM_TIRX_DLL Function	Dispatch (const Call &call, const DispatchContext &context)
	Invoke the Call's FTileOpDispatch attribute.
TVM_TIRX_DLL void	ConfigureTensorInstruction (Op op, ffi::Function validator)
	Attach the native Call contract and backend validator to a tile instruction.

Detailed Description

Dispatch interface for lowering tile instruction Calls.

Operator registration contracts for tile instructions.

Typedef Documentation

FTileOpDispatch

using tvm::tirx::tile::FTileOpDispatch = typedef ffi::TypedFunction<Function(const Call&, const DispatchContext&)>

Lower a tile instruction Call using its execution context.

Registering this attribute identifies an Op as a tile instruction. The dispatcher can be implemented in C++ or registered from Python.

Function Documentation

ConfigureTensorInstruction()

TVM_TIRX_DLL void tvm::tirx::tile::ConfigureTensorInstruction	(	Op	op,
		ffi::Function	validator
	)

Attach the native Call contract and backend validator to a tile instruction.

Dispatch()

TVM_TIRX_DLL Function tvm::tirx::tile::Dispatch	(	const Call &	call,
		const DispatchContext &	context
	)

Invoke the Call's FTileOpDispatch attribute.

Parameters

call	The tile instruction to lower.
context	The operator's execution context and accumulated callbacks.

Returns

The instruction implementation, whose body is expanded by TileDispatch.