torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sean Silva	ab5ad7af09	Add tracing suport to `torch_mlir.compile`. This also has a fix for the adjustment of types of TupleConstruct inputs, which I found when using this new functionality on a model. Some scenarios in tracing create situations where the output of TupleConstruct has a more refined type than the inputs. This introduces a helper `adjustStaticInformationForValues` which subsumes the `derefineValues` helper and the tensor static information adjustment we were doing.	2022-05-03 09:08:40 -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

Sean Silva

ab5ad7af09

Add tracing suport to `torch_mlir.compile`.

This also has a fix for the adjustment of types of TupleConstruct
inputs, which I found when using this new functionality on a model.

Some scenarios in tracing create situations where the output of
TupleConstruct has a more refined type than the inputs.

This introduces a helper `adjustStaticInformationForValues` which
subsumes the `derefineValues` helper and the tensor static information
adjustment we were doing.

2022-05-03 09:08:40 -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 (2b1b0f6e1970c9db13caea2515070c61d4dee167)