[CUSPARSE] Use cusparseSDDMM_preprocess (#2184)

JuliaGPU · Nov 29, 2023 · 62063dd · 62063dd
1 parent 9c281af
commit 62063dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/cusparse/generic.jl b/lib/cusparse/generic.jl
@@ -617,8 +617,8 @@ function sddmm!(transa::SparseChar, transb::SparseChar, alpha::Number, A::DenseC
         return out[]
     end
     with_workspace(bufferSize) do buffer
-        # Uncomment if we find a way to reuse the buffer (issue #1362)
-        # cusparseSDDMM_preprocess(handle(), transa, transb, Ref{T}(alpha), descA, descB, Ref{T}(beta), descC, T, algo, buffer)
+        # We should find a way to reuse the buffer (issue #1362)
+        cusparseSDDMM_preprocess(handle(), transa, transb, Ref{T}(alpha), descA, descB, Ref{T}(beta), descC, T, algo, buffer)
         cusparseSDDMM(handle(), transa, transb, Ref{T}(alpha), descA, descB, Ref{T}(beta), descC, T, algo, buffer)
     end
     return C