Function _restore_delegates

exir/_serialize/_program.py:595–624

Find and replace the Program's references to these segments, inlining the data. Args: program: The Program holding non-inlined delegates. Modified in-place. segments: List of bytes containing the delegate data. Not modified. Returns: The Program with delegates restored.

(program: Program, segments: List[bytes])

def _restore_delegates(program: Program, segments: List[bytes]) -> Program:
    """Find and replace the Program's references to these segments, inlining
    the data.

    Args:
        program: The Program holding non-inlined delegates. Modified in-place.
        segments: List of bytes containing the delegate data. Not modified.

    Returns: The Program with delegates restored.
    """
    for plan_index, plan in enumerate(program.execution_plan):
        for delegate_index, delegate in enumerate(plan.delegates):
            if delegate.processed.location == DataLocation.INLINE:
                continue
            assert delegate.processed.location == DataLocation.SEGMENT
            index = delegate.processed.index
            if index >= len(segments):
                raise ValueError(
                    f"Plan {plan_index} delegate {delegate_index} "
                    + f"segment index {index} >= num segments {len(segments)}"
                )

            data_index: int = len(program.backend_delegate_data)
            program.backend_delegate_data.append(
                BackendDelegateInlineData(data=segments[index])
            )
            delegate.processed = BackendDelegateDataReference(
                location=DataLocation.INLINE, index=data_index
            )
    return program


def _restore_constant_segment(