Makes 4KB alignment of .ptd

nitish2112 · facebook-github-bot · commit 813e26aa1245 · 2025-12-18T11:27:29.000-08:00
Differential Revision: D89237435
diff --git a/exir/program/_program.py b/exir/program/_program.py
@@ -79,6 +79,7 @@
     EXIREdgeDialectVerifier,
     get_aten_verifier,
 )
+from executorch.extension.flat_tensor.serialize.serialize import FlatTensorConfig
 from torch._export.passes import ReplaceViewOpsWithViewCopyOpsPass
 from torch._export.utils import _detect_fake_mode_from_gm
 from torch._export.verifier import Verifier
@@ -595,7 +596,7 @@ def __init__(
             FlatTensorSerializer,
         )
 
-        self._data_serializer: DataSerializer = FlatTensorSerializer()
+        self._data_serializer: DataSerializer = FlatTensorSerializer(FlatTensorConfig(tensor_alignment=4096, segment_alignment=4096))
 
     def _get_emitter_output(self) -> EmitterOutput:
         if self._emitter_output is None:

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,7 @@`
`79`	`79`	`EXIREdgeDialectVerifier,`
`80`	`80`	`get_aten_verifier,`
`81`	`81`	`)`
	`82`	`+from executorch.extension.flat_tensor.serialize.serialize import FlatTensorConfig`
`82`	`83`	`from torch._export.passes import ReplaceViewOpsWithViewCopyOpsPass`
`83`	`84`	`from torch._export.utils import _detect_fake_mode_from_gm`
`84`	`85`	`from torch._export.verifier import Verifier`
`@@ -595,7 +596,7 @@ def __init__(`
`595`	`596`	`FlatTensorSerializer,`
`596`	`597`	`)`
`597`	`598`
`598`		`- self._data_serializer: DataSerializer = FlatTensorSerializer()`
	`599`	`+ self._data_serializer: DataSerializer = FlatTensorSerializer(FlatTensorConfig(tensor_alignment=4096, segment_alignment=4096))`
`599`	`600`
`600`	`601`	`def _get_emitter_output(self) -> EmitterOutput:`
`601`	`602`	`if self._emitter_output is None:`