From dfe4cd1296575af5782dbde064ba1ce0a17b22cc Mon Sep 17 00:00:00 2001 From: Alexis Montoison Date: Wed, 29 Nov 2023 14:19:39 -0600 Subject: [PATCH] Don't use cusparseSpMM_process with old versions of CUSPARSE --- lib/cusparse/generic.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)