Workload

auto_scheduler Workload Definition.

class lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload

The workload for an op. A workload can be used to created an AutoScheduler task for tuning.

classmethod from_task(task: tvm.auto_scheduler.search_task.SearchTask) → lorien.dialect.tvm_dial.auto_scheduler_dial.workload.AutoSchedulerWorkload

Create a workload from an AutoScheduler task.

Parameters

task (SearchTask) -- The AutoScheduler task for the workload.

Returns

workload -- The initialized workload.

Return type

Workload

get_workload_key() → str

Get the primary key of this workload in DB.

Returns

key -- The primary key of this workload to index the records in DB.

Return type

str

mutate(rules: Any) → Sequence[lorien.workload.Workload]

auto_scheduler task mutation is not supported yet.

Parameters

• workload (Workload) -- The workload to be mutated.

• rules (Any) -- The mutation rules that can be customized.

Returns

workloads -- The mutated workloads.

Return type

Sequence[Workload]

to_job() → lorien.tune.job.Job

Create a job to tune this workload.

Returns

job -- The created job.

Return type

Job

to_task() → tvm.auto_scheduler.search_task.SearchTask

Create an AutoScheduler task from this workload.

Returns

task -- Return the created task, or raise RuntimeError if failed.

Return type

SearchTask