diff --git a/src/trace_link/trace_linker.py b/src/trace_link/trace_linker.py index 71696ca8..1123e45d 100644 --- a/src/trace_link/trace_linker.py +++ b/src/trace_link/trace_linker.py @@ -117,7 +117,7 @@ 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_analysis = TraceAnalysis(trace_dir=trace_dir, trace_files={rank: kineto_file}) cp_graph, success = trace_analysis.critical_path_analysis( rank=rank, annotation=annotation, instance_id=instance_id )