fixed cusparse linking

9fb07940 · rusty1s · 9c0419db · 9fb07940
Commit 9fb07940 authored Aug 28, 2018 by rusty1s
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
@@ -16,7 +16,7 @@ if torch.cuda.is_available():
        CUDAExtension(
            'spspmm_cuda',
            ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
-            extra_link_args=['-l', 'cusparse'],
+            extra_link_args=['-lcusparse'],
        )
    ]
    cmdclass['build_ext'] = BuildExtension