tvm::meta_schedule::TaskSchedulerNode Class Reference

The abstract interface of task schedulers. More...

#include <task_scheduler.h>

Inheritance diagram for tvm::meta_schedule::TaskSchedulerNode:

Collaboration diagram for tvm::meta_schedule::TaskSchedulerNode:

Public Member Functions
virtual	~TaskSchedulerNode ()=default
	The default destructor. More...
virtual int	NextTaskId ()=0
	Fetch the next task id. More...
virtual ffi::Array< RunnerResult >	JoinRunningTask (int task_id)
	Wait until the task is finished. More...
virtual void	Tune (ffi::Array< TuneContext > tasks, ffi::Array< FloatImm > task_weights, int max_trials_global, int max_trials_per_task, int num_trials_per_iter, Builder builder, Runner runner, ffi::Array< MeasureCallback > measure_callbacks, ffi::Optional< Database > database, ffi::Optional< CostModel > cost_model)
	Jointly tune a given list of tasks. More...
void	TerminateTask (int task_id)
	Terminate a task. More...
void	TouchTask (int task_id)
	Touch the task and update its status. More...
void	PrintTuningStatistics ()
	Print out a human-readable format of the tuning statistics. More...
	TVM_FFI_DECLARE_OBJECT_INFO ("meta_schedule.TaskScheduler", TaskSchedulerNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Function	logger
	The tuning task's logging function. More...
ffi::Array< TaskRecord >	tasks_
	Records for each task. More...
ffi::Array< MeasureCallback >	measure_callbacks_
	The list of measure callbacks of the scheduler. More...
ffi::Optional< Database >	database_
	The database used in tuning. More...
ffi::Optional< CostModel >	cost_model_
	The cost model used in tuning. More...
int	remaining_tasks_
	The number of remaining tasks to be tuned. More...

Static Public Attributes
static constexpr const bool	_type_mutable = true

Detailed Description

The abstract interface of task schedulers.

Note

The relationship between SpaceGenerator and other classes are as follows: +-----------------------------------------------------------—+ +–+--------------------------------------------------------—+ | +–+---------------— Tune Context --------------------------—+ | | | +------------------—+ | | | | | | Generate | | | | | Space Generator +-----------—+ | | | | | | | | | | | +------------------—+ v | | | | Design Space | | | | +------------------—+ | | | | | Generate | | Pretuning | | | | | +--------—+ Search Strategy |<----------—+ | | | | | | | | +–+ | | +------------------—+ +–+ +-—+-------------------------------------------------------—+ | | +-—+-------------— Managed By Task Scheduler ------------------—+ | | +--------—+ | | | Send to | | Send to | | v +----------—+| Builder +-------—+ | | Measure Candidate | Builder | | Runner | | | | | +--------—+ | | | | +---------—+---------—+ | | | | | | +--------—+ | | | +-—>| Task Scheduler | | | | | | | | | Runner |<--—+ | | +----------------------—+ | | | | ^ +--—+--—+ | | | | | | +-— Runner Future <----—+ | +------------------------------------------------------------------—+

Constructor & Destructor Documentation

~TaskSchedulerNode()

virtual tvm::meta_schedule::TaskSchedulerNode::~TaskSchedulerNode ( )

virtualdefault

The default destructor.

Member Function Documentation

JoinRunningTask()

virtual ffi::Array<RunnerResult> tvm::meta_schedule::TaskSchedulerNode::JoinRunningTask ( int  task_id )

virtual

Wait until the task is finished.

Parameters

task_id

The task id to be joined.

Returns

The results from the runner.

Reimplemented in tvm::meta_schedule::PyTaskSchedulerNode.

NextTaskId()

virtual int tvm::meta_schedule::TaskSchedulerNode::NextTaskId ( )

pure virtual

Fetch the next task id.

Returns

The next task id.

Implemented in tvm::meta_schedule::PyTaskSchedulerNode.

PrintTuningStatistics()

void tvm::meta_schedule::TaskSchedulerNode::PrintTuningStatistics

(

)

Print out a human-readable format of the tuning statistics.

RegisterReflection()

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

inlinestatic

TerminateTask()

void tvm::meta_schedule::TaskSchedulerNode::TerminateTask

(

int

task_id

)

Terminate a task.

Parameters

task_id

The id of the task to be terminated

TouchTask()

void tvm::meta_schedule::TaskSchedulerNode::TouchTask

(

int

task_id

)

Touch the task and update its status.

Parameters

task_id

The task id to be checked.

Tune()

virtual void tvm::meta_schedule::TaskSchedulerNode::Tune ( ffi::Array< TuneContext >  tasks, ffi::Array< FloatImm >  task_weights, int  max_trials_global, int  max_trials_per_task, int  num_trials_per_iter, Builder  builder, Runner  runner, ffi::Array< MeasureCallback >  measure_callbacks, ffi::Optional< Database >  database, ffi::Optional< CostModel >  cost_model  )

virtual

Jointly tune a given list of tasks.

Parameters

tasks	The tasks to be tuned
task_weights	The weight of each task
max_trials_global	The maximum number of trials to be performed globally
max_trials_per_task	The maximum number of trials to be performed for each task
num_trials_per_iter	The number of trials to be performed in each iteration
builder	The MetaSchedule builder
runner	The MetaSchedule runner
measure_callbacks	The callbacks to be called after each measurement
database	The database used in tuning
cost_model	The cost model used in tuning

Reimplemented in tvm::meta_schedule::PyTaskSchedulerNode.

TVM_FFI_DECLARE_OBJECT_INFO()

tvm::meta_schedule::TaskSchedulerNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"meta_schedule.TaskScheduler"	,
		TaskSchedulerNode	,
		Object
	)

Member Data Documentation

_type_mutable

constexpr const bool tvm::meta_schedule::TaskSchedulerNode::_type_mutable = true

staticconstexpr

cost_model_

ffi::Optional<CostModel> tvm::meta_schedule::TaskSchedulerNode::cost_model_

The cost model used in tuning.

database_

ffi::Optional<Database> tvm::meta_schedule::TaskSchedulerNode::database_

The database used in tuning.

logger

ffi::Function tvm::meta_schedule::TaskSchedulerNode::logger

The tuning task's logging function.

measure_callbacks_

ffi::Array<MeasureCallback> tvm::meta_schedule::TaskSchedulerNode::measure_callbacks_

The list of measure callbacks of the scheduler.

remaining_tasks_

int tvm::meta_schedule::TaskSchedulerNode::remaining_tasks_

The number of remaining tasks to be tuned.

tasks_

ffi::Array<TaskRecord> tvm::meta_schedule::TaskSchedulerNode::tasks_

Records for each task.

---

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

• include/tvm/meta_schedule/task_scheduler.h