Skip to content

Commit 33e81aa

Browse files
committed
updated setup.py as per @rusty1s's suggestion
1 parent 92082f9 commit 33e81aa

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

setup.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,10 @@
1313

1414
if torch.cuda.is_available():
1515
ext_modules += [
16-
CUDAExtension(
17-
'spspmm_cuda',
18-
['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
19-
extra_link_args=['-lcusparse'],
20-
),
21-
CUDAExtension('unique_cuda',
22-
['cuda/unique.cpp', 'cuda/unique_kernel.cu'])
16+
CUDAExtension('spspmm_cuda',
17+
['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
18+
extra_link_args=['-l', 'cusparse']
19+
)
2320
]
2421
cmdclass['build_ext'] = BuildExtension
2522

0 commit comments

Comments
 (0)