Skip to content

Commit 287458c

Browse files
authored
Pin PT version: Fix FPX Inductor error (#790)
* Pin PT version: Fix FPX Inductor error * Update regression_test.yml
1 parent 65f660d commit 287458c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/workflows/regression_test.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@ jobs:
3636
torch-spec: 'torch==2.4.0'
3737
gpu-arch-type: "cuda"
3838
gpu-arch-version: "12.1"
39-
- name: CUDA Nightly
39+
- name: CUDA Nightly (Aug 29, 2024)
4040
runs-on: linux.g5.12xlarge.nvidia.gpu
41-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
41+
torch-spec: '--pre torch==2.5.0.dev20240829+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121'
4242
gpu-arch-type: "cuda"
4343
gpu-arch-version: "12.1"
4444

@@ -57,9 +57,9 @@ jobs:
5757
torch-spec: 'torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu'
5858
gpu-arch-type: "cpu"
5959
gpu-arch-version: ""
60-
- name: CPU Nightly
60+
- name: CPU Nightly (Aug 29, 2024)
6161
runs-on: linux.4xlarge
62-
torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
62+
torch-spec: '--pre torch==2.5.0.dev20240829+cpu --index-url https://download.pytorch.org/whl/nightly/cpu'
6363
gpu-arch-type: "cpu"
6464
gpu-arch-version: ""
6565

0 commit comments

Comments
 (0)