updated setup.py as per @rusty1s's suggestion

33e81aa6 · Shagun Sodhani · 92082f98 · 33e81aa6
Commit 33e81aa6 authored Oct 19, 2018 by Shagun Sodhani
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 7 deletions

setup.py setup.py +4 -7

No files found.
--- a/setup.py
+++ b/setup.py
@@ -13,13 +13,10 @@ cmdclass = {}

 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=['-l', 'cusparse']
+                      )
    ]
    cmdclass['build_ext'] = BuildExtension