Skip to content

GitLab

Commit 072a17c4 authored by rusty1s's avatar rusty1s
Browse files

fix

parent bc522dd9
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
setup.py
View file @ 072a17c4
...@@ -21,9 +21,8 @@ cmdclass = {'build_ext': torch.utils.cpp_extension.BuildExtension} ...@@ -21,9 +21,8 @@ cmdclass = {'build_ext': torch.utils.cpp_extension.BuildExtension}
ext_modules = [] ext_modules = []
exts = [e.split(osp.sep)[-1][:-4] for e in glob(osp.join('cpu', '*.cpp'))] exts = [e.split(osp.sep)[-1][:-4] for e in glob(osp.join('cpu', '*.cpp'))]
ext_modules += [ ext_modules += [
CppExtension( CppExtension(f'torch_sparse.{ext}_cpu', [f'cpu/{ext}.cpp'],
f'torch_sparse.{ext}_cpu', [f'cpu/{ext}.cpp'], extra_compile_args=cxx_extra_compile_args) for ext in exts
extra_compile_args=cxx_extra_compile_args) for ext in exts
] ]
if CUDA_HOME is not None and '--cpu' not in argv: if CUDA_HOME is not None and '--cpu' not in argv:
...@@ -44,6 +43,8 @@ if CUDA_HOME is not None and '--cpu' not in argv: ...@@ -44,6 +43,8 @@ if CUDA_HOME is not None and '--cpu' not in argv:
extra_link_args=extra_link_args, extra_link_args=extra_link_args,
) for ext in exts ) for ext in exts
] ]
if '--cpu' in argv:
argv.remove('--cpu')
__version__ = '0.4.3' __version__ = '0.4.3'
url = 'https://github.com/rusty1s/pytorch_sparse' url = 'https://github.com/rusty1s/pytorch_sparse'
......
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