Skip to content

Commit 2ef8a0f

Browse files
authored
Fix serialization config
Differential Revision: D68583662 Pull Request resolved: #7909
1 parent 9a0b51c commit 2ef8a0f

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

exir/program/_program.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -524,7 +524,12 @@ def _get_pte_data(self) -> Cord:
524524
if self._pte_data is None:
525525
self._pte_data, self._tensor_data = serialize_for_executorch(
526526
self._get_emitter_output(),
527-
ExecutorchBackendConfig(),
527+
ExecutorchBackendConfig(
528+
extract_delegate_segments=self._extract_delegate_segments,
529+
segment_alignment=self._segment_alignment,
530+
constant_tensor_alignment=self._constant_tensor_alignment,
531+
delegate_alignment=self._delegate_alignment,
532+
),
528533
self._data_serializer,
529534
)
530535
assert self._pte_data is not None
@@ -1484,7 +1489,7 @@ def __init__(
14841489
# Serialize emitter output, ready to be written to a file.
14851490
self._data_serializer = FlatTensorSerializer()
14861491
self._pte_data, self._tensor_data = serialize_for_executorch(
1487-
self._emitter_output, ExecutorchBackendConfig(), self._data_serializer
1492+
self._emitter_output, backend_config, self._data_serializer
14881493
)
14891494
self._buffer: Optional[bytes] = None
14901495

0 commit comments

Comments
 (0)