The measure callback with customized methods on the python-side. More...

#include <measure_callback.h>

Inheritance diagram for tvm::s_tir::meta_schedule::PyMeasureCallbackNode:

Collaboration diagram for tvm::s_tir::meta_schedule::PyMeasureCallbackNode:

Public Types
using	FApply = ffi::TypedFunction< void(const TaskScheduler &task_scheduler, int task_id, const ffi::Array< MeasureCandidate > &measure_candidates, const ffi::Array< BuilderResult > &builds, const ffi::Array< RunnerResult > &results)>
	Apply a measure callback to the given schedule. More...

Public Member Functions
void	Apply (const TaskScheduler &task_scheduler, int task_id, const ffi::Array< MeasureCandidate > &measure_candidates, const ffi::Array< BuilderResult > &builds, const ffi::Array< RunnerResult > &results)
	Apply a measure callback rule with given arguments. More...

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("s_tir.meta_schedule.PyMeasureCallback", PyMeasureCallbackNode, MeasureCallbackNode)

Public Member Functions inherited from tvm::s_tir::meta_schedule::MeasureCallbackNode
virtual	~MeasureCallbackNode ()=default
	Virtual destructor. More...

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

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::s_tir::meta_schedule::MeasureCallbackNode
static void	RegisterReflection ()

Public Attributes
FApply	f_apply
	The packed function to the `Apply` function. More...

Additional Inherited Members
Static Public Attributes inherited from tvm::s_tir::meta_schedule::MeasureCallbackNode
static constexpr const bool	_type_mutable = true

Detailed Description

The measure callback with customized methods on the python-side.

Member Typedef Documentation

◆ FApply

using tvm::s_tir::meta_schedule::PyMeasureCallbackNode::FApply = ffi::TypedFunction<void(const TaskScheduler& task_scheduler, int task_id, const ffi::Array<MeasureCandidate>& measure_candidates, const ffi::Array<BuilderResult>& builds, const ffi::Array<RunnerResult>& results)>

Apply a measure callback to the given schedule.

Parameters

task_scheduler	The task scheduler.
tasks	The list of tune context to process.
measure_candidates	The measure candidates.
builds	The builder results by building the measure candidates.
results	The runner results by running the built measure candidates.

Returns: Whether the measure callback was successfully applied.

Member Function Documentation

◆ Apply()

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

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.

Implements tvm::s_tir::meta_schedule::MeasureCallbackNode.

◆ RegisterReflection()

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

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::s_tir::meta_schedule::PyMeasureCallbackNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"s_tir.meta_schedule.PyMeasureCallback"	,
		PyMeasureCallbackNode	,
		MeasureCallbackNode
	)

Member Data Documentation

◆ f_apply

FApply tvm::s_tir::meta_schedule::PyMeasureCallbackNode::f_apply

The packed function to the Apply function.

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

include/tvm/s_tir/meta_schedule/measure_callback.h

Public Types

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members