The context information of the kernel required by op schedule. More...

#include <tirx_op.h>

Inheritance diagram for tvm::tirx::ScheduleContextNode:

Collaboration diagram for tvm::tirx::ScheduleContextNode:

Public Member Functions
void	AddAllocBuffer (Buffer buffer)
	Add a buffer to be allocated in the kernel. More...

void	AddInitStmt (Stmt stmt, bool host=false)
	Add an initialization statement to be inserted. More...

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("tirx.ScheduleContext", ScheduleContextNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Target	target
	The target of the kernel. More...

ExecScope	exec_scope
	The exec scope of the operator. More...

ffi::Map< ffi::String, IterVar >	launch_params
	The kernel launch parameters. More...

ffi::Map< Var, Range >	var_range_map
	A map from loop variables to their ranges. More...

bool	alloc_only
	Whether the schedule context is only used for buffer allocation. More...

ffi::Map< ffi::String, ffi::ObjectRef >	callbacks
	Callback to be handled when the operator is scheduled. More...

Detailed Description

The context information of the kernel required by op schedule.

Member Function Documentation

◆ AddAllocBuffer()

void tvm::tirx::ScheduleContextNode::AddAllocBuffer ( Buffer buffer )

Add a buffer to be allocated in the kernel.

◆ AddInitStmt()

void tvm::tirx::ScheduleContextNode::AddInitStmt	(	Stmt	stmt,
		bool	host = `false`
	)

Add an initialization statement to be inserted.

Parameters

stmt	The statement to be inserted.
host	Whether the statement is a host statement. If True, the statement will be added to the host code (before the kernel). If False, the statement will be added to the kernel body (at the beginning of the kernel).

◆ RegisterReflection()

static void tvm::tirx::ScheduleContextNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::tirx::ScheduleContextNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"tirx.ScheduleContext"	,
		ScheduleContextNode	,
		ffi::Object
	)

Member Data Documentation

◆ alloc_only

bool tvm::tirx::ScheduleContextNode::alloc_only

Whether the schedule context is only used for buffer allocation.

◆ callbacks

ffi::Map<ffi::String, ffi::ObjectRef> tvm::tirx::ScheduleContextNode::callbacks

Callback to be handled when the operator is scheduled.

◆ exec_scope

ExecScope tvm::tirx::ScheduleContextNode::exec_scope

The exec scope of the operator.

◆ launch_params

ffi::Map<ffi::String, IterVar> tvm::tirx::ScheduleContextNode::launch_params

The kernel launch parameters.

◆ target

Target tvm::tirx::ScheduleContextNode::target

The target of the kernel.

◆ var_range_map

ffi::Map<Var, Range> tvm::tirx::ScheduleContextNode::var_range_map

A map from loop variables to their ranges.

The documentation for this class was generated from the following file:

include/tvm/tirx/tirx_op.h

Public Member Functions

Static Public Member Functions

Public Attributes