[BugFix] Fix bugs in calling cusparse API (#259)

Co-authored-by: fishmingyu <fishmingyu@github.com>
THUDM · Jul 17, 2021 · 6b8494d · 6b8494d
1 parent 26a2433
commit 6b8494d
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/cogdl/operators/mhspmm.py b/cogdl/operators/mhspmm.py
@@ -19,14 +19,16 @@
     )
     mhtranspose = load(
         name="mhtranspose",
+        extra_cflags=['-lcusparse'],
         sources=[os.path.join(path, "spmm/mhTranspose.cpp"), os.path.join(path, "spmm/mhTranspose.cu")],
-        verbose=False,
+        verbose=True,
     )
 
     spmm = load(
         name="spmm",
+        extra_cflags=['-lcusparse'],
         sources=[os.path.join(path, "spmm/spmm.cpp"), os.path.join(path, "spmm/spmm_kernel.cu")],
-        verbose=False,
+        verbose=True,
     )
 
     def csrmhspmm(rowptr, colind, feat, attention):

diff --git a/cogdl/operators/spmm.py b/cogdl/operators/spmm.py
@@ -9,8 +9,9 @@
 try:
     spmm = load(
         name="spmm",
+        extra_cflags=['-lcusparse'],
         sources=[os.path.join(path, "spmm/spmm.cpp"), os.path.join(path, "spmm/spmm_kernel.cu")],
-        verbose=False,
+        verbose=True,
     )
     sddmm = load(
         name="sddmm",

diff --git a/cogdl/operators/spmm/mhTranspose.cu b/cogdl/operators/spmm/mhTranspose.cu
@@ -54,6 +54,7 @@ void csr2cscKernel(int m, int n, int nnz,
 )
 {
     cusparseHandle_t handle;
+    checkCuSparseError(cusparseCreate(&handle));
     size_t bufferSize = 0;
     void* buffer = NULL;
     checkCuSparseError(cusparseCsr2cscEx2_bufferSize(handle,

diff --git a/cogdl/operators/spmm/spmm_kernel.cu b/cogdl/operators/spmm/spmm_kernel.cu
@@ -360,6 +360,7 @@ void csr2cscKernel(int m, int n, int nnz,
 )
 {
     cusparseHandle_t handle;
+    checkCuSparseError(cusparseCreate(&handle));
     size_t bufferSize = 0;
     void* buffer = NULL;
     checkCuSparseError(cusparseCsr2cscEx2_bufferSize(handle,