tvm::s_tir::meta_schedule::MeasureCallbackNode Class Reference

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

#include <measure_callback.h>

Inheritance diagram for tvm::s_tir::meta_schedule::MeasureCallbackNode:

Collaboration diagram for tvm::s_tir::meta_schedule::MeasureCallbackNode:

Public Member Functions
virtual	~MeasureCallbackNode ()=default
	Virtual destructor.
virtual void	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)=0
	Apply a measure callback rule with given arguments.
	TVM_FFI_DECLARE_OBJECT_INFO ("s_tir.meta_schedule.MeasureCallback", MeasureCallbackNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Static Public Attributes
static constexpr const bool	_type_mutable = true

Detailed Description

Rules to apply after measure results is available.

Constructor & Destructor Documentation

~MeasureCallbackNode()

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

virtualdefault

Virtual destructor.

Member Function Documentation

Apply()

virtual void tvm::s_tir::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::s_tir::meta_schedule::PyMeasureCallbackNode.

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO()

tvm::s_tir::meta_schedule::MeasureCallbackNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"s_tir.meta_schedule.MeasureCallback"	,
		MeasureCallbackNode	,
		ffi::Object
	)

Member Data Documentation

_type_mutable

constexpr const bool tvm::s_tir::meta_schedule::MeasureCallbackNode::_type_mutable = true

staticconstexpr

---

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

• include/tvm/s_tir/meta_schedule/measure_callback.h