iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.48k stars 553 forks source link

Reconcile PyTorch pass pipelines with those upstream in torch-mlir (if needed) #17021

Open ScottTodd opened 2 months ago

ScottTodd commented 2 months ago

We have some people running "e2e" test workflows in torch-mlir that end up differing when running through IREE. I suspect the pipelines may have drifted a bit.

linalg:

onnx:

ScottTodd commented 2 months ago

Related: https://github.com/nod-ai/SHARK-TestSuite/issues/47

AmosLewis commented 2 months ago

Related https://github.com/nod-ai/SHARK-TestSuite/issues/225