latest iree fix

nod-ai · Oct 10, 2024 · ee02170 · ee02170
1 parent 957a20a
commit ee02170
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/test_e2eshark.yml b/.github/workflows/test_e2eshark.yml
@@ -210,6 +210,7 @@ jobs:
           pip install -r ./alt_e2eshark/base_requirements.txt
           pip install -r ./alt_e2eshark/iree_requirements.txt
           pip install --no-deps -r ./alt_e2eshark/torch_mlir_requirements.txt
+          pip install --find-links https://iree.dev/pip-release-links.html iree-compiler iree-runtime --upgrade
         working-directory: ./test-suite
 
       - name: Run Onnx Bench Mode
@@ -299,6 +300,7 @@ jobs:
           pip install -r ./test-suite/alt_e2eshark/base_requirements.txt
           pip install -r ./test-suite/alt_e2eshark/iree_requirements.txt
           pip install --no-deps -r ./test-suite/alt_e2eshark/torch_mlir_requirements.txt
+          pip install --find-links https://iree.dev/pip-release-links.html iree-compiler iree-runtime --upgrade
       - uses: actions/download-artifact@master
         with:
           name: ci_reports_${{ matrix.backend }}_shark-test-suite_onnx_md