From aec4f0a659d739aef876f37ae4a5528a3833b893 Mon Sep 17 00:00:00 2001 From: ehsanhosseinzadehKhaligh Date: Wed, 22 Oct 2025 00:51:16 -0700 Subject: [PATCH] Updating npkit_trace_generator.py to check npkit directory (#1891) * create dir regardless of default or user-provided path if it doesn't exist * Fix npkit_dump_dir on npkit_trace_generator.py --------- Co-authored-by: BertanDogancay --- tools/scripts/npkit_trace_generator.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tools/scripts/npkit_trace_generator.py b/tools/scripts/npkit_trace_generator.py index d09b5b9ac0..727b85b7b4 100644 --- a/tools/scripts/npkit_trace_generator.py +++ b/tools/scripts/npkit_trace_generator.py @@ -287,6 +287,15 @@ def parse_cpu_event_file(npkit_dump_dir, npkit_event_def, rank, channel, cpu_clo def convert_npkit_dump_to_trace(npkit_dump_dir, output_dir, npkit_event_def, gpu_statistics, warmup_runs=0): + + # check if directory exists, if not create it + if not os.path.exists(npkit_dump_dir): + #print(f"Directory {npkit_dump_dir} does not exist. Creating it...") + os.makedirs(npkit_dump_dir, exist_ok=True) + else: + #print(f"Directory {npkit_dump_dir} already exists.") + pass + files_in_dump_dir = next(os.walk(npkit_dump_dir))[2] gpu_event_files = [x for x in files_in_dump_dir if x.startswith('gpu_events_rank_')] cpu_event_files = [x for x in files_in_dump_dir if x.startswith('cpu_events_rank_')] @@ -351,4 +360,4 @@ if __name__ == '__main__': if args.gpu_run_stats is not None: gpu_statistics = args.gpu_run_stats npkit_event_def = parse_npkit_event_header(args.npkit_event_header_path) - convert_npkit_dump_to_trace(args.npkit_dump_dir, args.output_dir, npkit_event_def, gpu_statistics, args.warmup_runs) \ No newline at end of file + convert_npkit_dump_to_trace(args.npkit_dump_dir, args.output_dir, npkit_event_def, gpu_statistics, args.warmup_runs)