File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments