updated setup.py to use -lcusparse flag

00566354 · Shagun Sodhani · 33e81aa6 · 00566354
Commit 00566354 authored Oct 20, 2018 by Shagun Sodhani
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@ if torch.cuda.is_available():
    ext_modules += [
        CUDAExtension('spspmm_cuda',
                      ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
-                      extra_link_args=['-l', 'cusparse']
+                      extra_link_args=['-lcusparse', '-l', 'cusparse'],
                      )
    ]
    cmdclass['build_ext'] = BuildExtension