Method add_executorch_program

devtools/etrecord/_etrecord.py:349–399

Add executorch program data to the ETRecord after it has been created. This method allows users to add executorch program data they want to record to an existing ETRecord instance. The executorch program data includes debug handle map, delegate map, reference output

(
        self,
        executorch_program: Union[
            ExecutorchProgram,
            ExecutorchProgramManager,
            BundledProgram,
        ],
    )

            _add_module_to_graph_map(graph_map, module_name, export_module)

    def add_executorch_program(
        self,
        executorch_program: Union[
            ExecutorchProgram,
            ExecutorchProgramManager,
            BundledProgram,
        ],
    ) -> None:
        """
        Add executorch program data to the ETRecord after it has been created.

        This method allows users to add executorch program data they want to record
        to an existing ETRecord instance. The executorch program data includes debug handle map,
        delegate map, reference outputs, and representative inputs that will be included
        when the ETRecord is saved.

        Args:
            executorch_program: The ExecuTorch program for this model returned by the call to
                `to_executorch()` or the `BundledProgram` of this model.

        Raises:
            RuntimeError: If executorch program data already exists in the ETRecord.
        """
        # Check if executorch program data already exists
        if (
            self._debug_handle_map is not None
            or self._delegate_map is not None
            or self._instruction_id_to_num_outs_map is not None
            or self._reference_outputs is not None
            or self._representative_inputs is not None
        ):
            raise RuntimeError(
                "Executorch program data already exists in the ETRecord. "
                "Cannot add executorch program data when it already exists."
            )

        # Process executorch program and extract data
        (
            debug_handle_map,
            delegate_map,
            instruction_id_to_num_outs_map,
            reference_outputs,
            representative_inputs,
        ) = _process_executorch_program(executorch_program)

        # Set the extracted data
        self._debug_handle_map = debug_handle_map
        self._delegate_map = delegate_map
        self._instruction_id_to_num_outs_map = instruction_id_to_num_outs_map
        self._reference_outputs = reference_outputs
        self._representative_inputs = representative_inputs

    def add_exported_program(
        self,