summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/behaviour.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/lib/behaviour.py b/lib/behaviour.py
index 626d3c2..fed0f6d 100644
--- a/lib/behaviour.py
+++ b/lib/behaviour.py
@@ -18,7 +18,12 @@ class SDKBehaviourModel:
is_loop = dict()
for annotation in annotations:
- am_tt_param_names = sorted(annotation.start.param.keys())
+ # annotation.start.param may be incomplete, for instance in cases
+ # where DPUs are allocated before the input file is loadeed (and
+ # thus before the problem size is known).
+ # Hence, we must use annotation.end.param whenever we deal
+ # with possibly problem size-dependent behaviour.
+ am_tt_param_names = sorted(annotation.end.param.keys())
if annotation.name not in delta_by_name:
delta_by_name[annotation.name] = dict()
delta_param_by_name[annotation.name] = dict()