Extract From Record

The module of AutoTVM workload extraction from tuning records. The graph-level optimization workloads such as data layout transform fit in this scenario.

lorien.dialect.tvm_dial.autotvm_dial.extract_from_record.create_layout_transform_task(record: Tuple[tvm.autotvm.measure.measure.MeasureInput, tvm.autotvm.measure.measure.MeasureResult]) → Optional[tvm.autotvm.task.task.Task]

Create an AutoTVM task of layout transform.

Parameters

record (Tuple[MeasureInput, MeasureResult]) -- The AutoTVM record pair to create the layout transform task.

Returns

task -- The created layout_transform task, or None if the record has no layout transform support.

Return type

Optional[Task]

lorien.dialect.tvm_dial.autotvm_dial.extract_from_record.extract_from_records(configs: argparse.Namespace)

Extract graph optimization workloads from a given DB table.

Parameters

configs (argparse.Namespace) -- The system configure of generate.extract-from-record.

Returns

workloads -- A list of collected workloads.

Return type

List[AutoTVMWorkload]