Support new zeros/ones methods from #19635

Author: timholy

src/Compat.jl

@@ -1411,6 +1411,18 @@ if VERSION < v"0.6.0-dev.2840"
     IndexStyle(args...) = Base.linearindexing(args...)
 end
 
+if VERSION < v"0.6.0-dev.1653"
+    for (fname, felt) in ((:zeros,:zero), (:ones,:one))
+        @eval begin
+            # allow signature of similar
+            Base.$fname(a::AbstractArray, T::Type, dims::Tuple) = fill!(similar(a, T, dims), $felt(T))
+            Base.$fname(a::AbstractArray, T::Type, dims...) = fill!(similar(a,T,dims...), $felt(T))
+            Base.$fname(a::AbstractArray, T::Type=eltype(a)) = fill!(similar(a,T), $felt(T))
+        end
+    end
+end
+
+
 include("to-be-deprecated.jl")
 
 end # module Compat

test/runtests.jl

@@ -1763,4 +1763,17 @@ let a = CompatArray.CartesianArray(rand(2,3)), b = CompatArray.LinearArray(rand(
     @test IndexStyle(b) === IndexLinear()
 end
 
+for (A,val) in ((zeros(1:5, Float32, 3, 2), 0),
+                (ones(1:5, Float32, 3, 2), 1),
+                (zeros(1:5, Float32, (3, 2)), 0),
+                (ones(1:5, Float32, (3, 2)), 1))
+    @test isa(A, Matrix{Float32}) && size(A) == (3,2) && all(x->x==val, A)
+end
+for (A,val) in ((zeros(1:5, Float32), 0),
+                (ones(1:5, Float32), 1))
+    @test isa(A, Vector{Float32}) && size(A) == (5,) && all(x->x==val, A)
+end
+
 include("to-be-deprecated.jl")
+
+nothing