torch-mlir/python/torch_mlir/tools/import_onnx
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
..
__main__.py [onnx] Add testing using the `onnx` compilation using torch tests (#2795) 2024-02-15 10:17:13 -08:00