Skip to content

Commit

Permalink
Merge pull request #69 from mlcommons/detailed-error-msg
Browse files Browse the repository at this point in the history
Print detailed error message on exception in get_start_timestamp_for_gpu_op
  • Loading branch information
srinivas212 authored May 22, 2024
2 parents a0834da + 61e3142 commit 6ed23d2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/trace_link/trace_linker.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 6ed23d2

Please sign in to comment.