tvm
|
Managed reference to ScheduleNode. More...
#include <schedule.h>
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... | |
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
|
static |
Construct a concrete TensorIR schedule from an IRModule.
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. |
|
static |
Construct a traced concrete TensorIR schedule from an IRModule.
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. |
tvm::tir::Schedule::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS | ( | Schedule | , |
runtime::ObjectRef | , | ||
ScheduleNode | |||
) |