Special case scalar broadcasting

Author: mtfishman

src/blocksparsearrayinterface/broadcast.jl

@@ -36,6 +36,14 @@ function Base.similar(bc::Broadcasted{<:BlockSparseArrayStyle}, elt::Type)
   return similar(first(m.args), elt, combine_axes(axes.(m.args)...))
 end
 
+# Catches cases like `dest .= value` or `dest .= value1 .+ value2`.
+# If the RHS is zero, this makes sure that the storage is emptied,
+# which is logic that is handled by `fill!`.
+function copyto_blocksparse!(dest::AbstractArray, bc::Broadcasted{<:AbstractArrayStyle{0}})
+  value = bc.f(bc.args...)
+  return fill!(dest, value)
+end
+
 # Broadcasting implementation
 # TODO: Delete this in favor of `DerivableInterfaces` version.
 function copyto_blocksparse!(dest::AbstractArray, bc::Broadcasted)

test/test_gradedunitrangesext.jl

@@ -325,7 +325,7 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
     a = cat(a1, a2; dims=(1, 2))
     b = a[r, dual(r)]
     @test eltype(b) === elt
-    @test b isa BlockSpareMatrix{elt}
+    @test b isa BlockSparseMatrix{elt}
     @test blockstoredlength(b) == 2
     @test b[Block(1, 1)] == a1
     @test iszero(b[Block(2, 1)])