Skip to content

Commit d5e0f7e

Browse files
Varun Purifacebook-github-bot
authored andcommitted
Write program_buffer during ETRecord Generation
Summary: D50129349 removed writing the program_buffer during ETRecord generation, causing tests looking for a buffer to fail (namely AIBench API tests). Inaccurate bisect blamed Yanan's diff. T166884022 Add this back. Differential Revision: D50400104
1 parent a1187c6 commit d5e0f7e

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

sdk/etrecord/_etrecord.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,13 @@ def generate_etrecord(
163163
)
164164
_handle_export_module(etrecord_zip, export_module, module_name)
165165

166+
if isinstance(
167+
executorch_program, (ExecutorchProgram, ExecutorchProgramManager)
168+
):
169+
etrecord_zip.writestr(
170+
ETRecordReservedFileNames.PROGRAM_BUFFER, executorch_program.buffer
171+
)
172+
166173
if isinstance(
167174
edge_dialect_program,
168175
(EdgeProgramManager, exir.program._program.EdgeProgramManager),

0 commit comments

Comments
 (0)