File tree Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Expand file tree Collapse file tree 2 files changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -64,32 +64,38 @@ jobs:
64
64
torch-spec : ' torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121'
65
65
gpu-arch-type : " cuda"
66
66
gpu-arch-version : " 12.6"
67
+ dev-requirements-overrides : " s/^pytest$/pytest==7.4.0/"
67
68
- name : CUDA 2.6
68
69
runs-on : linux.g5.12xlarge.nvidia.gpu
69
70
torch-spec : ' torch==2.6.0'
70
71
gpu-arch-type : " cuda"
71
72
gpu-arch-version : " 12.6"
73
+ dev-requirements-overrides : " "
72
74
- name : CUDA 2.7
73
75
runs-on : linux.g5.12xlarge.nvidia.gpu
74
76
torch-spec : ' torch==2.7.0'
75
77
gpu-arch-type : " cuda"
76
78
gpu-arch-version : " 12.6"
79
+ dev-requirements-overrides : " "
77
80
78
81
- name : CPU 2.5.1
79
82
runs-on : linux.4xlarge
80
83
torch-spec : ' torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu'
81
84
gpu-arch-type : " cpu"
82
85
gpu-arch-version : " "
86
+ dev-requirements-overrides : " s/^pytest$/pytest==7.4.0/"
83
87
- name : CPU 2.6
84
88
runs-on : linux.4xlarge
85
89
torch-spec : ' torch==2.6.0 --index-url https://download.pytorch.org/whl/cpu'
86
90
gpu-arch-type : " cpu"
87
91
gpu-arch-version : " "
92
+ dev-requirements-overrides : " "
88
93
- name : CPU 2.7
89
94
runs-on : linux.4xlarge
90
95
torch-spec : ' torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu'
91
96
gpu-arch-type : " cpu"
92
97
gpu-arch-version : " "
98
+ dev-requirements-overrides : " "
93
99
94
100
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
95
101
with :
@@ -106,6 +112,7 @@ jobs:
106
112
export PATH=/opt/rh/gcc-toolset-10/root/usr/bin/:$PATH
107
113
python -m pip install --upgrade pip
108
114
pip install ${{ matrix.torch-spec }}
115
+ sed -i '${{ matrix.dev-requirements-overrides }}' dev-requirements.txt
109
116
pip install -r dev-requirements.txt
110
117
pip install .
111
118
export CONDA=$(dirname $(dirname $(which conda)))
Original file line number Diff line number Diff line change 1
1
# Test utilities
2
- pytest==7.4.0
2
+ pytest
3
3
unittest-xml-reporting
4
4
parameterized
5
5
packaging
You can’t perform that action at this time.
0 commit comments