Install PyTorch for ROCM instead of CPU-only

These workflows run on MI300 machines but install a CPU-only version of
PyTorch instead of installing the ROCm enabled one.

.github/workflows/ci-llama-large-tests.yaml

@@ -56,7 +56,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
           pip install -r requirements-iree-unpinned.txt
           pip install --no-compile \
             -r sharktank/requirements-tests.txt \

.github/workflows/ci-llama-quick-tests.yaml

@@ -56,7 +56,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
           pip install -r requirements-iree-pinned.txt
           pip install --no-compile \
             -r sharktank/requirements-tests.txt \

.github/workflows/ci-sglang-benchmark.yml

@@ -64,7 +64,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
 
           # Use newest possible releases to be able to track commits that may
           # cause errors or performance changes.

.github/workflows/ci-sglang-integration-tests.yml

@@ -53,7 +53,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
 
           # Use newest possible releases to be able to track commits that may
           # cause errors.

.github/workflows/ci_eval.yaml

@@ -54,7 +54,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
           pip install -r requirements-iree-unpinned.txt
           pip install --no-compile \
             -r sharktank/requirements-tests.txt \
@@ -110,7 +110,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
           pip install -r requirements-iree-unpinned.txt
           pip install --no-compile \
             -r sharktank/requirements-tests.txt \

.github/workflows/ci_eval_short.yaml

@@ -53,7 +53,7 @@ jobs:
 
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first.
-          pip install --no-compile -r pytorch-cpu-requirements.txt
+          pip install --no-compile -r pytorch-rocm-requirements.txt
           pip install -r requirements-iree-pinned.txt
           pip install --no-compile \
             -r sharktank/requirements-tests.txt \