Skip to content

Commit 9750e13

Browse files
Explicitly specify docker-image to be used for ROCm yml (#2562)
1 parent 95d13d5 commit 9750e13

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/workflows/regression_test_rocm.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ jobs:
2525
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3'
2626
gpu-arch-type: "rocm"
2727
gpu-arch-version: "6.3"
28+
docker-image: pytorch/manylinux2_28-builder:rocm6.3
2829

2930
permissions:
3031
id-token: write
@@ -36,6 +37,7 @@ jobs:
3637
runner: ${{ matrix.runs-on }}
3738
gpu-arch-type: ${{ matrix.gpu-arch-type }}
3839
gpu-arch-version: ${{ matrix.gpu-arch-version }}
40+
docker-image: ${{ matrix.docker-image }}
3941
submodules: recursive
4042
script: |
4143
conda create -n venv python=3.9 -y

0 commit comments

Comments
 (0)