Repeated Stage Matching: teaching example

Python starts when you choose Run.

Repeated stage matching

This teaching model isolates the three matching cases in the figure. The production loader also validates ordered entries, handles unknown or misplaced stages, and works with the runtime checker. Its linked source and tests are the authority.

Sources:

https://github.com/torch-spyre/aiu-trace-analyzer/pull/98/files
https://github.com/torch-spyre/aiu-trace-analyzer/blob/0b4a5e538fc19e03cb9fdc6c6bed57a04e03ed9f/src/aiu_trace_analyzer/core/stage_profile.py

# A small teaching model of the three displayed cases, not the full loader.
def repeated_stage_flags(occurrences, key=None, complete=None):
    if complete is not None:
        if len(complete) != occurrences:
            raise ValueError("Complete profile must supply every displayed occurrence")
        return list(complete)
    if key == "pipeline_barrier":
        return [True] * occurrences
    prefix = "pipeline_barrier#"
    if not isinstance(key, str) or not key.startswith(prefix):
        raise ValueError("Use pipeline_barrier or pipeline_barrier#N")
    ordinal = int(key[len(prefix):])
    if not 1 <= ordinal <= occurrences:
        raise ValueError("Occurrence is out of range")
    return [i == ordinal for i in range(1, occurrences + 1)]

assert repeated_stage_flags(4, "pipeline_barrier") == [True] * 4
assert repeated_stage_flags(4, "pipeline_barrier#2") == [False, True, False, False]
assert repeated_stage_flags(4, complete=[False, True, True, True]) == [False, True, True, True]
for label, flags in [
    ("bare", repeated_stage_flags(4, "pipeline_barrier")),
    ("#2", repeated_stage_flags(4, "pipeline_barrier#2")),
    ("complete", repeated_stage_flags(4, complete=[False, True, True, True])),
]:
    print(label, flags)

bare [True, True, True, True]
#2 [False, True, False, False]
complete [False, True, True, True]

Python runs locally in this page. Saved results and figures are visible before execution.

Download full experiment notebook