Skip to content

GitLab

Commit 00566354 authored by Shagun Sodhani's avatar Shagun Sodhani
Browse files

updated setup.py to use -lcusparse flag

parent 33e81aa6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 00566354
...@@ -15,7 +15,7 @@ if torch.cuda.is_available(): ...@@ -15,7 +15,7 @@ if torch.cuda.is_available():
ext_modules += [ ext_modules += [
CUDAExtension('spspmm_cuda', CUDAExtension('spspmm_cuda',
['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'], ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
extra_link_args=['-l', 'cusparse'] extra_link_args=['-lcusparse', '-l', 'cusparse'],
) )
] ]
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