Skip to content

Commit 45ae8aa

Browse files
rusty1srusty1s
committed
cuda 110 builds
1 parent 87e30c5 commit 45ae8aa

File tree

7 files changed

+101
-78
lines changed

7 files changed

+101
-78
lines changed

‎.travis.yml‎

Lines changed: 42 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -10,38 +10,51 @@ env:
1010
- CUDA_HOME=/usr/local/cuda
1111
jobs:
1212
# Python 3.6
13+
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
14+
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
15+
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
16+
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
17+
- PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
1318
- PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cpu
1419
- PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu92
1520
- PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu101
1621
- PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu102
17-
- PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cpu
18-
- PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu92
19-
- PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu101
20-
- PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu102
2122
# Python 3.7
23+
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
24+
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
25+
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
26+
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
27+
- PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
2228
- PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cpu
2329
- PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu92
2430
- PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu101
2531
- PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu102
26-
- PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cpu
27-
- PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu92
28-
- PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu101
29-
- PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu102
3032
# Python 3.8
33+
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
34+
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
35+
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
36+
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
37+
- PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
3138
- PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
3239
- PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu92
3340
- PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
3441
- PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
35-
- PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cpu
36-
- PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu92
37-
- PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu101
38-
- PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu102
3942

4043
jobs:
41-
# For daily builds, we only test on *Python 3.6* and *CUDA 10.2*.
44+
# For daily builds, we only test on Python 3.6 with CUDA 10.2/11.0.
4245
# For deployment, we exclude *all* macOS CUDA and Windows CUDA 9.2 jobs.
4346
exclude:
4447
# Python 3.6
48+
- if: not (tag is present or commit_message =~ ci-deploy) and os != osx
49+
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cpu
50+
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
51+
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu92
52+
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
53+
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu101
54+
- if: os = osx
55+
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu102
56+
- if: os = osx
57+
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.7.0 IDX=cu110
4558
- if: not (tag is present or commit_message =~ ci-deploy) and os != osx
4659
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cpu
4760
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
@@ -50,15 +63,17 @@ jobs:
5063
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu101
5164
- if: os = osx
5265
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.6.0 IDX=cu102
53-
- if: not (tag is present or commit_message =~ ci-deploy) and os != osx
54-
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cpu
66+
# Python 3.7
67+
- if: not (tag is present or commit_message =~ ci-deploy)
68+
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cpu
5569
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
56-
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu92
70+
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu92
5771
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
58-
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu101
59-
- if: os = osx
60-
env: PYTHON_VERSION=3.6 TORCH_VERSION=1.5.0 IDX=cu102
61-
# Python 3.7
72+
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu101
73+
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
74+
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu102
75+
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
76+
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.7.0 IDX=cu110
6277
- if: not (tag is present or commit_message =~ ci-deploy)
6378
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cpu
6479
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
@@ -67,15 +82,17 @@ jobs:
6782
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu101
6883
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
6984
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.6.0 IDX=cu102
85+
# Python 3.8
7086
- if: not (tag is present or commit_message =~ ci-deploy)
71-
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cpu
87+
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cpu
7288
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
73-
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu92
89+
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu92
7490
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
75-
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu101
91+
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu101
7692
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
77-
env: PYTHON_VERSION=3.7 TORCH_VERSION=1.5.0 IDX=cu102
78-
# Python 3.8
93+
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu102
94+
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
95+
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.7.0 IDX=cu110
7996
- if: not (tag is present or commit_message =~ ci-deploy)
8097
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cpu
8198
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
@@ -84,14 +101,6 @@ jobs:
84101
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu101
85102
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
86103
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.6.0 IDX=cu102
87-
- if: not (tag is present or commit_message =~ ci-deploy)
88-
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cpu
89-
- if: not (tag is present or commit_message =~ ci-deploy) or os != linux
90-
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu92
91-
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
92-
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu101
93-
- if: not (tag is present or commit_message =~ ci-deploy) or os = osx
94-
env: PYTHON_VERSION=3.8 TORCH_VERSION=1.5.0 IDX=cu102
95104

96105
install:
97106
- source script/gcc.sh

‎CMakeLists.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
cmake_minimum_required(VERSION 3.0)
22
project(torchsparse)
33
set(CMAKE_CXX_STANDARD 14)
4-
set(TORCHSPARSE_VERSION 0.6.7)
4+
set(TORCHSPARSE_VERSION 0.6.8)
55

66
option(WITH_CUDA "Enable CUDA support" OFF)
77

‎README.md‎

Lines changed: 14 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -31,28 +31,29 @@ Note that only `value` comes with autograd support, as `index` is discrete and t
3131

3232
We provide pip wheels for all major OS/PyTorch/CUDA combinations, see [here](https://pytorch-geometric.com/whl).
3333

34-
#### PyTorch 1.6.0
34+
#### PyTorch 1.7.0
3535

36-
To install the binaries for PyTorch 1.6.0, simply run
36+
To install the binaries for PyTorch 1.7.0, simply run
3737

3838
```
39-
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
39+
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.7.0.html
4040
```
4141

42-
where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101` or `cu102` depending on your PyTorch installation.
42+
where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101`, `cu102`, or `cu110` depending on your PyTorch installation.
4343

44-
| | `cpu` | `cu92` | `cu101` | `cu102` |
45-
|-------------|-------|--------|---------|---------|
46-
| **Linux** | ✅ | ✅ | ✅ | ✅ |
47-
| **Windows** | ✅ | ❌ | ✅ | ✅ |
48-
| **macOS** | ✅ | | | |
44+
| | `cpu` | `cu92` | `cu101` | `cu102` | `cu110` |
45+
|-------------|-------|--------|---------|---------|---------|
46+
| **Linux** | ✅ | ✅ | ✅ | ✅ | ✅ |
47+
| **Windows** | ✅ | ❌ | ✅ | ✅ | ✅ |
48+
| **macOS** | ✅ | | | | |
4949

50-
#### PyTorch 1.5.0
5150

52-
To install the binaries for PyTorch 1.5.0, simply run
51+
#### PyTorch 1.6.0
52+
53+
To install the binaries for PyTorch 1.6.0, simply run
5354

5455
```
55-
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
56+
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
5657
```
5758

5859
where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101` or `cu102` depending on your PyTorch installation.
@@ -63,21 +64,7 @@ where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101` or `cu102` d
6364
| **Windows** | ✅ | ❌ | ✅ | ✅ |
6465
| **macOS** | ✅ | | | |
6566

66-
#### PyTorch 1.4.0
67-
68-
To install the binaries for PyTorch 1.4.0, simply run
69-
70-
```
71-
pip install torch-scatter==latest+${CUDA} torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
72-
```
73-
74-
where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu100` or `cu101` depending on your PyTorch installation.
75-
76-
| | `cpu` | `cu92` | `cu100` | `cu101` |
77-
|-------------|-------|--------|---------|---------|
78-
| **Linux** | ✅ | ✅ | ✅ | ✅ |
79-
| **Windows** | ✅ | ❌ | ❌ | ✅ |
80-
| **macOS** | ✅ | | | |
67+
**Note:** Binaries of older versions are also provided for PyTorch 1.4.0 and PyTorch 1.5.0 (following the same procedure).
8168

8269
### From source
8370

‎script/cuda.sh‎

Lines changed: 41 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,14 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu92" ]; then
88
export CUDA_SHORT=9.2
99
export CUDA=9.2.148-1
1010
export UBUNTU_VERSION=ubuntu1604
11-
export CUBLAS=cuda-cublas-dev-9-2
1211
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
1312
fi
1413

1514
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu101" ]; then
1615
export IDX=cu101
1716
export CUDA_SHORT=10.1
18-
export CUDA=10.1.105-1
17+
export CUDA=10.1.243-1
1918
export UBUNTU_VERSION=ubuntu1804
20-
export CUBLAS=libcublas-dev
2119
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
2220
fi
2321

@@ -26,7 +24,11 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu102" ]; then
2624
export CUDA_SHORT=10.2
2725
export CUDA=10.2.89-1
2826
export UBUNTU_VERSION=ubuntu1804
29-
export CUBLAS=libcublas-dev
27+
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
28+
fi
29+
30+
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "$IDX" = "cu110" ]; then
31+
export CUDA_SHORT=11.0
3032
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
3133
fi
3234

@@ -55,6 +57,13 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu102" ]; then
5557
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
5658
fi
5759

60+
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "$IDX" = "cu110" ]; then
61+
export CUDA_SHORT=11.0
62+
export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.2/local_installers
63+
export CUDA_FILE=cuda_${CUDA_SHORT}.2_451.48_win10.exe
64+
export TOOLKIT="cudatoolkit=${CUDA_SHORT}"
65+
fi
66+
5867
if [ "${TRAVIS_OS_NAME}" = "osx" ] && [ "$IDX" = "cpu" ]; then
5968
export TOOLKIT=""
6069
fi
@@ -65,14 +74,37 @@ else
6574
export FORCE_CUDA=1
6675
fi
6776

68-
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then
69-
INSTALLER=cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb
77+
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ] && [ "${IDX}" != "cu110" ]; then
78+
INSTALLER="cuda-repo-${UBUNTU_VERSION}_${CUDA}_amd64.deb"
7079
wget -nv "http://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/${INSTALLER}"
7180
sudo dpkg -i "${INSTALLER}"
7281
wget -nv "https://developer.download.nvidia.com/compute/cuda/repos/${UBUNTU_VERSION}/x86_64/7fa2af80.pub"
7382
sudo apt-key add 7fa2af80.pub
7483
sudo apt update -qq
75-
sudo apt install -y "cuda-core-${CUDA_SHORT/./-}" "cuda-cudart-dev-${CUDA_SHORT/./-}" "${CUBLAS}" "cuda-cusparse-dev-${CUDA_SHORT/./-}"
84+
sudo apt install "cuda-core-${CUDA_SHORT/./-}" "cuda-nvcc-${CUDA_SHORT/./-}" "cuda-libraries-dev-${CUDA_SHORT/./-}"
85+
sudo apt clean
86+
CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
87+
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
88+
PATH=${CUDA_HOME}/bin:${PATH}
89+
nvcc --version
90+
91+
# Fix cublas on CUDA 10.1:
92+
if [ -d "/usr/local/cuda-10.2/targets/x86_64-linux/include" ]; then
93+
sudo cp -r /usr/local/cuda-10.2/targets/x86_64-linux/include/* "${CUDA_HOME}/include/"
94+
fi
95+
if [ -d "/usr/local/cuda-10.2/targets/x86_64-linux/lib" ]; then
96+
sudo cp -r /usr/local/cuda-10.2/targets/x86_64-linux/lib/* "${CUDA_HOME}/lib/"
97+
fi
98+
fi
99+
100+
if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" = "cu110" ]; then
101+
wget -nv https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
102+
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
103+
wget -nv https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
104+
sudo dpkg -i cuda-repo-ubuntu1804-11-0-local_11.0.3-450.51.06-1_amd64.deb
105+
sudo apt-key add /var/cuda-repo-ubuntu1804-11-0-local/7fa2af80.pub
106+
sudo apt update -qq
107+
sudo apt install cuda-nvcc-11-0 cuda-libraries-dev-11-0
76108
sudo apt clean
77109
CUDA_HOME=/usr/local/cuda-${CUDA_SHORT}
78110
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
@@ -86,16 +118,11 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
86118
curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download" --output "/tmp/gpu_driver_dlls.zip"
87119
7z x "/tmp/gpu_driver_dlls.zip" -o"/c/Windows/System32"
88120

89-
# Install CUDA
121+
# Install CUDA:
90122
wget -nv "${CUDA_URL}/${CUDA_FILE}"
91-
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT}\" -Wait -NoNewWindow"
123+
PowerShell -Command "Start-Process -FilePath \"${CUDA_FILE}\" -ArgumentList \"-s nvcc_${CUDA_SHORT} cuobjdump_${CUDA_SHORT} nvprune_${CUDA_SHORT} cupti_${CUDA_SHORT} cublas_dev_${CUDA_SHORT} cudart_${CUDA_SHORT} cufft_dev_${CUDA_SHORT} curand_dev_${CUDA_SHORT} cusolver_dev_${CUDA_SHORT} cusparse_dev_${CUDA_SHORT} npp_dev_${CUDA_SHORT} nvrtc_dev_${CUDA_SHORT} nvml_dev_${CUDA_SHORT}\" -Wait -NoNewWindow"
92124
CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}
93125
PATH=${CUDA_HOME}/bin:$PATH
94126
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
95127
nvcc --version
96128
fi
97-
98-
# Fix Cuda9.2 on Windows: https://github.com/pytorch/pytorch/issues/6109
99-
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" = "cu92" ]; then
100-
sed -i.bak -e '129,141d' "${CUDA_HOME}/include/crt/host_config.h"
101-
fi

‎script/rename_wheel.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import shutil
66

77
idx = sys.argv[1]
8-
assert idx in ['cpu', 'cu92', 'cu101', 'cu102']
8+
assert idx in ['cpu', 'cu92', 'cu101', 'cu102', 'cu110']
99

1010
dist_dir = osp.join(osp.dirname(osp.abspath(__file__)), '..', 'dist')
1111
wheels = glob.glob(osp.join('dist', '**', '*.whl'), recursive=True)

‎setup.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def get_extensions():
8080

8181
setup(
8282
name='torch_sparse',
83-
version='0.6.7',
83+
version='0.6.8',
8484
author='Matthias Fey',
8585
author_email='[email protected]',
8686
url='https://github.com/rusty1s/pytorch_sparse',

‎torch_sparse/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import torch
55

6-
__version__ = '0.6.7'
6+
__version__ = '0.6.8'
77

88
for library in [
99
'_version', '_convert', '_diag', '_spmm', '_spspmm', '_metis', '_rw',

0 commit comments

Comments
 (0)