Skip to content

Commit 3426378

Browse files
Remove ROCm docker image default in linux_job_v2.yml (#6936)
Use `inputs.docker-image` to explicitly provide the docker image to be used. Relates to: pytorch/ao#2562
1 parent a9ec424 commit 3426378

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

.github/workflows/linux_job_v2.yml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,7 @@ jobs:
116116
name: ${{ inputs.job-name }}
117117
env:
118118
DOCKER_IMAGE: >-
119-
${{ inputs.gpu-arch-type == 'rocm' && format('pytorch/manylinux2_28-builder:{0}{1}',
120-
inputs.gpu-arch-type,
121-
inputs.gpu-arch-version)
122-
|| inputs.docker-image == 'pytorch/almalinux-builder' && format('pytorch/almalinux-builder:{0}{1}',
119+
${{ inputs.docker-image == 'pytorch/almalinux-builder' && format('pytorch/almalinux-builder:{0}{1}',
123120
inputs.gpu-arch-type,
124121
inputs.gpu-arch-version)
125122
|| inputs.docker-image }}

0 commit comments

Comments
 (0)