File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1010from executorch .backends .apple .coreml .partition import CoreMLPartitioner
1111from executorch .exir import to_edge
1212from executorch .backends .xnnpack .partition .xnnpack_partitioner import XnnpackPartitioner
13- from executorch .exir import to_edge_transform_and_lower
13+ from executorch .exir import EdgeCompileConfig , to_edge_transform_and_lower
1414
1515
1616def main () -> None :
@@ -24,9 +24,9 @@ def main() -> None:
2424 et_program_coreml = to_edge_transform_and_lower (
2525 torch .export .export (model , sample_inputs ),
2626 partitioner = [CoreMLPartitioner ()],
27+ compile_config = EdgeCompileConfig (_skip_dim_order = True ),
2728 ).to_executorch ()
2829
29-
3030 # MPS backend doesn't work yet with pip install today.
3131 # Currently, it is just falling back to portable ops instead.
3232 #
You can’t perform that action at this time.
0 commit comments