Use run_only_rank0 to broadcast allocation data

BlueBrain · Feb 21, 2024 · 8e3b3ba · 8e3b3ba
1 parent a844031
commit 8e3b3ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neurodamus/utils/memory.py b/neurodamus/utils/memory.py
@@ -200,7 +200,8 @@ def export_allocation_stats(rank_allocation, filename):
         f.write(compressed_data)
 
 
-def import_allocation_stats(filename):
+@run_only_rank0
+def import_allocation_stats(filename) -> dict:
     """
     Import allocation dictionary from serialized pickle file.
     """