Update on "[ExecuTorch][#10375] Add extension.BundledModule to Wrap extension.Module with Bundled Program Logic"

#10375

# 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`.

The issue is that `pybindings.PyModule` is dependent on the `method` getter from `pybindings.Module`, which `extension.Module` do not have. Since we don't want to expose `method` getter in `extension.Module`, we have to protect the getter, wrap the functions that is dependent on it and use the protected getter there, ultimately decouple `pybindings` from a `method` getter.

# Proposal

Now that we have a protected `method` getter, we can introduce a `extension.BundledModule`, a child class inheriting `extension.Module` which wraps up bundled program logic that is dependent on the `method` getter.

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

[ghstack-poisoned]

Author: zhenyan-zhang-meta

extension/module/bundled_module.h

@@ -13,8 +13,6 @@
 namespace executorch {
 namespace extension {
 
-using executorch::extension::ET_MODULE_NAMESPACE::Module;
-
 /**
  * A facade class for loading bundled programs and executing methods within
  * them.

extension/module/test/bundled_module_test.cpp

@@ -19,8 +19,8 @@ class BundledModuleTest : public ::testing::Test {
     if (const char* env = std::getenv("RESOURCES_PATH")) {
       resources_path = env;
     }
+    pte_path_ = std::getenv("ET_MODULE_PTE_PATH");
     bpte_path_ = resources_path + "/bundled_program.bpte";
-    pte_path_ = resources_path + "/add.pte";
   }
 
   static inline std::string bpte_path_;

extension/module/test/targets.bzl

@@ -44,24 +44,25 @@ def define_common_targets(is_fbcode=False):
             )
 
 
-        runtime.cxx_test(
-            name = "bundled_test" + aten_suffix,
-            srcs = [
-                "bundled_module_test.cpp",
-            ],
-            deps = [
-                "//executorch/kernels/portable:generated_lib" + aten_suffix,
-                "//executorch/extension/module:bundled_module" + aten_suffix,
-                "//executorch/extension/tensor:tensor" + aten_suffix,
-            ],
-            env = {
-                "RESOURCES_PATH": "$(location :resources)/resources",
-            },
-            platforms = [CXX, ANDROID],  # Cannot bundle resources on Apple platform.
-            compiler_flags = [
-                "-Wno-error=deprecated-declarations",
-            ],
-        )
+            runtime.cxx_test(
+                name = "bundled_test" + aten_suffix,
+                srcs = [
+                    "bundled_module_test.cpp",
+                ],
+                deps = [
+                    "//executorch/kernels/portable:generated_lib" + aten_suffix,
+                    "//executorch/extension/module:bundled_module" + aten_suffix,
+                    "//executorch/extension/tensor:tensor" + aten_suffix,
+                ],
+                env = {
+                    "RESOURCES_PATH": "$(location :resources)/resources",
+                    "ET_MODULE_PTE_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleAdd.pte])",
+                },
+                platforms = [CXX, ANDROID],  # Cannot bundle resources on Apple platform.
+                compiler_flags = [
+                    "-Wno-error=deprecated-declarations",
+                ],
+            )
 
     runtime.filegroup(
         name = "resources",