Skip to content

Commit 6e8dfb2

Browse files
committed
Drop support for pytorch 1.12 and 1.13
1 parent 03a38fb commit 6e8dfb2

File tree

2 files changed

+5
-28
lines changed

2 files changed

+5
-28
lines changed

.github/workflows/publish.yaml

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ jobs:
4343
# Using ubuntu-20.04 instead of 22.04 for more compatibility (glibc). Ideally we'd use the
4444
# manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
4545
os: [ubuntu-20.04]
46-
python-version: ['3.7', '3.8', '3.9', '3.10', '3.11', '3.12']
47-
torch-version: ['1.12.1', '1.13.1', '2.0.1', '2.1.2', '2.2.2', '2.3.0', '2.4.0.dev20240420']
46+
python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
47+
torch-version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.0.dev20240505']
4848
cuda-version: ['11.8.0', '12.2.2']
4949
# We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
5050
# Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.
@@ -53,33 +53,11 @@ jobs:
5353
cxx11_abi: ['FALSE', 'TRUE']
5454
exclude:
5555
# Pytorch < 2.2 does not support Python 3.12
56-
- torch-version: '1.12.1'
57-
python-version: '3.12'
58-
- torch-version: '1.13.1'
59-
python-version: '3.12'
6056
- torch-version: '2.0.1'
6157
python-version: '3.12'
6258
- torch-version: '2.1.2'
6359
python-version: '3.12'
64-
# Pytorch <= 1.12 does not support Python 3.11
65-
- torch-version: '1.12.1'
66-
python-version: '3.11'
67-
# Pytorch >= 2.0 only supports Python >= 3.8
68-
- torch-version: '2.0.1'
69-
python-version: '3.7'
70-
- torch-version: '2.1.2'
71-
python-version: '3.7'
72-
- torch-version: '2.2.2'
73-
python-version: '3.7'
74-
- torch-version: '2.3.0'
75-
python-version: '3.7'
76-
- torch-version: '2.4.0.dev20240420'
77-
python-version: '3.7'
7860
# Pytorch <= 2.0 only supports CUDA <= 11.8
79-
- torch-version: '1.12.1'
80-
cuda-version: '12.2.2'
81-
- torch-version: '1.13.1'
82-
cuda-version: '12.2.2'
8361
- torch-version: '2.0.1'
8462
cuda-version: '12.2.2'
8563

@@ -138,8 +116,8 @@ jobs:
138116
# e.g. we can have system CUDA version being 11.7 but if torch==1.12 then we need to download the wheel from cu116
139117
# This code is ugly, maybe there's a better way to do this.
140118
export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
141-
minv = {'1.12': 113, '1.13': 116, '2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118}[env['MATRIX_TORCH_VERSION']]; \
142-
maxv = {'1.12': 116, '1.13': 117, '2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121}[env['MATRIX_TORCH_VERSION']]; \
119+
minv = {'2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118}[env['MATRIX_TORCH_VERSION']]; \
120+
maxv = {'2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121}[env['MATRIX_TORCH_VERSION']]; \
143121
print(max(min(int(env['MATRIX_CUDA_VERSION']), maxv), minv))" \
144122
)
145123
if [[ ${{ matrix.torch-version }} == *"dev"* ]]; then

setup.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,6 @@ def append_nvcc_threads(nvcc_extra_args):
207207
f"--offload-arch={os.getenv('HIP_ARCHITECTURES', 'native')}",
208208
"-U__CUDA_NO_HALF_OPERATORS__",
209209
"-U__CUDA_NO_HALF_CONVERSIONS__",
210-
"-DCK_FMHA_FWD_FAST_EXP2=1",
211210
"-fgpu-flush-denormals-to-zero",
212211
]
213212
+ cc_flag,
@@ -367,7 +366,7 @@ def run(self):
367366
else {
368367
"bdist_wheel": CachedWheelsCommand,
369368
},
370-
python_requires=">=3.7",
369+
python_requires=">=3.8",
371370
install_requires=[
372371
"torch",
373372
"packaging",

0 commit comments

Comments
 (0)