Closed AmosLewis closed 2 weeks ago
Those versions are pinned to the same torch-mlir commit, so I'm not sure there can be something missing. But what there may be is an issue between running the full iree pipeline and the torch-mlir pipeline.
To triage that, you can run iree-compile --compile-to=input
to see how IREE pre-processes and see if/how that differs from torch-mlir-opt.
The usual suspect is the createTorchToIREEPipeline having drifted from the upstream version (https://github.com/llvm/torch-mlir/blob/main/lib/Dialect/TorchConversion/Transforms/Passes.cpp#L65).
I've been wanting to extract the torch-mlir code for that into something more modular that can be shared for a while... but time...
Just test with candidate-20240809.980, inconsistency for Inception_v4_vaiq_int8 disappear
What happened?
Inconsistency found when lowering Inception_v4_vaiq_int8 model https://github.com/nod-ai/SHARK-TestSuite/issues/190
Failed log:
Steps to reproduce your issue
git clone https://github.com/nod-ai/SHARK-TestSuite
cd SHARK-TestSuite/e2eshark/
Passed: standalone torch-mlir-opt + iree: onnx -> torch -> linalg -> vmfb
python ./run.py --torchmlirbuild ../../torch-mlir/build --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f onnx -g models --mode onnx --report --tests onnx/models/Inception_v4_vaiq_int8 --torchtolinalg
Failed: iree: onnx -> vmfb:
python ./run.py --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f onnx -g models --mode onnx --report --tests onnx/models/Inception_v4_vaiq_int8
You can find the Inception_v4_vaiq_int8.default.torch-onnx.mlir file by
cd SHARK-TestSuite/e2eshark/test-run/onnx/models/Inception_v4_vaiq_int8
What component(s) does this issue relate to?
Compiler
Version information
iree: candidate-20240704.944 torch-mlir : ca0e9066755b35c0889c6ab792265b0886325f50
Additional context
No response