diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 2c3b594eea..ad4a1cf5fe 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -36,9 +36,9 @@ jobs: torch-spec: 'torch==2.4.0' gpu-arch-type: "cuda" gpu-arch-version: "12.1" - - name: CUDA Nightly + - name: CUDA Nightly (Aug 29, 2024) runs-on: linux.g5.12xlarge.nvidia.gpu - torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121' + torch-spec: '--pre torch==2.5.0.dev20240829+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.1" @@ -57,9 +57,9 @@ jobs: torch-spec: 'torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu' gpu-arch-type: "cpu" gpu-arch-version: "" - - name: CPU Nightly + - name: CPU Nightly (Aug 29, 2024) runs-on: linux.4xlarge - torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu' + torch-spec: '--pre torch==2.5.0.dev20240829+cpu --index-url https://download.pytorch.org/whl/nightly/cpu' gpu-arch-type: "cpu" gpu-arch-version: ""