AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updating npkit_trace_generator.py to check npkit directory (#1891)

Browse Source 
* 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 <bertan.dogancay@gmail.com>
This commit is contained in:
ehsanhosseinzadehKhaligh
2025-10-22 00:51:16 -07:00
committed by GitHub
parent 724680f87c
commit aec4f0a659
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/scripts/npkit_trace_generator.py
+10 -1
View File
@@ -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)
convert_npkit_dump_to_trace(args.npkit_dump_dir, args.output_dir, npkit_event_def, gpu_statistics, args.warmup_runs)