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 0e0326d929db8e328b0aa1ce2b091db248d6c956..73eb9dc045b71319e050437bfceb5a3390ac2791 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]