tvm
Public Member Functions | Static Public Member Functions | List of all members
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_DEFINE_MUTABLE_OBJECT_REF_METHODS (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
modThe IRModule to be scheduled
seedThe seed value for schedule's random state
debug_maskDo extra correctness checking after the class creation and each time after calling the Replace method.
error_render_levelThe level of error rendering
enable_checkWhether 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
modThe IRModule to be scheduled
seedThe seed value for schedule's random state
debug_maskDo extra correctness checking after the class creation and each time after calling the Replace method.
error_render_levelThe level of error rendering
enable_checkWhether 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_DEFINE_MUTABLE_OBJECT_REF_METHODS()

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

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