From ff2a4b17b9ac9c11af69fb62051622c97e5fcdf2 Mon Sep 17 00:00:00 2001
From: Bachir Djermani <bachir.djermani@physicsx.ai>
Date: Wed, 19 Mar 2025 21:09:07 +0000
Subject: [PATCH] chore: support torch 2.6.0

---
 .github/workflows/building.yml |  6 +++++-
 .github/workflows/testing.yml  |  2 +-
 README.md                      | 16 ++++++++++++++++
 3 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
index a4bbb38d..613ab476 100644
--- a/.github/workflows/building.yml
+++ b/.github/workflows/building.yml
@@ -12,9 +12,13 @@ jobs:
       matrix:
         os: [ubuntu-20.04, macos-14, windows-2019]
         python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
-        torch-version: [2.5.0]  # [2.3.0, 2.4.0, 2.5.0]
+        torch-version: [2.6.0]  # [2.3.0, 2.4.0, 2.5.0, 2.6.0]
         cuda-version: ['cpu', 'cu118', 'cu121', 'cu124']
         exclude:
+          - torch-version: 2.6.0
+            python-version: '3.8'
+          - torch-version: 2.6.0
+            cuda-version: 'cu121'
           - torch-version: 2.5.0
             python-version: '3.8'
           - torch-version: 2.3.0
diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
index b99298f9..1e760e13 100644
--- a/.github/workflows/testing.yml
+++ b/.github/workflows/testing.yml
@@ -16,7 +16,7 @@ jobs:
       matrix:
         os: [ubuntu-latest, windows-latest]
         python-version: [3.9]
-        torch-version: [2.4.0, 2.5.0]
+        torch-version: [2.4.0, 2.5.0, 2.6.0]
 
     steps:
       - uses: actions/checkout@v2
diff --git a/README.md b/README.md
index 2039c83e..b9cab727 100644
--- a/README.md
+++ b/README.md
@@ -54,6 +54,22 @@ conda install pytorch-scatter -c pyg
 
 We alternatively provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://data.pyg.org/whl).
 
+#### PyTorch 2.6
+
+To install the binaries for PyTorch 2.6.0, simply run
+
+```
+pip install torch-scatter -f https://data.pyg.org/whl/torch-2.6.0+${CUDA}.html
+```
+
+where `${CUDA}` should be replaced by either `cpu`, `cu118` or `cu124` depending on your PyTorch installation.
+
+|             | `cpu` | `cu118` | `cu124` |
+|-------------|-------|---------|---------|
+| **Linux**   | ✅    | ✅      | ✅      |
+| **Windows** | ✅    | ✅      | ✅      |
+| **macOS**   | ✅    |         |         |
+
 #### PyTorch 2.5
 
 To install the binaries for PyTorch 2.5.0, simply run