diff --git a/.github/workflows/test_e2eshark.yml b/.github/workflows/test_e2eshark.yml
index ef3e0de2..fea6b433 100644
--- 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