Skip to content

Commit f9db02a

Browse files
zingofacebook-github-bot
authored andcommitted
Fix backends/arm test after _skip_dim_order update (#3529)
Summary: This fixes the backends/arm unit tests Pull Request resolved: #3529 Reviewed By: larryliu0820 Differential Revision: D57054106 Pulled By: digantdesai fbshipit-source-id: cec2b14e85bb445114533c2a6cd6702acf2ded6e
1 parent 251aa74 commit f9db02a

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

backends/arm/test/tester/arm_tester.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,16 @@ def quantize(self, quantize_stage: Optional[Quantize] = None):
178178
)
179179
return super().quantize(quantize_stage)
180180

181-
def to_edge(self, to_edge_stage: Optional[ToEdge] = None):
181+
def to_edge(
182+
self,
183+
to_edge_stage: Optional[ToEdge] = None,
184+
config: Optional[EdgeCompileConfig] = None,
185+
):
182186
if to_edge_stage is None:
183-
to_edge_stage = ToEdge(EdgeCompileConfig(_check_ir_validity=False))
187+
to_edge_stage = ToEdge(config)
188+
else:
189+
if config is not None:
190+
to_edge_stage.edge_compile_conf = config
184191

185192
# TODO(T182928844): Delegate dim order op to backend.
186193
to_edge_stage.edge_compile_conf._skip_dim_order = True

0 commit comments

Comments
 (0)