diff --git a/lib/cusparse/generic.jl b/lib/cusparse/generic.jl index 5d2dcc7e4d..df933da29a 100644 --- a/lib/cusparse/generic.jl +++ b/lib/cusparse/generic.jl @@ -236,7 +236,7 @@ function mm!(transa::SparseChar, transb::SparseChar, alpha::Number, A::Union{CuS end with_workspace(bufferSize) do buffer # We should find a way to reuse the buffer (issue #1362) - if !(A isa CuSparseMatrixCOO) + if !(A isa CuSparseMatrixCOO) && (CUSPARSE.version() ≥ v"11.7.2") cusparseSpMM_preprocess( handle(), transa, transb, Ref{T}(alpha), descA, descB, Ref{T}(beta), descC, T, algo, buffer) @@ -295,7 +295,7 @@ function mm!(transa::SparseChar, transb::SparseChar, alpha::Number, A::DenseCuMa end with_workspace(bufferSize) do buffer # We should find a way to reuse the buffer (issue #1362) - if !(B isa CuSparseMatrixCOO) + if !(B isa CuSparseMatrixCOO) && (CUSPARSE.version() ≥ v"11.7.2") cusparseSpMM_preprocess( handle(), transb, transa, Ref{T}(alpha), descB, descA, Ref{T}(beta), descC, T, algo, buffer)