Don't use cusparseSpMM_process with old versions of CUSPARSE

JuliaGPU · Nov 29, 2023 · dfe4cd1 · dfe4cd1
1 parent 5aa719b
commit dfe4cd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)