Arm backend: Improve chain handling (#15738)

Improve chain handling in Model Explorer visualise script.

cc @freddan80 @per @zingo @oscarandersson8218 @digantdesai

Co-authored-by: Zingo Andersen <[email protected]>

Author: jmahbs

examples/arm/visualize.py

@@ -95,19 +95,20 @@ def is_end_of_command(qread_offset: int, end_idx: int) -> bool:
 
         qread_offset = 4 * int(event["args"]["qread"])
 
+        while (cmd_index + chain_len <= queue_df_len - 1) and queue_df.iloc[
+            cmd_index + chain_len
+        ]["scheduled_id"] in sub_ops:
+            chain_len += 1
+
         end_idx = cmd_index + chain_len
         if is_end_of_command(qread_offset, end_idx):
             end_ts = int(event["ts"]) - 1
             queue_df.loc[cmd_index, ["duration"]] = [
                 end_ts - start_ts,
             ]
             start_ts = end_ts
-            cmd_index += chain_len
+            cmd_index = end_idx
             chain_len = 1
-            while (cmd_index + chain_len <= queue_df_len - 1) and queue_df.iloc[
-                cmd_index + chain_len
-            ]["scheduled_id"] in sub_ops:
-                chain_len += 1
 
 
 Agg = Union[str, Callable[[pd.Series], Any]]