tvm
Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
tvm::meta_schedule::MeasureCallbackNode Class Referenceabstract

Rules to apply after measure results is available. More...

#include <measure_callback.h>

Inheritance diagram for tvm::meta_schedule::MeasureCallbackNode:
Collaboration diagram for tvm::meta_schedule::MeasureCallbackNode:

Public Member Functions

virtual ~MeasureCallbackNode ()=default
 Virtual destructor. More...
 
virtual void Apply (const TaskScheduler &task_scheduler, int task_id, const Array< MeasureCandidate > &measure_candidates, const Array< BuilderResult > &builder_results, const Array< RunnerResult > &runner_results)=0
 Apply a measure callback rule with given arguments. More...
 
 TVM_DECLARE_BASE_OBJECT_INFO (MeasureCallbackNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Static Public Attributes

static constexpr const char * _type_key = "meta_schedule.MeasureCallback"
 

Detailed Description

Rules to apply after measure results is available.

Constructor & Destructor Documentation

◆ ~MeasureCallbackNode()

virtual tvm::meta_schedule::MeasureCallbackNode::~MeasureCallbackNode ( )
virtualdefault

Virtual destructor.

Member Function Documentation

◆ Apply()

virtual void tvm::meta_schedule::MeasureCallbackNode::Apply ( const TaskScheduler task_scheduler,
int  task_id,
const Array< MeasureCandidate > &  measure_candidates,
const Array< BuilderResult > &  builder_results,
const Array< RunnerResult > &  runner_results 
)
pure virtual

Apply a measure callback rule with given arguments.

Parameters
task_schedulerThe task scheduler.
task_idThe id of the task (tune context) to apply measure callbacks.
measure_candidatesThe measure candidates.
builder_resultsThe builder results by building the measure candidates.
runner_resultsThe runner results by running the built measure candidates.

Implemented in tvm::meta_schedule::PyMeasureCallbackNode.

◆ RegisterReflection()

static void tvm::meta_schedule::MeasureCallbackNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_BASE_OBJECT_INFO()

tvm::meta_schedule::MeasureCallbackNode::TVM_DECLARE_BASE_OBJECT_INFO ( MeasureCallbackNode  ,
Object   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::meta_schedule::MeasureCallbackNode::_type_key = "meta_schedule.MeasureCallback"
staticconstexpr

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