Rules to apply after measure results is available.
More...
#include <measure_callback.h>
Rules to apply after measure results is available.
◆ ~MeasureCallbackNode()
virtual tvm::meta_schedule::MeasureCallbackNode::~MeasureCallbackNode |
( |
| ) |
|
|
virtualdefault |
◆ Apply()
virtual void tvm::meta_schedule::MeasureCallbackNode::Apply |
( |
const TaskScheduler & |
task_scheduler, |
|
|
int |
task_id, |
|
|
const ffi::Array< MeasureCandidate > & |
measure_candidates, |
|
|
const ffi::Array< BuilderResult > & |
builder_results, |
|
|
const ffi::Array< RunnerResult > & |
runner_results |
|
) |
| |
|
pure virtual |
Apply a measure callback rule with given arguments.
- Parameters
-
task_scheduler | The task scheduler. |
task_id | The id of the task (tune context) to apply measure callbacks. |
measure_candidates | The measure candidates. |
builder_results | The builder results by building the measure candidates. |
runner_results | The runner results by running the built measure candidates. |
Implemented in tvm::meta_schedule::PyMeasureCallbackNode.
◆ RegisterReflection()
static void tvm::meta_schedule::MeasureCallbackNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO()
tvm::meta_schedule::MeasureCallbackNode::TVM_FFI_DECLARE_OBJECT_INFO |
( |
"meta_schedule.MeasureCallback" |
, |
|
|
MeasureCallbackNode |
, |
|
|
Object |
|
|
) |
| |
◆ _type_mutable
constexpr const bool tvm::meta_schedule::MeasureCallbackNode::_type_mutable = true |
|
staticconstexpr |
The documentation for this class was generated from the following file: