Include directory path in create_directory barrier key.

PiperOrigin-RevId: 658790764
google · Aug 5, 2024 · f892887 · f892887
1 parent f84f454
commit f892887
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/checkpoint/orbax/checkpoint/checkpoint_manager.py b/checkpoint/orbax/checkpoint/checkpoint_manager.py
@@ -368,7 +368,7 @@ def _create_root_directory(
     logging.info('Created directory=%s', directory)
   multihost.sync_global_processes(
       multihost.unique_barrier_key(
-          'CheckpointManager:create_directory',
+          'CheckpointManager:create_directory:' + str(directory),
           prefix=multiprocessing_options.barrier_sync_key_prefix,
           suffix=None,
       ),