format

Author: ErikQQY

ext/SparseMatrixColoringsCUDAExt.jl

@@ -7,10 +7,14 @@ using cuSPARSE: AbstractCuSparseMatrix, CuSparseMatrixCSC, CuSparseMatrixCSR
 
 ## Basic support for GPU sparsity pattern stuff
 
-SMC.SparsityPatternCSC(A::CuSparseMatrixCSC) = SMC.SparsityPatternCSC(first(A.dims), last(A.dims), A.colPtr, A.rowVal)
+function SMC.SparsityPatternCSC(A::CuSparseMatrixCSC)
+    SMC.SparsityPatternCSC(first(A.dims), last(A.dims), A.colPtr, A.rowVal)
+end
 
 for R in (:Diagonal, :Bidiagonal, :Tridiagonal)
-    @eval function SMC.BipartiteGraph(A::$R{T, <:CuArray}; symmetric_pattern::Bool=false) where {T}
+    @eval function SMC.BipartiteGraph(
+        A::$R{T,<:CuArray}; symmetric_pattern::Bool=false
+    ) where {T}
         return SMC.BipartiteGraph(CuSparseMatrixCSC(A); symmetric_pattern)
     end
 end

ext/SparseMatrixColoringsColorsExt.jl

@@ -271,8 +271,9 @@ function show_colors!(
     A_ccolor_indices = mod1.(column_colors(res), length(colorscheme))
     A_rcolor_indices = mod1.(row_shift .+ row_colors(res), length(colorscheme))
     B_ccolor_indices = mod1.(1:maximum(column_colors(res)), length(colorscheme))
-    B_rcolor_indices =
-        mod1.((row_shift + 1):(row_shift + maximum(row_colors(res))), length(colorscheme))
+    B_rcolor_indices = mod1.(
+        (row_shift + 1):(row_shift + maximum(row_colors(res))), length(colorscheme)
+    )
     A_ccolors = colorscheme[A_ccolor_indices]
     A_rcolors = colorscheme[A_rcolor_indices]
     B_ccolors = colorscheme[B_ccolor_indices]

ext/SparseMatrixColoringsJuMPExt.jl

@@ -29,7 +29,7 @@ function optimal_distance2_coloring(
     model = Model(optimizer)
     silent && set_silent(model)
     # one variable per vertex to color, removing some renumbering symmetries
-    @variable(model, 1 <= color[i=1:n] <= i, Int)
+    @variable(model, 1 <= color[i = 1:n] <= i, Int)
     # one variable to count the number of distinct colors
     @variable(model, ncolors, Int)
     @constraint(model, [ncolors; color] in MOI.CountDistinct(n + 1))