tvm::tir::Schedule Class Reference

Managed reference to ScheduleNode. More...

#include <schedule.h>

Inheritance diagram for tvm::tir::Schedule:

Collaboration diagram for tvm::tir::Schedule:

Public Member Functions
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (Schedule, runtime::ObjectRef, ScheduleNode)

Static Public Member Functions
static Schedule	Concrete (IRModule mod, support::LinearCongruentialEngine::TRandState seed, int debug_mask, ScheduleErrorRenderLevel error_render_level, bool enable_check=true)
	Construct a concrete TensorIR schedule from an IRModule. More...
static Schedule	Traced (IRModule mod, support::LinearCongruentialEngine::TRandState seed, int debug_mask, ScheduleErrorRenderLevel error_render_level, bool enable_check=true)
	Construct a traced concrete TensorIR schedule from an IRModule. More...

Detailed Description

Managed reference to ScheduleNode.

A schedule is a set of transformations that change the order of computation but preserve the semantics of computation. Some example of schedules: 1) Split a loop into two; 2) Reorder two loops; 3) Inline the computation of a specific buffer into its consumer

The schedule class stores auxiliary information to schedule correctly and efficiently.

Link to tutorial: https://tvm.apache.org/docs/tutorials/language/schedule_primitives.html

See also

ScheduleNode

Member Function Documentation

Concrete()

static Schedule tvm::tir::Schedule::Concrete ( IRModule  mod, support::LinearCongruentialEngine::TRandState  seed, int  debug_mask, ScheduleErrorRenderLevel  error_render_level, bool  enable_check = true  )

static

Construct a concrete TensorIR schedule from an IRModule.

Parameters

mod	The IRModule to be scheduled
seed	The seed value for schedule's random state
debug_mask	Do extra correctness checking after the class creation and each time after calling the Replace method.
error_render_level	The level of error rendering
enable_check	Whether to enable some prequisite checks for schedule primitives, it's user's duty to guarantee the schedule correctness if we disable the checks.

Returns

The concrete schedule created

See also

ScheduleDebugMask

Note

The checks performed includes: 1) VerifySRefTree 2) VerifyCachedFlags

Traced()

static Schedule tvm::tir::Schedule::Traced ( IRModule  mod, support::LinearCongruentialEngine::TRandState  seed, int  debug_mask, ScheduleErrorRenderLevel  error_render_level, bool  enable_check = true  )

static

Construct a traced concrete TensorIR schedule from an IRModule.

Parameters

mod	The IRModule to be scheduled
seed	The seed value for schedule's random state
debug_mask	Do extra correctness checking after the class creation and each time after calling the Replace method.
error_render_level	The level of error rendering
enable_check	Whether to enable prequisite checks for schedule primitives.

Returns

The concrete schedule created

See also

ScheduleDebugMask

Note

The checks performed include: 1) VerifySRefTree 2) VerifyCachedFlags

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::tir::Schedule::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	Schedule	,
		runtime::ObjectRef	,
		ScheduleNode
	)

---

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

• include/tvm/tir/schedule/schedule.h