From c95c645f595ab79b4827968975932853f178eee1 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Mon, 15 Jul 2024 11:45:54 -0500
Subject: [PATCH 1/2] Switch to mi250 runner for nightly CI.

---
 .github/workflows/test_sdxl.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/test_sdxl.yml b/.github/workflows/test_sdxl.yml
index bf97cc392..b0e59b5df 100644
--- a/.github/workflows/test_sdxl.yml
+++ b/.github/workflows/test_sdxl.yml
@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         version: [3.11]
-        os: [nodai-amdgpu-w7900-x86-64]
+        os: [nodai-amdgpu-mi250-x86-64]
 
     runs-on: ${{matrix.os}}
     steps:

From 5bdb0258d693d0faefc768b7288d7c3149a9eb58 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Mon, 15 Jul 2024 11:53:18 -0500
Subject: [PATCH 2/2] Explicitly install torch cpu in SHARK CI

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

diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
index 6f2e4b4ed..45ada02d0 100644
--- a/.github/workflows/test_shark.yml
+++ b/.github/workflows/test_shark.yml
@@ -49,6 +49,7 @@ jobs:
           cd $GITHUB_WORKSPACE/SHARK
           python${{ matrix.version }} -m venv shark.venv
           source shark.venv/bin/activate
+          pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           pip install -r requirements.txt --no-cache-dir
           pip install -e .
           python apps/shark_studio/tests/api_test.py