From 810af5dc1eb027ddddb7e5fe6c1e7101c192a4b5 Mon Sep 17 00:00:00 2001 From: archana-ramalingam Date: Thu, 28 Nov 2024 01:33:45 +0000 Subject: [PATCH] Update IREE to pinned versions --- .github/workflows/ci-llama-quick-tests.yaml | 5 ++--- .github/workflows/ci-shark-ai.yml | 5 ++--- .github/workflows/ci-sharktank.yml | 10 ++++------ .github/workflows/ci_eval_short.yaml | 2 +- 4 files changed, 9 insertions(+), 13 deletions(-) diff --git a/.github/workflows/ci-llama-quick-tests.yaml b/.github/workflows/ci-llama-quick-tests.yaml index 4ee8d24ed..81d5ac5e8 100644 --- a/.github/workflows/ci-llama-quick-tests.yaml +++ b/.github/workflows/ci-llama-quick-tests.yaml @@ -65,9 +65,8 @@ jobs: pip install --no-compile -r pytorch-cpu-requirements.txt pip install --no-compile -r requirements.txt -e sharktank/ shortfin/ - # Get latest stable IREE release - pip install --src deps \ - -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" + # Get latest working stable IREE release + pip install iree-turbine==3.0.0 - name: Run llama 8b f16 decomposed test run: pytest sharktank/tests/models/llama/benchmark_amdgpu_test.py -v -s --iree-hip-target=gfx942 --run-quick-llama-test diff --git a/.github/workflows/ci-shark-ai.yml b/.github/workflows/ci-shark-ai.yml index 16705712b..9c65a28de 100644 --- a/.github/workflows/ci-shark-ai.yml +++ b/.github/workflows/ci-shark-ai.yml @@ -60,9 +60,8 @@ jobs: pip install --no-compile -r pytorch-cpu-requirements.txt pip install --no-compile -r requirements.txt -e sharktank/ shortfin/ - # Get latest stable IREE release - pip install --src deps \ - -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" + # Get latest working stable IREE release + pip install iree-turbine==3.0.0 - name: Run LLM Integration Tests run: pytest -v app_tests/integration_tests/llm/shortfin --log-cli-level=INFO diff --git a/.github/workflows/ci-sharktank.yml b/.github/workflows/ci-sharktank.yml index eadefa65f..b7cf57753 100644 --- a/.github/workflows/ci-sharktank.yml +++ b/.github/workflows/ci-sharktank.yml @@ -100,9 +100,8 @@ jobs: pip install --no-compile -r pytorch-cpu-requirements.txt pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/ - # Get latest stable IREE release - pip install --src deps \ - -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" + # Get latest working stable IREE release + pip install iree-turbine==3.0.0 - name: Run sharktank tests if: ${{ !cancelled() }} @@ -151,9 +150,8 @@ jobs: pip install --no-compile -r pytorch-cpu-requirements.txt pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/ - # Get latest stable IREE release - pip install --src deps \ - -e "git+https://github.com/iree-org/iree-turbine.git#egg=iree-turbine" + # Get latest working stable IREE release + pip install iree-turbine==3.0.0 - name: Run tests run: | diff --git a/.github/workflows/ci_eval_short.yaml b/.github/workflows/ci_eval_short.yaml index 80a74cacb..de55777dc 100644 --- a/.github/workflows/ci_eval_short.yaml +++ b/.github/workflows/ci_eval_short.yaml @@ -62,7 +62,7 @@ jobs: pip install --no-compile -r pytorch-cpu-requirements.txt pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/ - # Get latest stable IREE release + # Get latest working stable IREE release pip install iree-turbine==3.0.0 - name: Run perplexity test with vmfb