Job

TVM Job definition module.

class lorien.dialect.tvm_dial.job.TVMJob(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

is_target_compatible(target: str) → bool

Check if the taret is compatible to this job.

Parameters

target (str) -- The target string

Returns

compatible -- Whether the target is compatible to this job.

Return type

bool

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.job.TVMJobConfigs(configs: argparse.Namespace)

AutoTVM job configurations.

check_tvm_build_config() → bool

Check if the TVM build config on this machine matches the expected one.

Returns

match -- Return True if matcheing.

Return type

bool

localize(target: str, **kwargs)

Localize options on worker.

Parameters

• target (str) -- The target string.

• **kwargs -- The kwargs of job configuration for updating.

class lorien.dialect.tvm_dial.job.TuneMetadata

Metadata for a tuning process.