torch-mlir/test/python/onnx_importer
Rob Suderman 074f112d6a
[onnx] Add testing using the `onnx` compilation using torch tests (#2795)
We can route the torch tests via `onnx` using the `torch.onnx.export`
tooling. We can then reimport, lower to torch, and compile to linalg to
validate the onnx path is working correctly.

The current implementation exposes some failures in the `onnx` path so
we cannot enable the onnx test suite yet due to segmentation faults.
2024-02-15 10:17:13 -08:00
..
.gitignore Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00
LeakyReLU.onnx [onnx] Add torch-mlir-import-onnx tool. (#2637) 2023-12-12 22:01:30 -08:00
_torch_mlir_config.py Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00
command_line_test.py Make the onnx importer more robust for internal/external and large models (#2794) 2024-01-31 21:58:43 -08:00
import_onnx_tool.runlit [onnx] Add torch-mlir-import-onnx tool. (#2637) 2023-12-12 22:01:30 -08:00
import_smoke_test.py [onnx] Add testing using the `onnx` compilation using torch tests (#2795) 2024-02-15 10:17:13 -08:00
lit.local.cfg Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00