From 0f787f2b797f3fb1f5cb9b632255e99eda4dac5d Mon Sep 17 00:00:00 2001
From: Marius Brehler <marius.brehler@amd.com>
Date: Fri, 13 Dec 2024 16:39:43 +0000
Subject: [PATCH] Move `+cpu` from matrix to pip command

---
 .github/workflows/ci-sharktank.yml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/ci-sharktank.yml b/.github/workflows/ci-sharktank.yml
index 01d72ff2e..6edb356c8 100644
--- a/.github/workflows/ci-sharktank.yml
+++ b/.github/workflows/ci-sharktank.yml
@@ -27,16 +27,16 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.11", "3.12"]
-        torch-version: ["2.3.0+cpu", "2.4.1+cpu", "2.5.1+cpu"]
+        torch-version: ["2.3.0", "2.4.1", "2.5.1"]
         os: [ubuntu-24.04]
         include:
           - os: windows-2022
             python-version: "3.11"
-            torch-version: "2.3.0+cpu"
+            torch-version: "2.3.0"
         exclude:
           - python-version: "3.12"
              # `torch.compile` requires torch>=2.4.0 for Python 3.12+
-            torch-version: "2.3.0+cpu"
+            torch-version: "2.3.0"
       fail-fast: false
     runs-on: ${{matrix.os}}
     defaults:
@@ -66,7 +66,7 @@ jobs:
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first. Installing the PyTorch CPU
           # wheels saves multiple minutes and a lot of bandwidth on runner setup.
-          pip install --no-compile --pre --index-url https://download.pytorch.org/whl/test/cpu torch==${{matrix.torch-version}}
+          pip install --no-compile --pre --index-url https://download.pytorch.org/whl/test/cpu torch==${{matrix.torch-version}}+cpu
           pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/
 
           # Install nightly IREE packages.