CUSPARSE: Adding a value to the diagonal

[ChrisRackauckas]

I cannot figure out how to add a value to the diagonal, and it's showing some missing dispatches.

using CUDA
CUDA.allowscalar(false)
W = cu(sprand(100,100,0.1))
J = cu(sprand(100,100,0.1))
idxs = diagind(W)
λ = 1.0
@. @view(W[idxs]) = @view(W[idxs]) + λ # Scalar indexing

Base.:+(A::CUSPARSE.CuSparseMatrix,I::UniformScaling) = A + Diagonal(CuArray(ones(size(A,1))))
Base.:-(A::CUSPARSE.CuSparseMatrix,I::UniformScaling) = A - Diagonal(CuArray(ones(size(A,1))))

@show typeof(W + λ*I) # CuArray not sparse :(
W .= W + λ*I # Scalar index

This is the last piece to finish SciML/OrdinaryDiffEq.jl#1566 I think.

[maleadt]

#1390