Merge pull request #2 from shagunsodhani/master

updated setup.py as per @rusty1s's  suggestion

Author: rusty1s

setup.py

@@ -13,13 +13,10 @@
 
 if torch.cuda.is_available():
     ext_modules += [
-        CUDAExtension(
-            'spspmm_cuda',
-            ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
-            extra_link_args=['-lcusparse'],
-        ),
-        CUDAExtension('unique_cuda',
-                      ['cuda/unique.cpp', 'cuda/unique_kernel.cu'])
+        CUDAExtension('spspmm_cuda',
+                      ['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
+                      extra_link_args=['-lcusparse', '-l', 'cusparse'],
+                      )
     ]
     cmdclass['build_ext'] = BuildExtension