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:
524
524
if self ._pte_data is None :
525
525
self ._pte_data , self ._tensor_data = serialize_for_executorch (
526
526
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
+ ),
528
533
self ._data_serializer ,
529
534
)
530
535
assert self ._pte_data is not None
@@ -1484,7 +1489,7 @@ def __init__(
1484
1489
# Serialize emitter output, ready to be written to a file.
1485
1490
self ._data_serializer = FlatTensorSerializer ()
1486
1491
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
1488
1493
)
1489
1494
self ._buffer : Optional [bytes ] = None
1490
1495
You can’t perform that action at this time.
0 commit comments