Method to_backend

exir/program/_program.py:1680–1728

Returns a semantically-equivalent program to the one given as input, but with portions of each program in the EdgeProgramManager targeted for delegation as determined by the partitioner. Args: partitioner: The partitioner can either be a Partitioner subc

(
        self,
        partitioner: Union[Partitioner, Dict[str, Partitioner]],
    )

    @et_logger("to_backend")
    def to_backend(
        self,
        partitioner: Union[Partitioner, Dict[str, Partitioner]],
    ) -> "EdgeProgramManager":
        """
        Returns a semantically-equivalent program to the one given as input,
        but with portions of each program in the EdgeProgramManager targeted
        for delegation as determined by the partitioner.

        Args:
            partitioner: The partitioner can either be a Partitioner subclass instance, or a
                dictionary mapping method names to Partitioner subclass instance. If it is a
                Partitioner subclass, all programs in the given EdgeProgramManager
                will be lowered using the given partitioner. If it is a
                dictionary, only method names specified in the dictionary will be
                lowered with the given partitioner.

                The Partitioner subclass instance is in charge with tagging portions of the
                input program for delegation. A valid partitioner must return PartitionerResult including valid
                partition_tags: Dict[str, DelegationSpec], where each key is a tag
                name and the nodes with same tag will be fused a one subgraph and
                delegated to backend specififed in delegation spec.

        Returns:
            EdgeProgramManager: A copy of the calling EdgeProgramManager with the
            specified subgraphs lowered.
        """
        new_edge_programs: Dict[str, ExportedProgram] = {}
        method_to_partitioner: Dict[str, Partitioner] = {}
        if not isinstance(partitioner, dict):
            method_to_partitioner = {name: partitioner for name in self._edge_programs}
        else:
            method_to_partitioner = partitioner

        method_to_programs_and_partitioners = MethodProgramsPartitionerSpec(
            self._edge_programs,
            method_to_partitioner,
        )

        new_edge_programs = to_backend(method_to_programs_and_partitioners)
        config = EdgeCompileConfig(_check_ir_validity=False)
        epm = EdgeProgramManager(
            new_edge_programs,
            copy.deepcopy(self._config_methods),
            config,
        )

        epm._etrecord = self._etrecord
        return epm

    @et_logger("to_executorch")
    def to_executorch(  # noqa (FLAKE8) C901