[CUSPARSE] Use CUSPARSE_SPMV_COO_ALG2 (#2182)

JuliaGPU · Dec 5, 2023 · 7fb3f79 · 7fb3f79
1 parent 9b52f36
commit 7fb3f79
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/libraries/cusparse/generic.jl b/test/libraries/cusparse/generic.jl
@@ -37,7 +37,6 @@ if CUSPARSE.version() >= v"11.4.1" # lower CUDA version doesn't support these al
 
     if CUSPARSE.version() >= v"11.7.2"
         SPMV_ALGOS[CuSparseMatrixCOO] = [CUSPARSE.CUSPARSE_SPMV_ALG_DEFAULT,
-                                         # CUSPARSE.CUSPARSE_SPMV_COO_ALG2,
                                          CUSPARSE.CUSPARSE_SPMV_COO_ALG1]
 
         SPMM_ALGOS[CuSparseMatrixCOO] = [CUSPARSE.CUSPARSE_SPMM_ALG_DEFAULT,
@@ -50,6 +49,10 @@ if CUSPARSE.version() >= v"11.4.1" # lower CUDA version doesn't support these al
         push!(SPMM_ALGOS[CuSparseMatrixCOO], CUSPARSE.CUSPARSE_SPMM_COO_ALG2)
     end
 
+    if CUSPARSE.version() >= v"12.1.3"
+        push!(SPMV_ALGOS[CuSparseMatrixCOO], CUSPARSE.CUSPARSE_SPMV_COO_ALG2)
+    end
+
     for SparseMatrixType in keys(SPMV_ALGOS)
         @testset "$SparseMatrixType -- mv! algo=$algo" for algo in SPMV_ALGOS[SparseMatrixType]
             @testset "mv! $T" for T in [Float32, Float64, ComplexF32, ComplexF64]