Skip to content

Commit 91feaa5

Browse files
authored
Fix CUDA wheels (part 2) (#403)
* PyTorch 2.7 support * PyTorch 2.7 support
1 parent 463eb83 commit 91feaa5

File tree

1 file changed

+5
-13
lines changed

1 file changed

+5
-13
lines changed

.github/workflows/building.yml

Lines changed: 5 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ jobs:
4949
- name: Upgrade pip
5050
run: |
5151
pip install --upgrade setuptools
52+
pip install wheel
5253
5354
- name: Install scipy
5455
if: ${{ (matrix.python-version == '3.8') || (matrix.python-version == '3.9') }}
@@ -96,30 +97,21 @@ jobs:
9697
run: |
9798
bash .github/workflows/metis-${{ runner.os }}.sh
9899
99-
- name: Install main package for CPU
100+
- name: Build wheel for CPU
100101
if: ${{ matrix.cuda-version == 'cpu' }}
101102
run: |
102-
FORCE_ONLY_CPU=1 WITH_METIS=1 pip install -v -e . --no-build-isolation
103+
FORCE_ONLY_CPU=1 WITH_METIS=1 python setup.py bdist_wheel --dist-dir=dist
103104
shell:
104105
bash
105106

106-
- name: Install main package for GPU
107+
- name: Build wheel for GPU
107108
if: ${{ matrix.cuda-version != 'cpu' }}
108109
run: |
109110
source .github/workflows/cuda/${{ matrix.cuda-version }}-${{ runner.os }}-env.sh
110-
FORCE_CUDA=1 WITH_METIS=1 pip install -v -e . --no-build-isolation
111+
FORCE_CUDA=1 WITH_METIS=1 python setup.py bdist_wheel --dist-dir=dist
111112
shell:
112113
bash
113114

114-
- name: Test installation
115-
run: |
116-
python -c "import torch_sparse; print('torch-sparse:', torch_sparse.__version__)"
117-
118-
- name: Build wheel
119-
run: |
120-
pip install wheel
121-
python setup.py bdist_wheel --dist-dir=dist
122-
123115
- name: Configure AWS
124116
uses: aws-actions/configure-aws-credentials@v1
125117
with:

0 commit comments

Comments
 (0)