Job

TVM auto_scheduler Job definition module.

class lorien.dialect.tvm_dial.auto_scheduler_dial.job.AutoSchedulerJob(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.auto_scheduler_dial.job.AutoSchedulerJobConfigs(configs: argparse.Namespace)

AutoScheduler 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).

class lorien.dialect.tvm_dial.auto_scheduler_dial.job.RecordToMetadata(metadata: lorien.dialect.tvm_dial.job.TuneMetadata)

A Lorien customized callback to update the tuning metadata.

callback(_, inputs, results)

A callback to log the so far best status.