Algebra with UniformScaling Uses Generic Fallback Scalar Indexing

[ChrisRackauckas]

using CuArrays, LinearAlgebra
CuArrays.allowscalar(false)
cu(rand(3,3)) + I

scalar getindex is disallowed
error(::String) at error.jl:33
assertscalar(::String) at indexing.jl:41
getindex at indexing.jl:96 [inlined]
_getindex at abstractarray.jl:1003 [inlined]
getindex at abstractarray.jl:980 [inlined]
+(::CuArray{Float32,2,Nothing}, ::UniformScaling{Bool}) at uniformscaling.jl:167
top-level scope at test.jl:18

[ChrisRackauckas]

It seems like broadcast is having issues too? I was trying:

using CuArrays, LinearAlgebra
A = cu(rand(3,3))
idxs = 1:3
@. setindex!((A,),getindex((A,),idxs,idxs)+getindex((I,),idxs,idxs),idxs,idxs)

scalar getindex is disallowed
error(::String) at error.jl:33
assertscalar(::String) at indexing.jl:41
getindex at indexing.jl:96 [inlined]
_getindex at abstractarray.jl:1003 [inlined]
getindex at abstractarray.jl:980 [inlined]
_broadcast_getindex_evalf at broadcast.jl:631 [inlined]
_broadcast_getindex at broadcast.jl:604 [inlined]
_getindex at broadcast.jl:627 [inlined]
_broadcast_getindex at broadcast.jl:603 [inlined]
_getindex at broadcast.jl:627 [inlined]
_getindex at broadcast.jl:627 [inlined]
_broadcast_getindex at broadcast.jl:603 [inlined]
getindex at broadcast.jl:564 [inlined]
macro expansion at broadcast.jl:910 [inlined]
macro expansion at simdloop.jl:77 [inlined]
copyto! at broadcast.jl:909 [inlined]
copyto! at broadcast.jl:864 [inlined]
copy at broadcast.jl:840 [inlined]
materialize(::Base.Broadcast.Broadcasted{Base.Broadcast.DefaultArrayStyle{1},Nothing,typeof(setindex!),Tuple{Tuple{CuArray{Float32,2,Nothing}},Base.Broadcast.Broadcasted{Base.Broadcast.DefaultArrayStyle{1},Nothing,typeof(+),Tuple{Base.Broadcast.Broadcasted{Base.Broadcast.DefaultArrayStyle{1},Nothing,typeof(getindex),Tuple{Tuple{CuArray{Float32,2,Nothing}},UnitRange{Int64},UnitRange{Int64}}},Base.Broadcast.Broadcasted{Base.Broadcast.DefaultArrayStyle{1},Nothing,typeof(getindex),Tuple{Tuple{UniformScaling{Bool}},UnitRange{Int64},UnitRange{Int64}}}}},UnitRange{Int64},UnitRange{Int64}}}) at broadcast.jl:820
top-level scope at simple_gpu.jl:47

[ChrisRackauckas]

Base.:+(A::CuArray,I::UniformScaling) = A + CuArray(I,size(A,1),size(A,2))

works but seems wasteful.

[maleadt]

This works now.