Skip to content

Commit de6a9fd

Browse files
committed
fixed unique install
1 parent 8f8398e commit de6a9fd

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

setup.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@
1717
'spspmm_cuda',
1818
['cuda/spspmm.cpp', 'cuda/spspmm_kernel.cu'],
1919
extra_link_args=['-lcusparse', '-l', 'cusparse'],
20-
)
20+
),
21+
CUDAExtension('unique_cuda',
22+
['cuda/unique.cpp', 'cuda/unique_kernel.cu']),
2123
]
2224
cmdclass['build_ext'] = BuildExtension
2325

0 commit comments

Comments
 (0)