From 61e31428bfd88e85dc9bfac3a0ce20dde38dbc1d Mon Sep 17 00:00:00 2001 From: Taekyung Heo <7621438+TaekyungHeo@users.noreply.github.com> Date: Wed, 22 May 2024 15:02:48 -0700 Subject: [PATCH] Print detailed error messages on exceptions in get_start_timestamp_for_gpu_op --- 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 ded5506e..88d4e308 100644 --- a/src/trace_link/trace_linker.py +++ b/src/trace_link/trace_linker.py @@ -594,7 +594,7 @@ def get_start_timestamp_for_gpu_op(self, kineto_gpu_op: KinetoOperator) -> int: return cpu_launcher_op.timestamp + cpu_launcher_op.inclusive_dur if kineto_gpu_op.external_id in self.kineto_id_arrow_op_map: return self.kineto_id_arrow_op_map[kineto_gpu_op.external_id].timestamp - raise RuntimeError(f"No valid timestamp found for GPU operator: {kineto_gpu_op.name}") + raise RuntimeError(f"No valid timestamp found for GPU operator: {kineto_gpu_op}") def find_closest_op( self, kineto_gpu_op: KinetoOperator, kineto_ops: List[KinetoOperator], ts: int