diff --git a/test/libraries/cusparse/generic.jl b/test/libraries/cusparse/generic.jl index e47c17ad77..3f1c15ee62 100644 --- 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]