Support Ref

Author: dlfivefifty

src/fillbroadcast.jl

@@ -51,6 +51,8 @@ end
 broadcasted(::DefaultArrayStyle{N}, op, r::AbstractFill{T,N}) where {T,N} = Fill(op(getindex_value(r)), size(r))
 broadcasted(::DefaultArrayStyle{N}, op, r::AbstractFill{T,N}, x::Number) where {T,N} = Fill(op(getindex_value(r),x), size(r))
 broadcasted(::DefaultArrayStyle{N}, op, x::Number, r::AbstractFill{T,N}) where {T,N} = Fill(op(x, getindex_value(r)), size(r))
+broadcasted(::DefaultArrayStyle{N}, op, r::AbstractFill{T,N}, x::Ref) where {T,N} = Fill(op(getindex_value(r),x[]), size(r))
+broadcasted(::DefaultArrayStyle{N}, op, x::Ref, r::AbstractFill{T,N}) where {T,N} = Fill(op(x[], getindex_value(r)), size(r))
 function broadcasted(::DefaultArrayStyle{N}, op, r1::AbstractFill{T,N}, r2::AbstractFill{V,N}) where {T,V,N}
     size(r1) ≠ size(r2) && throw(DimensionMismatch("dimensions must match."))
     Fill(op(getindex_value(r1),getindex_value(r2)), size(r1))

test/runtests.jl

@@ -448,6 +448,11 @@ end
     @test Zeros(5) .* Fill(5.0, 5) == Zeros(5)
     @test Ones(5) .* Zeros(5) == Zeros(5)
     @test Fill(5.0, 5) .* Zeros(5) == Zeros(5)
+
+    # support Ref
+    @test Fill(1,10) .- 1 ≡ Fill(1,10) .- Ref(1) ≡ Fill(1,10) .- Ref(1I)
+    @test Fill([1 2; 3 4],10) .- Ref(1I) == Fill([0 2; 3 3],10)
+    @test Ref(1I) .+ Fill([1 2; 3 4],10) == Fill([2 2; 3 5],10)
 end
 
 @testset "Sub-arrays" begin