Function _extract_named_data

exir/_serialize/_program.py:384–419

Modifies the program in-place to add references to the named data segments. Args: program: The program to extract segments from. Modified in-place. segments: A list of buffers to append extracted segments to. Modified in-place. buffers: A list of unique buffers a

(
    program: Program,
    segments: List[AlignedData],
    buffers: Sequence[bytes],
    name_to_data_entry: Dict[str, DataEntry],
)

def _extract_named_data(
    program: Program,
    segments: List[AlignedData],
    buffers: Sequence[bytes],
    name_to_data_entry: Dict[str, DataEntry],
) -> None:
    """Modifies the program in-place to add references to the named data
        segments.

    Args:
        program: The program to extract segments from. Modified in-place.
        segments: A list of buffers to append extracted segments to. Modified in-place.
        buffers: A list of unique buffers and the information required to
            serialize them. Not modified.
        name_to_data_entry: A map from the blob name to DataEntry.
            Not modified.
    """
    if program.named_data is not None and len(program.named_data) > 0:
        raise ValueError("Program already has named data.")

    # Map from buffer_idx to segment_idx.
    segment_index_map: Dict[int, int] = {}

    named_data: List[NamedData] = []
    for name, data_entry in name_to_data_entry.items():
        segment_index = segment_index_map.get(data_entry.buffer_index, None)
        if segment_index is None:
            segment_index = len(segments)
            segment_index_map[data_entry.buffer_index] = segment_index
            segments.append(
                AlignedData(
                    Cord(buffers[data_entry.buffer_index]), data_entry.alignment
                )
            )
        named_data.append(NamedData(key=name, segment_index=segment_index))
    program.named_data = named_data


def serialize_pte_binary(