Add cublas as link library for pylibcugraph

Signed-off-by: Jordan Jacobelli <[email protected]>
rapidsai · May 13, 2022 · 3362771 · 3362771
1 parent cba547e
commit 3362771
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pylibcugraph/setup.py b/python/pylibcugraph/setup.py
@@ -122,7 +122,7 @@ def run(self):
                   os.path.join(os.sys.prefix, "lib")
               ],
               libraries=['cudart', 'cusparse', 'cusolver', 'cugraph', 'nccl',
-                         'cugraph_c'],
+                         'cugraph_c', 'cublas'],
               language='c++',
               extra_compile_args=['-std=c++17'])
 ]