From b178b9f8c286e184dce8ec014dd475fb187ec788 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Wed, 21 May 2025 13:44:11 +0200 Subject: RunAnnotation: store name --- lib/loader/plain.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'lib/loader/plain.py') diff --git a/lib/loader/plain.py b/lib/loader/plain.py index 502dcdf..488dd2a 100644 --- a/lib/loader/plain.py +++ b/lib/loader/plain.py @@ -87,6 +87,7 @@ class TraceAnnotation: class RunAnnotation: + name = None start = None kernels = list() end = None @@ -106,9 +107,7 @@ class RunAnnotation: return self def __repr__(self): - return ( - f"RunAnnotation" - ) + return f"RunAnnotation<{self.name}, start={self.start}, kernels={self.kernels}, end={self.end}>" class Logfile: @@ -216,9 +215,15 @@ class Logfile: offset=len(observations), name=name_str, param=param ) if trace_start is not None: + assert trace_start.name == trace_end.name + for kernel in trace_kernels: + assert trace_start.name == kernel.name annotations.append( RunAnnotation( - start=trace_start, kernels=trace_kernels, end=trace_end + name=trace_start.name, + start=trace_start, + kernels=trace_kernels, + end=trace_end, ) ) -- cgit v1.2.3