Support ranges with different integer widths in preduce (#28651)

JuliaLang · Aug 24, 2018 · 145224f · 145224f · nanosoldier · Aug 25, 2018
1 parent 35c67e5
commit 145224f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/stdlib/Distributed/src/macros.jl b/stdlib/Distributed/src/macros.jl
@@ -251,7 +251,7 @@ end
 
 function preduce(reducer, f, R)
     N = length(R)
-    chunks = splitrange(N, nworkers())
+    chunks = splitrange(Int(N), nworkers())
     all_w = workers()[1:length(chunks)]
 
     w_exec = Task[]

diff --git a/stdlib/Distributed/test/distributed_exec.jl b/stdlib/Distributed/test/distributed_exec.jl
@@ -1527,6 +1527,14 @@ end
 a27933 = :_not_defined_27933
 @test remotecall_fetch(()->a27933, first(workers())) === a27933
 
+# PR #28651
+for T in (UInt8, Int8, UInt16, Int16, UInt32, Int32, UInt64)
+    n = @distributed (+) for i in Base.OneTo(T(10))
+        i
+    end
+    @test n == 55
+end
+
 # Run topology tests last after removing all workers, since a given
 # cluster at any time only supports a single topology.
 rmprocs(workers())