make et program serialization in bp in line with et program manager

Summary: This change ensures that the serialization of bundled programs is consistent with the serialization of programs in the `et` program manager.

Differential Revision: D69395142

Author: Gasoonjia

devtools/bundled_program/core.py

@@ -92,14 +92,6 @@ def serialize_to_schema(self) -> bp_schema.BundledProgram:
         if self._bundled_program_in_schema is not None:
             return self._bundled_program_in_schema
 
-        if self.executorch_program:
-            program = self._extract_program(self.executorch_program)
-        else:
-            assert self._pte_file_path is not None
-            with open(self._pte_file_path, "rb") as f:
-                p_bytes = f.read()
-            program = _deserialize_pte_binary(p_bytes)
-
         bundled_method_test_suites: List[bp_schema.BundledMethodTestSuite] = []
 
         # Emit data and metadata of bundled tensor
@@ -149,8 +141,13 @@ def serialize_to_schema(self) -> bp_schema.BundledProgram:
                 )
             )
 
-        # TODO(T181463742): avoid calling bytes(..) which may incur large copies.
-        program_bytes: bytes = bytes(_serialize_pte_binary(program))
+        if self.executorch_program:
+            program_bytes = self.executorch_program.buffer
+        else:
+            assert self._pte_file_path is not None
+            with open(self._pte_file_path, "rb") as f:
+                program_bytes = f.read()
+
         self._bundled_program_in_schema = bp_schema.BundledProgram(
             version=BUNDLED_PROGRAM_SCHEMA_VERSION,
             method_test_suites=bundled_method_test_suites,

devtools/bundled_program/test/test_bundle_data.py

@@ -115,7 +115,7 @@ def test_bundled_program_from_pte(self) -> None:
 
             self.assertEqual(
                 bundled_program.serialize_to_schema().program,
-                bytes(_serialize_pte_binary(executorch_program.executorch_program)),
+                executorch_program.buffer,
             )
 
     def test_bundled_miss_methods(self) -> None: