diff --git a/.github/workflows/regression_test_rocm.yml b/.github/workflows/regression_test_rocm.yml index a9fdb68c13..c12f2a61da 100644 --- a/.github/workflows/regression_test_rocm.yml +++ b/.github/workflows/regression_test_rocm.yml @@ -22,10 +22,10 @@ jobs: include: - name: ROCM Nightly runs-on: linux.rocm.gpu.gfx942.2 - torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3' + torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm7.0' gpu-arch-type: "rocm" - gpu-arch-version: "6.3" - docker-image: pytorch/manylinux2_28-builder:rocm6.3 + gpu-arch-version: "7.0" + docker-image: pytorch/manylinux2_28-builder:rocm7.0 permissions: id-token: write