Skip to content

Commit 235b378

Browse files
authored
Pin CUDA nightlies for mx failures (#428)
* Pin CUDA nightlies for mx failures * Update regression_test.yml * Update regression_test.yml
1 parent fd9f95d commit 235b378

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/workflows/regression_test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ jobs:
3333
gpu-arch-version: "12.1"
3434
- name: CUDA Nightly
3535
runs-on: linux.g5.12xlarge.nvidia.gpu
36-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
36+
torch-spec: '--pre torch==2.5.0.dev20240620+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121'
3737
gpu-arch-type: "cuda"
3838
gpu-arch-version: "12.1"
3939
- name: CPU 2.2.2

0 commit comments

Comments
 (0)