File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 3333 ExecutorchProgramManager ,
3434 to_edge ,
3535)
36+ from executorch .exir .dialects ._ops import ops as exir_ops
3637from executorch .exir .pass_base import PassResult
3738from executorch .exir .passes import ToOutVarPass
3839from executorch .exir .passes .sym_shape_eval_pass import HintBasedSymShapeEvalPass
@@ -185,14 +186,17 @@ def export_to_edge(
185186 edge_prog_manager = to_edge (
186187 expo_program ,
187188 compile_config = EdgeCompileConfig (
188- _skip_dim_order = True ,
189189 # Allow specific non-core aten ops in the IR.
190190 _core_aten_ops_exception_list = [
191191 torch .ops .aten ._native_batch_norm_legit_functional .default ,
192192 torch .ops .aten .linear .default ,
193193 torch .ops .aten .linalg_vector_norm .default ,
194194 torch .ops .aten .unfold .default ,
195195 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 ,
196200 ],
197201 ),
198202 )
You can’t perform that action at this time.
0 commit comments