new build dir

MANIFEST.in

@@ -2,7 +2,4 @@ include LICENSE
 include build.py
 include build.sh
 
-recursive-include torch_cluster/src *
-recursive-include torch_cluster/kernel *
-
 recursive-exclude torch_cluster/_ext *

build.py

@@ -24,7 +24,7 @@ if torch.cuda.is_available():
     sources += ['aten/THCC/THCC{}.c'.format(f) for f in files]
     include_dirs += ['aten/THCC']
     define_macros += [('WITH_CUDA', None)]
-    extra_objects += ['aten/build/THC.so']
+    extra_objects += ['torch_cluster/_ext/THC.so']
     with_cuda = True
 
 ffi = create_extension(

build.sh

@@ -4,7 +4,7 @@ echo "Compiling kernel..."
 
 if [ -z "$1" ]; then TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))"); else TORCH="$1"; fi
 SRC_DIR=aten/THC
-BUILD_DIR=aten/build
+BUILD_DIR=torch_cluster/_ext
 
 mkdir -p $BUILD_DIR
 $(which nvcc) -c -o "$BUILD_DIR/THC.so" "$SRC_DIR/THC.cu" -arch=sm_52 -Xcompiler -fPIC -shared "-I$TORCH/lib/include/TH" "-I$TORCH/lib/include" "-I$SRC_DIR"

setup.cfg

@@ -5,4 +5,4 @@ description-file = README.md
 test = pytest
 
 [tool:pytest]
-addopts = --capture=no --cov
+addopts = --capture=no

setup.py

@@ -2,7 +2,7 @@ from os import path as osp
 
 from setuptools import setup, find_packages
 
-__version__ = '1.0.0'
+__version__ = '1.0.1'
 url = 'https://github.com/rusty1s/pytorch_cluster'
 
 install_requires = ['cffi']

torch_cluster/__init__.py

 from .graclus import graclus_cluster
 from .grid import grid_cluster
 
-__version__ = '1.0.0'
+__version__ = '1.0.1'
 
 __all__ = ['graclus_cluster', 'grid_cluster', '__version__']