1- using CUDA . CUSPARSE : CuSparseMatrixCSC, CuSparseMatrixCSR
1+ using cuSPARSE : CuSparseMatrixCSC, CuSparseMatrixCSR
22using LinearAlgebra
33using SparseArrays
44using SparseMatrixColorings
@@ -21,34 +21,34 @@ symmetric_params = vcat(
2121)
2222
2323@testset verbose = true " Column coloring & decompression" begin
24- problem = ColoringProblem (; structure= :nonsymmetric , partition= :column )
25- algo = GreedyColoringAlgorithm (; decompression= :direct )
24+ problem = ColoringProblem (; structure = :nonsymmetric , partition = :column )
25+ algo = GreedyColoringAlgorithm (; decompression = :direct )
2626 @testset for T in (CuSparseMatrixCSC, CuSparseMatrixCSR)
2727 @testset " $((; m, n, p)) " for (m, n, p) in asymmetric_params
2828 A0 = T (sprand (rng, m, n, p))
29- test_coloring_decompression (A0, problem, algo; gpu= true )
29+ test_coloring_decompression (A0, problem, algo; gpu = true )
3030 end
3131 end
3232end ;
3333
3434@testset verbose = true " Row coloring & decompression" begin
35- problem = ColoringProblem (; structure= :nonsymmetric , partition= :row )
36- algo = GreedyColoringAlgorithm (; decompression= :direct )
35+ problem = ColoringProblem (; structure = :nonsymmetric , partition = :row )
36+ algo = GreedyColoringAlgorithm (; decompression = :direct )
3737 @testset for T in (CuSparseMatrixCSC, CuSparseMatrixCSR)
3838 @testset " $((; m, n, p)) " for (m, n, p) in asymmetric_params
3939 A0 = T (sprand (rng, m, n, p))
40- test_coloring_decompression (A0, problem, algo; gpu= true )
40+ test_coloring_decompression (A0, problem, algo; gpu = true )
4141 end
4242 end
4343end ;
4444
4545@testset verbose = true " Symmetric coloring & direct decompression" begin
46- problem = ColoringProblem (; structure= :symmetric , partition= :column )
47- algo = GreedyColoringAlgorithm (; postprocessing= false , decompression= :direct )
46+ problem = ColoringProblem (; structure = :symmetric , partition = :column )
47+ algo = GreedyColoringAlgorithm (; postprocessing = false , decompression = :direct )
4848 @testset for T in (CuSparseMatrixCSC, CuSparseMatrixCSR)
4949 @testset " $((; n, p)) " for (n, p) in symmetric_params
5050 A0 = T (sparse (Symmetric (sprand (rng, n, n, p))))
51- test_coloring_decompression (A0, problem, algo; gpu= true )
51+ test_coloring_decompression (A0, problem, algo; gpu = true )
5252 end
5353 A0 = T (sparse (Diagonal (ones (10 ))))
5454 result = coloring (A0, problem, algo)
0 commit comments