Currently, the preconditioning step requires a "power sum" over columns or rows of a sparse matrix. It is very likely that this operation is not implemented efficiently for all the matrix types we care about (even though we have a good SparseMatrixCSC version).
If we replace this operation with a map (raising to the correct power) followed by a multiplication with a vector of ones, we might get something that most GPU sparse matrices support out of the box
Currently, the preconditioning step requires a "power sum" over columns or rows of a sparse matrix. It is very likely that this operation is not implemented efficiently for all the matrix types we care about (even though we have a good
SparseMatrixCSCversion).If we replace this operation with a
map(raising to the correct power) followed by a multiplication with a vector of ones, we might get something that most GPU sparse matrices support out of the box