diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index d6f77b1e1..ce9c57d1f 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -43,6 +43,11 @@ jobs: torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.1" + - name: ROCM Nightly + runs-on: linux.rocm.gpu + torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.2' + gpu-arch-type: "rocm" + gpu-arch-version: "6.2" - name: CPU 2.2.2 runs-on: linux.4xlarge @@ -64,7 +69,7 @@ jobs: torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu' gpu-arch-type: "cpu" gpu-arch-version: "" - + uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: timeout: 60