nod-ai / SHARK-Turbine

Unified compiler/runtime for interfacing with PyTorch Dynamo.
Apache License 2.0
90 stars 44 forks source link

[tracking] HF CNN fp32 model tests #801

Open jinchen62 opened 1 month ago

jinchen62 commented 1 month ago

Below is the list of HF CNN fp32 model issues for Full model list. Tests are imported in https://github.com/nod-ai/SHARK-TestSuite/pull/313.

Status on 9/12: items tests model-run onnx-import torch-mlir iree-compile inference
total-count 529 499 489 418 310 154

Steps: Build torch-mlir and iree locally

cd SHARK-TestSuite/e2eshark/
python ./run.py  --tolerance 0.001 0.001 -c path/to/torch-mlir/build -i path/to/iree-build --cachedir path/to/cache -f onnx -g models --mode onnx --report -j 12 --torchtolinalg --testsfile model_list.txt

model-run

onnx-import

torch-mlir

iree-compile

inference

AmosLewis commented 1 month ago

onnx.Conv in #566 has been fixed by disable quantized 1d conv. This should be a different conv error.

AmosLewis commented 1 month ago

Better to add the failed op into https://github.com/nod-ai/SHARK-Turbine/issues/797

AmosLewis commented 1 month ago

onnx.Batchnorm in https://github.com/nod-ai/SHARK-Turbine/issues/566 also has been fixed by bump iree. This should be a different error. Maybe you should check vivek's patch https://github.com/llvm/torch-mlir/pull/3597

vivekkhandelwal1 commented 1 month ago

Hi @jinchen62, can you please update this issue, since it seems some of these have been fixed.

jinchen62 commented 1 month ago

@vivekkhandelwal1 Nothing was changed by Monday, will verify and update weekly.