torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sean Silva	93f1c3138b	torch_mlir.compile: Allow OutputType as a string. A lot of code was super verbose with `torch_mlir.OutputType.XYZ`. Now, you can simply do `"xyz"`. I updated a few examples.	2022-07-08 17:37:27 -07:00
Sean Silva	075464fa74	Add a new `torch_mlir.compile` method. This makes it much easier to convert models and hides all the ClassAnnotator complexity. This also adds a new example `torchscript_resnet18_all_output_types.py` which shows the ResNet18 IR for all output types. Also, - This moves `run_pipeline_with_repro_report` to `torch_mlir.compiler_utils`.	2022-04-20 10:06:01 -07:00
Sean Silva	1960ba76fb	Remove "e2e" name from `examples/torchscript_resnet18_e2e.py` That was back from an earlier stage in the project when e2e was a big deal because we didn't have anything working e2e yet :)	2022-03-28 18:26:54 +00:00

Author

SHA1

Message

Date

Sean Silva

93f1c3138b

torch_mlir.compile: Allow OutputType as a string.

A lot of code was super verbose with `torch_mlir.OutputType.XYZ`. Now,
you can simply do `"xyz"`. I updated a few examples.

2022-07-08 17:37:27 -07:00

Sean Silva

075464fa74

Add a new `torch_mlir.compile` method.

This makes it much easier to convert models and hides all the
ClassAnnotator complexity.

This also adds a new example `torchscript_resnet18_all_output_types.py`
which shows the ResNet18 IR for all output types.

Also,

- This moves `run_pipeline_with_repro_report` to
  `torch_mlir.compiler_utils`.

2022-04-20 10:06:01 -07:00

Sean Silva

1960ba76fb

Remove "e2e" name from `examples/torchscript_resnet18_e2e.py`

That was back from an earlier stage in the project when e2e was a big
deal because we didn't have anything working e2e yet :)

2022-03-28 18:26:54 +00:00

3 Commits (1dfe5efe9ee8f58deabf9382b58894a00b2d440c)