File tree 1 file changed +5
-1
lines changed
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 33
33
ExecutorchProgramManager ,
34
34
to_edge ,
35
35
)
36
+ from executorch .exir .dialects ._ops import ops as exir_ops
36
37
from executorch .exir .pass_base import PassResult
37
38
from executorch .exir .passes import ToOutVarPass
38
39
from executorch .exir .passes .sym_shape_eval_pass import HintBasedSymShapeEvalPass
@@ -185,14 +186,17 @@ def export_to_edge(
185
186
edge_prog_manager = to_edge (
186
187
expo_program ,
187
188
compile_config = EdgeCompileConfig (
188
- _skip_dim_order = True ,
189
189
# Allow specific non-core aten ops in the IR.
190
190
_core_aten_ops_exception_list = [
191
191
torch .ops .aten ._native_batch_norm_legit_functional .default ,
192
192
torch .ops .aten .linear .default ,
193
193
torch .ops .aten .linalg_vector_norm .default ,
194
194
torch .ops .aten .unfold .default ,
195
195
torch .ops .aten .angle .default ,
196
+ # cadence replaced to_dim_order_copy with _to_copy for performance
197
+ # skip _to_copy op to get around of dim order check
198
+ # We should remove this op once cadence can support dim order
199
+ exir_ops .edge .aten ._to_copy .default ,
196
200
],
197
201
),
198
202
)
You can’t perform that action at this time.
0 commit comments