From e1381e129f3a694e80787efd2243c0ab032be8bd Mon Sep 17 00:00:00 2001 From: Lucy Qiu Date: Thu, 23 Jan 2025 18:24:58 -0800 Subject: [PATCH] Fix serialization config (#7909) Summary: T213034039, caused by D66523267 Reviewed By: tarun292 Differential Revision: D68583662 --- exir/program/_program.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/exir/program/_program.py b/exir/program/_program.py index a87400cf7df..86f111f2f98 100644 --- a/exir/program/_program.py +++ b/exir/program/_program.py @@ -524,7 +524,12 @@ def _get_pte_data(self) -> Cord: if self._pte_data is None: self._pte_data, self._tensor_data = serialize_for_executorch( self._get_emitter_output(), - ExecutorchBackendConfig(), + ExecutorchBackendConfig( + extract_delegate_segments=self._extract_delegate_segments, + segment_alignment=self._segment_alignment, + constant_tensor_alignment=self._constant_tensor_alignment, + delegate_alignment=self._delegate_alignment, + ), self._data_serializer, ) assert self._pte_data is not None @@ -1484,7 +1489,7 @@ def __init__( # Serialize emitter output, ready to be written to a file. self._data_serializer = FlatTensorSerializer() self._pte_data, self._tensor_data = serialize_for_executorch( - self._emitter_output, ExecutorchBackendConfig(), self._data_serializer + self._emitter_output, backend_config, self._data_serializer ) self._buffer: Optional[bytes] = None