torch-mlir

Commit Graph

Author	SHA1	Message	Date
Ramiro Leal-Cavazos	22c0893ec6	Update debug options in compilation errors (#913 ) The flag for printing the IR after each pass is now prefixed with "mlir". This commit updates the flag in the error reporting for the compiler.	2022-06-07 10:55:54 -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

Author

SHA1

Message

Date

Ramiro Leal-Cavazos

22c0893ec6

Update debug options in compilation errors (#913 )

The flag for printing the IR after each pass is now prefixed with
"mlir". This commit updates the flag in the error reporting for the
compiler.

2022-06-07 10:55:54 -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

2 Commits (5fbf2a376c3d36f1f31a2cda8a242bd36f090383)