[Distributed] make nextproc use atomic operations

JuliaLang · Sep 13, 2021 · 4627729 · 4627729
1 parent 70cc57c
commit 4627729
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/stdlib/Distributed/src/macros.jl b/stdlib/Distributed/src/macros.jl
@@ -1,14 +1,10 @@
 # This file is a part of Julia. License is MIT: https://julialang.org/license
 
-let nextidx = 0
+let nextidx = Threads.Atomic{Int}(0)
     global nextproc
     function nextproc()
-        p = -1
-        if p == -1
-            p = workers()[(nextidx % nworkers()) + 1]
-            nextidx += 1
-        end
-        p
+        idx = Threads.atomic_add!(nextidx, 1)
+        return workers()[(idx % nworkers()) + 1]
     end
 end