llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,298 Commits
47 Branches
1195 Tags
72 MiB
Commit Graph

3 Commits (85f383ce0b6855592c7d440f744d9f41d73aa21e)

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