From f3c60bfaf709591b495bcb53b37e03767b9c2ba8 Mon Sep 17 00:00:00 2001
From: rusty1s <matthias.fey@tu-dortmund.de>
Date: Fri, 30 May 2025 21:16:32 +0200
Subject: [PATCH 1/2] PyTorch 2.7 support

---
 .github/workflows/building.yml | 19 +++++--------------
 1 file changed, 5 insertions(+), 14 deletions(-)

diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
index 492f8a1..75d597e 100644
--- a/.github/workflows/building.yml
+++ b/.github/workflows/building.yml
@@ -46,6 +46,7 @@ jobs:
       - name: Upgrade pip
         run: |
           pip install --upgrade setuptools
+          pip install wheel
 
       - name: Install scipy
         if: ${{ (matrix.python-version == '3.8') || (matrix.python-version == '3.9') }}
@@ -79,30 +80,20 @@ jobs:
         shell:
           bash
 
-      - name: Install main package for CPU
+      - name: Build wheel on CPU
         if: ${{ matrix.cuda-version == 'cpu' }}
         run: |
-          FORCE_ONLY_CPU=1 pip install -v -e . --no-build-isolation
+          FORCE_ONLY_CPU=1 python setup.py bdist_wheel --dist-dir=dist
         shell:
           bash
 
-      - name: Install main package for GPU
+      - name: Build wheel on GPU
         if: ${{ matrix.cuda-version != 'cpu' }}
         run: |
-          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
-          FORCE_CUDA=1 pip install -v -e . --no-build-isolation
+          FORCE_CUDA=1 python setup.py bdist_wheel --dist-dir=dist
         shell:
           bash
 
-      - name: Test installation
-        run: |
-          python -c "import torch_cluster; print('torch-cluster:', torch_cluster.__version__)"
-
-      - name: Build wheel
-        run: |
-          pip install wheel
-          python setup.py bdist_wheel --dist-dir=dist
-
       - name: Configure AWS
         uses: aws-actions/configure-aws-credentials@v1
         with:

From 125aa90ef45671006cb5141300cc6944a52dc369 Mon Sep 17 00:00:00 2001
From: rusty1s <matthias.fey@tu-dortmund.de>
Date: Fri, 30 May 2025 21:19:14 +0200
Subject: [PATCH 2/2] PyTorch 2.7 support

---
 .github/workflows/building.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/building.yml b/.github/workflows/building.yml
index 75d597e..1f859e9 100644
--- a/.github/workflows/building.yml
+++ b/.github/workflows/building.yml
@@ -80,16 +80,17 @@ jobs:
         shell:
           bash
 
-      - name: Build wheel on CPU
+      - name: Build wheel for CPU
         if: ${{ matrix.cuda-version == 'cpu' }}
         run: |
           FORCE_ONLY_CPU=1 python setup.py bdist_wheel --dist-dir=dist
         shell:
           bash
 
-      - name: Build wheel on GPU
+      - name: Build wheel for GPU
         if: ${{ matrix.cuda-version != 'cpu' }}
         run: |
+          source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
           FORCE_CUDA=1 python setup.py bdist_wheel --dist-dir=dist
         shell:
           bash