Skip to content

Commit 580d27f

Browse files
Gasoonjiafacebook-github-bot
authored andcommitted
turn on dim order in cadence test (#7756)
Summary: This diff turns on dim order in cadence test. Also we get around `to_copy` operator in verifier to keep the verifier check enable. Differential Revision: D68246404
1 parent 1bf20e3 commit 580d27f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backends/cadence/aot/compiler.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
ExecutorchProgramManager,
3434
to_edge,
3535
)
36+
from executorch.exir.dialects._ops import ops as exir_ops
3637
from executorch.exir.pass_base import PassResult
3738
from executorch.exir.passes import ToOutVarPass
3839
from 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
)

0 commit comments

Comments
 (0)