Use extension module for pybind (#13858)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: #13717 by
@lucylq
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/lucylq/102/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/lucylq/102/head
Merge bot PR base: https://github.com/pytorch/executorch/tree/main
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/lucylq/102/orig
@diff-train-skip-merge

Co-authored-by: lucylq <[email protected]>

Author: pytorchbot

exir/backend/test/test_backends.py

@@ -319,16 +319,16 @@ def forward(self, x):
         )
 
         buff = exec_prog.buffer
-
+        executorch_module = _load_for_executorch_from_buffer(buff)
         # This line should raise an exception like
         # RuntimeError: failed with error 0x12
-        _load_for_executorch_from_buffer(buff)
+        executorch_module.run_method("forward")
 
     @vary_segments
     def test_backend_with_compiler_out_of_range(self, extract_delegate_segments: bool):
         with self.assertRaisesRegex(
             RuntimeError,
-            "loading method forward failed with error 0x12",
+            "Failed to get method forward, error: 0x12",
         ):
             self.run_model_in_unsupported_backend(
                 extract_delegate_segments=extract_delegate_segments

exir/backend/test/test_backends_lifted.py

@@ -347,15 +347,16 @@ def forward(self, x):
 
         buff = exec_prog.buffer
 
+        executorch_module = _load_for_executorch_from_buffer(buff)
         # This line should raise an exception like
         # RuntimeError: failed with error 0x12
-        _load_for_executorch_from_buffer(buff)
+        executorch_module.run_method("forward")
 
     @vary_segments
     def test_backend_with_compiler_out_of_range(self, extract_delegate_segments: bool):
         with self.assertRaisesRegex(
             RuntimeError,
-            "loading method forward failed with error 0x12",
+            "Failed to get method forward, error: 0x12",
         ):
             self.run_model_in_unsupported_backend(
                 extract_delegate_segments=extract_delegate_segments

exir/backend/test/test_compatibility.py

@@ -62,12 +62,13 @@ def forward(self, x):
             )
         )
 
+        executorch_module = _load_for_executorch_from_buffer(buff)
         # Throw runtime error with error code 0x30, meaning delegate is incompatible.
         with self.assertRaisesRegex(
             RuntimeError,
-            "loading method forward failed with error 0x30",
+            "Failed to get method forward, error: 0x30",
         ):
-            executorch_module = _load_for_executorch_from_buffer(buff)
+            executorch_module.run_method("forward")
 
     def test_compatibility_in_runtime_edge_program_manager(self):
         class SinModule(torch.nn.Module):
@@ -108,9 +109,10 @@ def forward(self, x):
             )
         )
 
+        executorch_module = _load_for_executorch_from_buffer(buff)
         # Throw runtime error with error code 0x30, meaning delegate is incompatible.
         with self.assertRaisesRegex(
             RuntimeError,
-            "loading method forward failed with error 0x30",
+            "Failed to get method forward, error: 0x30",
         ):
-            executorch_module = _load_for_executorch_from_buffer(buff)
+            executorch_module.run_method("forward")