Method run_method

export/export.py:616–657

Run a specific method with the given inputs. Args: method_name: Name of the method to run, defaults to "forward" example_inputs: Optional inputs to use, defaults to the example inputs Returns: The outputs of the method execution

(
        self,
        method_name: str = "forward",
        example_inputs: Optional[Tuple[torch.Tensor, ...]] = None,
    )

        return self._example_inputs[method_name][0]

    def run_method(
        self,
        method_name: str = "forward",
        example_inputs: Optional[Tuple[torch.Tensor, ...]] = None,
    ) -> Sequence[Any]:
        """
        Run a specific method with the given inputs.

        Args:
            method_name: Name of the method to run, defaults to "forward"
            example_inputs: Optional inputs to use, defaults to the example inputs

        Returns:
            The outputs of the method execution

        Raises:
            RuntimeError: If the method cannot be loaded
        """
        # Lazy import to avoid forcing portable_lib dependency at module load time.
        try:
            from executorch.runtime import Runtime, Verification
        except ModuleNotFoundError as e:
            raise ModuleNotFoundError(
                "executorch.runtime is not available. "
                "In OSS: Please ensure executorch is properly installed via pip. "
                "In fbcode: Please add //executorch/runtime:runtime to your dependencies."
            ) from e

        et_runtime = Runtime.get()
        program = et_runtime.load_program(
            self.get_pte_buffer(), verification=Verification.Minimal
        )
        forward = program.load_method(method_name)

        if forward is None:
            raise RuntimeError(
                f"Failed to load method '{method_name}' from the program"
            )
        if example_inputs is None:
            example_inputs = self.get_example_input(method_name)

        return forward.execute(example_inputs)

    def print_delegation_info(self) -> None:
        """