Skip to content

Commit 14d844a

Browse files
jcaipjainapurva
authored andcommitted
pin nightlies to deal with std::badalloc (#1256)
* pin nightlies to deal with std::badalloc * Update regression_test.yml
1 parent 90544b9 commit 14d844a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/regression_test.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ jobs:
4040
gpu-arch-version: "12.1"
4141
- name: CUDA Nightly
4242
runs-on: linux.g5.12xlarge.nvidia.gpu
43-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
43+
torch-spec: '--pre torch==2.6.0.dev20241101 --index-url https://download.pytorch.org/whl/nightly/cu121'
4444
gpu-arch-type: "cuda"
4545
gpu-arch-version: "12.1"
4646

@@ -61,7 +61,7 @@ jobs:
6161
gpu-arch-version: ""
6262
- name: CPU Nightly
6363
runs-on: linux.4xlarge
64-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
64+
torch-spec: '--pre torch==2.6.0.dev20241101 --index-url https://download.pytorch.org/whl/nightly/cpu'
6565
gpu-arch-type: "cpu"
6666
gpu-arch-version: ""
6767

0 commit comments

Comments
 (0)