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

#include <measure_callback.h>

Inheritance diagram for tvm::meta_schedule::PyMeasureCallbackNode:

Collaboration diagram for tvm::meta_schedule::PyMeasureCallbackNode:

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

using	FAsString = ffi::TypedFunction< String()>
	Get the measure callback function as string with name. More...

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

	TVM_DECLARE_FINAL_OBJECT_INFO (PyMeasureCallbackNode, MeasureCallbackNode)

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

	TVM_DECLARE_BASE_OBJECT_INFO (MeasureCallbackNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

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

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

FAsString	f_as_string
	The packed function to the `AsString` function. More...

Static Public Attributes
static constexpr const char *	_type_key = "meta_schedule.PyMeasureCallback"

Static Public Attributes inherited from tvm::meta_schedule::MeasureCallbackNode
static constexpr const char *	_type_key = "meta_schedule.MeasureCallback"

Detailed Description

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

Member Typedef Documentation

◆ FApply

using tvm::meta_schedule::PyMeasureCallbackNode::FApply = ffi::TypedFunction<void(const TaskScheduler& task_scheduler, int task_id, const Array<MeasureCandidate>& measure_candidates, const Array<BuilderResult>& builds, const 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.

◆ FAsString

using tvm::meta_schedule::PyMeasureCallbackNode::FAsString = ffi::TypedFunction<String()>

Get the measure callback function as string with name.

Returns: The string of the measure callback function.

Member Function Documentation

◆ Apply()

void tvm::meta_schedule::PyMeasureCallbackNode::Apply	(	const TaskScheduler &	task_scheduler,
		int	task_id,
		const Array< MeasureCandidate > &	measure_candidates,
		const Array< BuilderResult > &	builder_results,
		const 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::meta_schedule::MeasureCallbackNode.

◆ RegisterReflection()

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

inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::meta_schedule::PyMeasureCallbackNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PyMeasureCallbackNode	,
		MeasureCallbackNode
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::meta_schedule::PyMeasureCallbackNode::_type_key = "meta_schedule.PyMeasureCallback"

staticconstexpr

◆ f_apply

FApply tvm::meta_schedule::PyMeasureCallbackNode::f_apply

The packed function to the Apply function.

◆ f_as_string

FAsString tvm::meta_schedule::PyMeasureCallbackNode::f_as_string

The packed function to the AsString function.

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

include/tvm/meta_schedule/measure_callback.h

Public Types

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes