updated setup.py to use -lcusparse flag

shagunsodhani · shagunsodhani · commit 00566354d243 · 2018-10-20T09:41:44.000-04:00
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     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

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	`ext_modules += [`
`16`	`16`	`CUDAExtension('spspmm_cuda',`
`17`	`17`	`['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],`
`18`		`- extra_link_args=['-l', 'cusparse']`
	`18`	`+ extra_link_args=['-lcusparse', '-l', 'cusparse'],`
`19`	`19`	`)`
`20`	`20`	`]`
`21`	`21`	`cmdclass['build_ext'] = BuildExtension`