Update on "[ExecuTorch][#10447] Extend PyBundledModule with extension.BundledModule"

#10447

# Context

This issue is a step of #9638.

In #9638, we want to have `extension.Module` as the single source of implementation in `pybindings`, which means that `pybindings.PyModule` should use `extension.Module` rather than its own `pybindings.Module`.

# Proposal

Now that we have `extension.BundledModule` ready, we want to test it out by having our existing `PyBundledModule` to extend it, and let `verify_result_with_bundled_expected_output` to use it, so that we can test out the whole thing with https://github.com/pytorch/executorch/blob/fb45e19055a92d2a91a4d4b7008e135232cbb14b/devtools/bundled_program/test/test_end2end.py

Differential Revision: [D73564127](https://our.internmc.facebook.com/intern/diff/D73564127/)

[ghstack-poisoned]

Author: zhenyan-zhang-meta

extension/module/test/resources/gen_bundled_program.py

@@ -10,6 +10,7 @@
 from executorch.exir import to_edge_transform_and_lower
 from torch.export import export, export_for_training
 
+
 # Step 1: ExecuTorch Program Export
 class SampleModel(torch.nn.Module):
     """An example model with multi-methods. Each method has multiple input and single output"""
@@ -27,6 +28,7 @@ def forward(self, x: torch.Tensor, q: torch.Tensor) -> torch.Tensor:
         torch.add(y, q, out=y)
         return y
 
+
 def main() -> None:
     """Sample code to generate bundled program and save it to file. It is the same as in https://pytorch.org/executorch/0.6/bundled-io.html#emit-example"""
     # Inference method name of SampleModel we want to bundle testcases to.
@@ -46,7 +48,6 @@ def main() -> None:
         capture_input,
     )
 
-
     # Emit the traced method into ET Program.
     et_program = to_edge_transform_and_lower(method_graph).to_executorch()
 

extension/pybindings/pybindings.cpp

@@ -95,9 +95,9 @@ using ::executorch::ET_RUNTIME_NAMESPACE::Kernel;
 using ::executorch::ET_RUNTIME_NAMESPACE::Method;
 using ::executorch::ET_RUNTIME_NAMESPACE::Program;
 using ::executorch::extension::BufferDataLoader;
+using ::executorch::extension::BundledModule;
 using ::executorch::extension::MallocMemoryAllocator;
 using ::executorch::extension::MmapDataLoader;
-using ::executorch::extension::BundledModule;
 using ::executorch::runtime::ArrayRef;
 using ::executorch::runtime::DataLoader;
 using ::executorch::runtime::Error;