barrier to avoid race condn

zjwegert · May 24, 2024 · 11bea63 · 11bea63
1 parent a0fa593
commit 11bea63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Io.jl b/src/Io.jl
@@ -43,7 +43,7 @@ a set of JLD2 files corresponding to each part.
 function psave(dir::AbstractString, x)
   ranks = get_parts(x)
   i_am_main(ranks) && mkpath(dir)
-  fetch(ranks)
+  PartitionedArrays.barrier(ranks)
   arr = to_local_storage(x)
   map(ranks,arr) do id, arr
     filename = joinpath(dir,basename(dir)*"_$id.jdl2")