tvm
Public Member Functions | List of all members
tvm::tir::ScheduleState Class Reference

Managed reference to ScheduleStateNode. More...

#include <state.h>

Inheritance diagram for tvm::tir::ScheduleState:
Collaboration diagram for tvm::tir::ScheduleState:

Public Member Functions

 ScheduleState (IRModule mod, int debug_mask=0, bool enable_check=true)
 Construct a schedule state from an IRModule. More...
 
ScheduleStateNodeget () const
 
 TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS (ScheduleState, ObjectRef, ScheduleStateNode)
 

Detailed Description

Managed reference to ScheduleStateNode.

See also
ScheduleStateNode

Constructor & Destructor Documentation

◆ ScheduleState()

tvm::tir::ScheduleState::ScheduleState ( IRModule  mod,
int  debug_mask = 0,
bool  enable_check = true 
)
explicit

Construct a schedule state from an IRModule.

Parameters
modThe IRModule to be scheduled
debug_maskDo extra correctness checking after the class creation and each time after calling the Replace method.
enable_checkWhether enables prerequisite checks for schedule primitives.

Member Function Documentation

◆ get()

ScheduleStateNode* tvm::tir::ScheduleState::get ( ) const
inline
Returns
The mutable pointer to the ScheduleStateNode

◆ TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS()

tvm::tir::ScheduleState::TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS ( ScheduleState  ,
ObjectRef  ,
ScheduleStateNode   
)

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