Resolve kineto_file path bug with absolute path conversion

mlcommons · Oct 7, 2024 · 7306c14 · 7306c14
1 parent f7e5ca3
commit 7306c14
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
@@ -115,7 +115,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
         )