Resolve kineto_file path bug with absolute path conversion

mlcommons · Jul 13, 2024 · c5db738 · c5db738
1 parent f1820bb
commit c5db738
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/trace_link/trace_linker.py b/src/trace_link/trace_linker.py
@@ -440,7 +440,9 @@ def load_sync_dependencies(
                 that have synchronization dependencies.
         """
         sync_dependencies = {}
-        trace_analysis = TraceAnalysis(trace_dir=os.path.dirname(kineto_file))
+        absolute_kineto_file = os.path.abspath(kineto_file)
+        trace_dir = os.path.dirname(absolute_kineto_file)
+        trace_analysis = TraceAnalysis(trace_dir=trace_dir)
         cp_graph, success = trace_analysis.critical_path_analysis(
             rank=rank, annotation=annotation, instance_id=instance_id
         )