Skip to content

Commit 426bc72

Browse files
committed
Fix ambiguity
1 parent 1d2359e commit 426bc72

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

ext/SparseMatrixColoringsGPUArraysExt.jl

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,18 @@ SMC.matrix_versions(A::AbstractGPUSparseMatrix) = (A,)
88

99
## Compression (slow, through CPU)
1010

11-
function SMC.compress(A::AbstractGPUSparseMatrix, result::SMC.AbstractColoringResult)
11+
function SMC.compress(
12+
A::AbstractGPUSparseMatrix, result::SMC.AbstractColoringResult{structure,:column}
13+
) where {structure}
14+
A_cpu = SparseMatrixCSC(A)
15+
B_cpu = SMC.compress(A_cpu, result)
16+
B = dense_array_type(A)(B_cpu)
17+
return B
18+
end
19+
20+
function SMC.compress(
21+
A::AbstractGPUSparseMatrix, result::SMC.AbstractColoringResult{structure,:row}
22+
) where {structure}
1223
A_cpu = SparseMatrixCSC(A)
1324
B_cpu = SMC.compress(A_cpu, result)
1425
B = dense_array_type(A)(B_cpu)

0 commit comments

Comments
 (0)