Job

AutoTVM Job definition module.

class lorien.dialect.tvm_dial.autotvm_dial.job.AutoTVMJob(workload: lorien.workload.Workload)

A tuning job including a workload as well as tuning related configurations.

static create_job_configs(configs: argparse.Namespace) → lorien.tune.job.JobConfigs

Create a JobConfigs. See JobConfigs.

Parameters

configs (argparse.Namespace) -- The system configuration of tuner.

Returns

job_configs -- The job configurations.

Return type

JobConfigs

tune(tune_options: Dict[str, Any], measure_options: Dict[str, Any], commit_options: Optional[Dict[str, Any]] = None)

Tune the job with the given configuration and update the result. If the commit options are provided, then this function also in charge of committing the tuning results, or the job manager will commit the result, otherwise.

class lorien.dialect.tvm_dial.autotvm_dial.job.AutoTVMJobConfigs(configs: argparse.Namespace)

AutoTVM job configurations.

localize(target: str, **kwargs)

Localize options on worker.

Parameters

  • target (str) -- The target string.

  • **kwargs --

    The kwargs of AutoTVM job configuration for updating.

    configs:

    System configuration that may include RPC information (argparse.Namespace).

lorien.dialect.tvm_dial.autotvm_dial.job.callback_metadata(metadata: lorien.dialect.tvm_dial.job.TuneMetadata) → Callable

An AutoTVM callback function to update the tuning metadata.

Parameters

metadata (TuneMetadata) -- The statistic information for the tuning process.

Returns

callback -- The callable function that maintains the metadata.

Return type

Callable

lorien.dialect.tvm_dial.autotvm_dial.job.create_autotvm_tuner(tuner_name: str, task: tvm.autotvm.task.task.Task) → tvm.autotvm.tuner.tuner.Tuner

Create an AutoTVM tuner by its name.

Parameters

  • tuner_name (str) -- The AutoTVM tuner name.

  • task (autotvm.task.Task) -- The AutoTVM task to be tuned.

Returns

autotvm_tuner -- The AutoTVM tuner.

Return type

autotvm.tuner.Tuner