diff --git a/src/trace_link/trace_linker.py b/src/trace_link/trace_linker.py index af247dd7..4a08d0b4 100644 --- a/src/trace_link/trace_linker.py +++ b/src/trace_link/trace_linker.py @@ -117,7 +117,8 @@ def load_sync_dependencies( sync_dependencies = {} absolute_kineto_file = os.path.abspath(kineto_file) trace_dir = os.path.dirname(absolute_kineto_file) - trace_analysis = TraceAnalysis(trace_dir=trace_dir, trace_files={rank: kineto_file}) + kineto_file_base = os.path.basename(kineto_file) + trace_analysis = TraceAnalysis(trace_dir=trace_dir, trace_files={rank: kineto_file_base}) try: cp_graph, success = trace_analysis.critical_path_analysis( rank=rank, annotation=annotation, instance_id=instance_id