Skip to content

Commit c63899b

Browse files
authored
Fix rocm CI (#3136)
1 parent 3679631 commit c63899b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/regression_test_rocm.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ jobs:
2222
include:
2323
- name: ROCM Nightly
2424
runs-on: linux.rocm.gpu.gfx942.2
25-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3'
25+
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.0'
2626
gpu-arch-type: "rocm"
27-
gpu-arch-version: "6.3"
28-
docker-image: pytorch/manylinux2_28-builder:rocm6.3
27+
gpu-arch-version: "7.0"
28+
docker-image: pytorch/manylinux2_28-builder:rocm7.0
2929

3030
permissions:
3131
id-token: write

0 commit comments

Comments
 (0)