From dd72dfec89de44024cf0510cc54c21eb401bb587 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Tue, 18 Jun 2024 12:05:14 -0700
Subject: [PATCH 1/4] Update test_models.yml

---
 .github/workflows/test_models.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml
index 8af87230..21dd933e 100644
--- a/.github/workflows/test_models.yml
+++ b/.github/workflows/test_models.yml
@@ -49,6 +49,7 @@ jobs:
           pip install --no-compile --pre --upgrade -r requirements.txt
           pip install --no-compile --pre -e .[testing]
           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
+          pip install --no-cache-dir -r iree-requirements.txt -f https://iree.dev/pip-release-links.html
       - name: Show current free memory
         run: |
           free -mh

From 71b1f751af26201f98c51b2088a7c875ce2be1e9 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Tue, 18 Jun 2024 12:45:06 -0700
Subject: [PATCH 2/4] Update iree versioning

---
 .github/workflows/test_models.yml | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/test_models.yml b/.github/workflows/test_models.yml
index 21dd933e..3c61cec1 100644
--- a/.github/workflows/test_models.yml
+++ b/.github/workflows/test_models.yml
@@ -49,7 +49,11 @@ jobs:
           pip install --no-compile --pre --upgrade -r requirements.txt
           pip install --no-compile --pre -e .[testing]
           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
-          pip install --no-cache-dir -r iree-requirements.txt -f https://iree.dev/pip-release-links.html
+          pip install \
+            --find-links https://iree.dev/pip-release-links.html \
+            --upgrade \
+            iree-compiler \
+            iree-runtime
       - name: Show current free memory
         run: |
           free -mh
@@ -86,6 +90,11 @@ jobs:
           pip install --no-compile --pre --upgrade -r requirements.txt
           pip install --no-compile --pre -e .[testing]
           pip install --no-compile --pre --upgrade -e ${{ github.workspace }}/SHARK-Turbine/models -r ${{ github.workspace }}/SHARK-Turbine/models/requirements.txt
+          pip install \
+            --find-links https://iree.dev/pip-release-links.html \
+            --upgrade \
+            iree-compiler \
+            iree-runtime
       - name: Show current free memory
         run: |
           free -mh

From 477393e54f6070193488d6c90be07cd76b916a83 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Tue, 18 Jun 2024 12:47:23 -0700
Subject: [PATCH 3/4] Resolve SHARK test

---
 .github/workflows/test_shark.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
index b50bb359..7291345a 100644
--- a/.github/workflows/test_shark.yml
+++ b/.github/workflows/test_shark.yml
@@ -36,9 +36,9 @@ jobs:
       - name: "Checkout SHARK"
         uses: actions/checkout@v4
         with:
-          repository: "nod-ai/SHARK.git"
+          repository: "nod-ai/SHARK"
           path: SHARK
-          ref: "iree-turbine-switch"
+          ref: "diffusers-version"
 
       - name: "Install SHARK"
         run: |

From 5b56bcf8b91a899ae401aced16a2f490173a04d9 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Tue, 18 Jun 2024 13:42:14 -0700
Subject: [PATCH 4/4] Update test_shark.yml back to main

---
 .github/workflows/test_shark.yml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
index 7291345a..d38d852e 100644
--- a/.github/workflows/test_shark.yml
+++ b/.github/workflows/test_shark.yml
@@ -38,7 +38,6 @@ jobs:
         with:
           repository: "nod-ai/SHARK"
           path: SHARK
-          ref: "diffusers-version"
 
       - name: "Install SHARK"
         run: |