update

pyg-team · Feb 2, 2024 · 68856c6 · 68856c6
1 parent 897de03
commit 68856c6
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 10 deletions.
diff --git a/.github/actions/setup/action.yml b/.github/actions/setup/action.yml
@@ -6,7 +6,7 @@ inputs:
     default: '3.8'
   torch-version:
     required: false
-    default: '2.1.0'
+    default: '2.2.0'
   cuda-version:
     required: false
     default: cpu

diff --git a/.github/workflows/aws/upload_nightly_index.py b/.github/workflows/aws/upload_nightly_index.py
@@ -27,6 +27,9 @@
         wheels_dict[torch_version.replace('1.13.0', '1.13.1')].append(wheel)
     if '2.0.0' in torch_version:
         wheels_dict[torch_version.replace('2.0.0', '2.0.1')].append(wheel)
+    if '2.1.0' in torch_version:
+        wheels_dict[torch_version.replace('2.1.0', '2.1.1')].append(wheel)
+        wheels_dict[torch_version.replace('2.1.0', '2.1.2')].append(wheel)
 
 index_html = html.format('\n'.join([
     href.format(f'{version}.html'.replace('+', '%2B'), version)

diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
@@ -10,11 +10,20 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-20.04, macos-11]  # windows-2019
-        python-version: ['3.8', '3.9', '3.10', '3.11']
-        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0]
+        os: [ubuntu-20.04, macos-11, windows-2019]
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
+        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0]
         cuda-version: ['cpu', 'cu113', 'cu116', 'cu117', 'cu118', 'cu121']
         exclude:
+          - os: windows-2019  # No windows support yet :(
+          - torch-version: 1.12.0
+            python-version: '3.12'
+          - torch-version: 1.13.0
+            python-version: '3.12'
+          - torch-version: 2.0.0
+            python-version: '3.12'
+          - torch-version: 2.1.0
+            python-version: '3.12'
           - torch-version: 1.12.0
             python-version: '3.11'
           - torch-version: 1.12.0
@@ -43,6 +52,12 @@ jobs:
             cuda-version: 'cu116'
           - torch-version: 2.1.0
             cuda-version: 'cu117'
+          - torch-version: 2.2.0
+            cuda-version: 'cu113'
+          - torch-version: 2.2.0
+            cuda-version: 'cu116'
+          - torch-version: 2.2.0
+            cuda-version: 'cu117'
           - os: macos-11
             cuda-version: 'cu113'
           - os: macos-11
@@ -53,6 +68,13 @@ jobs:
             cuda-version: 'cu118'
           - os: macos-11
             cuda-version: 'cu121'
+            # Fails with:
+            # * note: 'value' has been explicitly marked unavailable here
+            # * error: 'value' is unavailable: introduced in macOS 10.13
+          - os: macos-11
+            python-version: '3.11'
+          - os: macos-11
+            python-version: '3.12'
 
     steps:
       - name: Checkout repository

diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml
@@ -15,10 +15,18 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-20.04]
-        python-version: ['3.8', '3.9', '3.10', '3.11']
-        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0]
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
+        torch-version: [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.2.0]
         cuda-version: ['cpu', 'cu113', 'cu116', 'cu117', 'cu118', 'cu121']
         exclude:
+          - torch-version: 1.12.0
+            python-version: '3.12'
+          - torch-version: 1.13.0
+            python-version: '3.12'
+          - torch-version: 2.0.0
+            python-version: '3.12'
+          - torch-version: 2.1.0
+            python-version: '3.12'
           - torch-version: 1.12.0
             python-version: '3.11'
           - torch-version: 1.12.0
@@ -47,6 +55,12 @@ jobs:
             cuda-version: 'cu116'
           - torch-version: 2.1.0
             cuda-version: 'cu117'
+          - torch-version: 2.2.0
+            cuda-version: 'cu113'
+          - torch-version: 2.2.0
+            cuda-version: 'cu116'
+          - torch-version: 2.2.0
+            cuda-version: 'cu117'
 
     steps:
       - name: Checkout repository

diff --git a/README.md b/README.md
@@ -18,7 +18,7 @@
 
 ## Installation
 
-We provide pre-built Python wheels for all major OS/PyTorch/CUDA combinations from Python 3.8 till 3.11, see [here](https://data.pyg.org/whl).
+We provide pre-built Python wheels for all major OS/PyTorch/CUDA combinations from Python 3.8 till 3.12, see [here](https://data.pyg.org/whl).
 Note that currently, Windows wheels are not supported (we are working on fixing this as soon as possible).
 
 To install the wheels, simply run
@@ -29,11 +29,17 @@ pip install pyg-lib -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
 
 where
 
-* `${TORCH}` should be replaced by either `1.12.0`, `1.13.0`, `2.0.0` or `2.1.0`
-* `${CUDA}` should be replaced by either `cpu`, `cu102`, `cu113`, `cu116`, `cu117`, `cu118` or `cu121`
+* `${TORCH}` should be replaced by either `1.12.0`, `1.13.0`, `2.0.0`, `2.1.0`, or `2.2.0`
+* `${CUDA}` should be replaced by either `cpu`, `cu102`, `cu113`, `cu116`, `cu117`, `cu118`, or `cu121`
 
 The following combinations are supported:
 
+| PyTorch 2.2  | `cpu` | `cu102` | `cu113` | `cu116` | `cu117` | `cu118` | `cu121` |
+|--------------|-------|---------|---------|---------|---------|---------|---------|
+| **Linux**    | ✅    |         |         |         |         | ✅      | ✅      |
+| **Windows**  |       |         |         |         |         |         |         |
+| **macOS**    | ✅    |         |         |         |         |         |         |
+
 | PyTorch 2.1  | `cpu` | `cu102` | `cu113` | `cu116` | `cu117` | `cu118` | `cu121` |
 |--------------|-------|---------|---------|---------|---------|---------|---------|
 | **Linux**    | ✅    |         |         |         |         | ✅      | ✅      |
@@ -60,7 +66,7 @@ The following combinations are supported:
 
 ### Form nightly
 
-Nightly wheels are provided for Linux from Python 3.8 till 3.11:
+Nightly wheels are provided for Linux from Python 3.8 till 3.12:
 
 ```
 pip install pyg-lib -f https://data.pyg.org/whl/nightly/torch-${TORCH}+${CUDA}.html

diff --git a/setup.cfg b/setup.cfg
@@ -10,6 +10,7 @@ classifiers=
     Programming Language :: Python :: 3.9
     Programming Language :: Python :: 3.10
     Programming Language :: Python :: 3.11
+    Programming Language :: Python :: 3.12
     Programming Language :: Python :: 3 :: Only
 
 [aliases]