From 6ed6e6647de50f7ac4dd3d2f16a7ee0b4deb1266 Mon Sep 17 00:00:00 2001 From: Alex Denisov Date: Mon, 11 Nov 2024 12:06:04 +0100 Subject: [PATCH] Specify the kineto filepath explicitly when running HTA analysis Without specifying the kineto filepath explicitly, HTA may pick arbitrary files from the `trace_dir` and either provide incorrect analysis results, or fail in some weird ways. --- src/trace_link/trace_linker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 )