Skip to content

Commit b003ac8

Browse files
authored
Skip dim order when exporting MV3 to Core ML (#23)
1 parent 58c9f70 commit b003ac8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mv3/python/export.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from executorch.backends.apple.coreml.partition import CoreMLPartitioner
1111
from executorch.exir import to_edge
1212
from 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

1616
def 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
#

0 commit comments

Comments
 (0)