Skip to content

Commit 998d4db

Browse files
committed
Use copyto! of a view
1 parent 25b1dc7 commit 998d4db

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ext/SparseMatrixColoringsCUDAExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -103,15 +103,15 @@ for R in (:ColumnColoringResult, :RowColoringResult, :StarSetColoringResult)
103103
A::CuSparseMatrixCSC, B::CuMatrix, result::SMC.$R{<:CuSparseMatrixCSC}
104104
)
105105
compressed_indices = result.additional_info.compressed_indices_gpu_csc
106-
map!(Base.Fix1(getindex, B), A.nzVal, compressed_indices)
106+
copyto!(A.nzVal, view(B, compressed_indices))
107107
return A
108108
end
109109

110110
@eval function SMC.decompress!(
111111
A::CuSparseMatrixCSR, B::CuMatrix, result::SMC.$R{<:CuSparseMatrixCSR}
112112
)
113113
compressed_indices = result.additional_info.compressed_indices_gpu_csr
114-
map!(Base.Fix1(getindex, B), A.nzVal, compressed_indices)
114+
copyto!(A.nzVal, view(B, compressed_indices))
115115
return A
116116
end
117117
end

0 commit comments

Comments
 (0)