Skip to content

GitLab

Unverified Commit 85c6c492 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #2 from shagunsodhani/master 

updated setup.py as per @rusty1s's  suggestion
parents 43c50459 00566354
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 7 deletions
setup.py
View file @ 85c6c492
...@@ -13,13 +13,10 @@ cmdclass = {} ...@@ -13,13 +13,10 @@ cmdclass = {}
if torch.cuda.is_available(): if torch.cuda.is_available():
ext_modules += [ ext_modules += [
CUDAExtension( CUDAExtension('spspmm_cuda',
'spspmm_cuda', ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'], extra_link_args=['-lcusparse', '-l', 'cusparse'],
extra_link_args=['-lcusparse'], )
),
CUDAExtension('unique_cuda',
['cuda/unique.cpp', 'cuda/unique_kernel.cu'])
] ]
cmdclass['build_ext'] = BuildExtension cmdclass['build_ext'] = BuildExtension
......
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