From 702ccfdbc2ee8b5b75d38bc6fd486db8704f1c19 Mon Sep 17 00:00:00 2001 From: gitee Date: Mon, 21 Apr 2025 16:30:44 +0800 Subject: [PATCH] fix multi-node bug --- .../experimental_tools/mindie_benchmark/mindie_llm.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ais-bench_workload/experimental_tools/mindie_benchmark/mindie_llm.py b/ais-bench_workload/experimental_tools/mindie_benchmark/mindie_llm.py index 0e0326d9..73eb9dc0 100644 --- a/ais-bench_workload/experimental_tools/mindie_benchmark/mindie_llm.py +++ b/ais-bench_workload/experimental_tools/mindie_benchmark/mindie_llm.py @@ -54,8 +54,11 @@ def main(): node_rank = run_cfg.get(Const.NODE_RANK, 0) if not os.path.exists(cfg.work_dir) or not os.listdir( - cfg.work_dir) and node_rank == 0: - raise ValueError(f"No performance data found in {cfg.work_dir}. Please check if the run was successful.") + cfg.work_dir): + if node_rank == 0: + raise ValueError(f"No performance data found in {cfg.work_dir}. Please check if the run was successful.") + else: + print("Please check performance data in node 0 machine.") else: dirs = os.listdir(cfg.work_dir) dir_time_str = sorted(dirs)[-1] -- Gitee