From 2af13430937c7b545870219f9fc6382126b3066c Mon Sep 17 00:00:00 2001
From: jishnub <jishnub@users.noreply.github.com>
Date: Sat, 8 May 2021 18:21:03 +0400
Subject: [PATCH] Add zero

---
 Project.toml   | 2 +-
 src/darray.jl  | 1 +
 test/darray.jl | 6 ++++++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index b1f2f4c..9ed2197 100644
--- a/Project.toml
+++ b/Project.toml
@@ -12,8 +12,8 @@ SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [compat]
-julia = "1"
 Primes = "0.4"
+julia = "1"
 
 [extras]
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
diff --git a/src/darray.jl b/src/darray.jl
index c4b12b1..902aa4c 100644
--- a/src/darray.jl
+++ b/src/darray.jl
@@ -485,6 +485,7 @@ dzeros(::Type{T}, d1::Integer, drest::Integer...) where {T} = dzeros(T, convert(
 dzeros(d1::Integer, drest::Integer...) = dzeros(Float64, convert(Dims, tuple(d1, drest...)))
 dzeros(d::Dims) = dzeros(Float64, d)
 
+Base.zero(d::DArray) = dzeros(eltype(d), size(d), procs(d))
 
 """
     dones(dims, ...)
diff --git a/test/darray.jl b/test/darray.jl
index 4a8d3df..fc25612 100644
--- a/test/darray.jl
+++ b/test/darray.jl
@@ -562,6 +562,12 @@ check_leaks()
         @test size(A) == (10,10)
         close(A)
     end
+
+    @testset "zero" begin
+        A = dzeros(Int, 10, 10)
+        B = zero(A)
+        @test A == B
+    end
 end
 
 check_leaks()