torch-mlir

Commit Graph

Author	SHA1	Message	Date
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
Yi Zhang	e09e2cbe70	Include IR dump options on e2e failure report	2022-02-09 11:19:34 -05:00
Sean Silva	19e9fc4ef1	Bring some more order to the e2e error reporting situation. - Move `run_pipeline_with_repro_report` to a more common place, and use it consistently - Attach a `torch.debug_module_name` to the enclosing `builtin.module` op to allow for self-contained error reporting (not needing to pass the names around. - Remove redundant error reporting in linalg_on_tensors_backend.py and tosa_backend.py (their respective backend abstract base classes now take care of the error reports themselves) - Save off original value of sys.stderr, rather than always resetting to `sys.__stderr__`. This is just more hygienic, and allows nesting if desired.	2021-10-08 13:00:12 -07:00

Author

SHA1

Message

Date

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

Yi Zhang

e09e2cbe70

Include IR dump options on e2e failure report

2022-02-09 11:19:34 -05:00

Sean Silva

19e9fc4ef1

Bring some more order to the e2e error reporting situation.

- Move `run_pipeline_with_repro_report` to a more common place, and use it
  consistently
- Attach a `torch.debug_module_name` to the enclosing `builtin.module`
  op to allow for self-contained error reporting (not needing to pass
  the names around.
- Remove redundant error reporting in linalg_on_tensors_backend.py and
  tosa_backend.py (their respective backend abstract base classes now
  take care of the error reports themselves)
- Save off original value of sys.stderr, rather than always resetting to
  `sys.__stderr__`. This is just more hygienic, and allows nesting if
  desired.

2021-10-08 13:00:12 -07:00

3 Commits (2d75654b2c2e7e5c4aba2f125229ab3cc3c5c4d8)