Commit e87a2251 authored by rusty1s's avatar rusty1s
Browse files

rename cuda files

parent 452097ba
......@@ -15,8 +15,8 @@ with_cuda = False
if torch.cuda.is_available():
subprocess.call(['./build.sh', osp.dirname(torch.__file__)])
headers += ['torch_scatter/src/cuda.h']
sources += ['torch_scatter/src/cuda.c']
headers += ['torch_scatter/src/gpu.h']
sources += ['torch_scatter/src/gpu.c']
include_dirs += ['torch_scatter/kernel']
define_macros += [('WITH_CUDA', None)]
extra_objects += ['torch_scatter/build/kernel.so']
......
......@@ -2,7 +2,7 @@ from os import path as osp
from setuptools import setup, find_packages
__version__ = '1.0.2'
__version__ = '1.0.3'
url = 'https://github.com/rusty1s/pytorch_scatter'
install_requires = ['cffi']
......
......@@ -6,7 +6,7 @@ from .mean import scatter_mean
from .max import scatter_max
from .min import scatter_min
__version__ = '1.0.2'
__version__ = '1.0.3'
__all__ = [
'scatter_add', 'scatter_sub', 'scatter_mul', 'scatter_div', 'scatter_mean',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment