fix load dir logic

PaddlePaddle · Nov 20, 2023 · 0bff0af · 0bff0af
1 parent 5cfd132
commit 0bff0af
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/paddle/distributed/auto_parallel/static/profiler_helper_static.py b/python/paddle/distributed/auto_parallel/static/profiler_helper_static.py
@@ -141,9 +141,10 @@ def process_one_machine_log(log_dir, multi_machine_idx=-1):
     if args.multi_machine:
         multi_machine_dirs = os.listdir(args.log_dir)
         multi_machine_dirs = [
-            os.path.join(args.log_dir, f"machine{i}")
-            for i in range(len(multi_machine_dirs))
-            if os.path.isdir(os.path.join(args.log_dir, f"machine{i}"))
+            os.path.join(args.log_dir, d)
+            for d in multi_machine_dirs
+            if d.startswith("machine")
+            and os.path.isdir(os.path.join(args.log_dir, d))
         ]
         machine_num = len(multi_machine_dirs)
         for i, d in enumerate(multi_machine_dirs):