From 782e961a70ca8f655d6df683451372657535c5a6 Mon Sep 17 00:00:00 2001 From: Jake Stevens Date: Fri, 26 Jun 2026 19:17:18 -0700 Subject: [PATCH] Remove _use_edge_ops=True (#20516) Summary: In preparation of removing _use_edge_ops (see TODO in _config.py for EdgeCompileConfig), removing all explicit usages where =True, since this is already the case by default. Differential Revision: D109702323 --- exir/backend/test/hta_partitioner_demo.py | 2 +- exir/backend/test/test_backends_lifted.py | 8 ++------ exir/backend/test/test_debug_handle_map.py | 4 +--- exir/backend/test/test_lowered_backend_module.py | 12 +++--------- exir/emit/test/test_emit.py | 4 +--- exir/tests/test_passes.py | 4 +--- exir/tests/test_quant_fusion_pass.py | 3 --- 7 files changed, 9 insertions(+), 28 deletions(-) diff --git a/exir/backend/test/hta_partitioner_demo.py b/exir/backend/test/hta_partitioner_demo.py index b83b95622a1..2504c33e434 100644 --- a/exir/backend/test/hta_partitioner_demo.py +++ b/exir/backend/test/hta_partitioner_demo.py @@ -81,7 +81,7 @@ def forward(self, x, y): pattern_sub = ( to_edge( export(SubModule(), (input_x, input_h), strict=True), - compile_config=exir.EdgeCompileConfig(_use_edge_ops=True), + compile_config=exir.EdgeCompileConfig(), ) .exported_program() .graph_module diff --git a/exir/backend/test/test_backends_lifted.py b/exir/backend/test/test_backends_lifted.py index 8a2bdaa77a5..7cdfc8b2ab5 100644 --- a/exir/backend/test/test_backends_lifted.py +++ b/exir/backend/test/test_backends_lifted.py @@ -627,9 +627,7 @@ def forward(self, x_raw, h, c): traced = to_edge( export(composite_m, inputs, strict=True), - compile_config=exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ), + compile_config=exir.EdgeCompileConfig(_check_ir_validity=False), ) program_without_delegates = to_edge( @@ -748,9 +746,7 @@ def forward(self, x_raw, h, c): traced = to_edge( export(composite_m, inputs, strict=True), - compile_config=exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ), + compile_config=exir.EdgeCompileConfig(_check_ir_validity=False), ) program_without_delegates = to_edge( diff --git a/exir/backend/test/test_debug_handle_map.py b/exir/backend/test/test_debug_handle_map.py index c7000f53c60..8df9410ce07 100644 --- a/exir/backend/test/test_debug_handle_map.py +++ b/exir/backend/test/test_debug_handle_map.py @@ -66,9 +66,7 @@ def test_lowered_the_whole_model(self): models.ModelWithUnusedArg(), ] - edge_compile_config = exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ) + edge_compile_config = exir.EdgeCompileConfig(_check_ir_validity=False) for model in module_list: model_inputs = model.get_random_inputs() diff --git a/exir/backend/test/test_lowered_backend_module.py b/exir/backend/test/test_lowered_backend_module.py index 01b8277feb5..19425c2e358 100644 --- a/exir/backend/test/test_lowered_backend_module.py +++ b/exir/backend/test/test_lowered_backend_module.py @@ -78,9 +78,7 @@ def forward(self, x): expected_res = sin_module(*model_inputs) edgeir_m = to_edge( export(sin_module, model_inputs, strict=True), - compile_config=exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ), + compile_config=exir.EdgeCompileConfig(_check_ir_validity=False), ) max_value = model_inputs[0].shape[0] compile_specs = [CompileSpec("max_value", bytes([max_value]))] @@ -117,9 +115,7 @@ def test_emit_lowered_backend_module(self): models.ModelWithUnusedArg(), ] - edge_compile_config = exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ) + edge_compile_config = exir.EdgeCompileConfig(_check_ir_validity=False) for model in module_list: model_inputs = model.get_random_inputs() @@ -170,9 +166,7 @@ def test_emit_nested_lowered_backend_module(self): models.ModelWithUnusedArg(), ] - edge_compile_config = exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ) + edge_compile_config = exir.EdgeCompileConfig(_check_ir_validity=False) for model in module_list: model_inputs = model.get_random_inputs() diff --git a/exir/emit/test/test_emit.py b/exir/emit/test/test_emit.py index 6f2923d1b81..5b7fee7777d 100644 --- a/exir/emit/test/test_emit.py +++ b/exir/emit/test/test_emit.py @@ -2176,9 +2176,7 @@ def forward(self, x): edge_program_manager = exir.to_edge( {"forward1": ep1, "forward2": ep2}, - compile_config=exir.EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ), + compile_config=exir.EdgeCompileConfig(_check_ir_validity=False), ) edge_program_manager = edge_program_manager.to_backend( diff --git a/exir/tests/test_passes.py b/exir/tests/test_passes.py index 20906fe92e9..07a5d71a3aa 100644 --- a/exir/tests/test_passes.py +++ b/exir/tests/test_passes.py @@ -1758,9 +1758,7 @@ def forward(self, query, key, value): # lower to edge dialect edge_prog = to_edge( aten_prog, - compile_config=EdgeCompileConfig( - _check_ir_validity=False, _use_edge_ops=True - ), + compile_config=EdgeCompileConfig(_check_ir_validity=False), ) edge_prog = edge_prog.to_backend(XnnpackPartitioner()) diff --git a/exir/tests/test_quant_fusion_pass.py b/exir/tests/test_quant_fusion_pass.py index 8622fca0bd8..d87c68f7063 100644 --- a/exir/tests/test_quant_fusion_pass.py +++ b/exir/tests/test_quant_fusion_pass.py @@ -300,7 +300,6 @@ def forward(self, indices): m = _convert_to_reference_decomposed_fx(m) compile_config = EdgeCompileConfig( _check_ir_validity=False, - _use_edge_ops=True, ) m = to_edge( export(m, example_inputs, strict=True), compile_config=compile_config @@ -358,7 +357,6 @@ def forward(self, indices): m = _convert_to_reference_decomposed_fx(m) compile_config = EdgeCompileConfig( _check_ir_validity=False, - _use_edge_ops=True, ) m = to_edge( export(m, example_inputs, strict=True), compile_config=compile_config @@ -431,7 +429,6 @@ def _test_embedding_torchao( compile_config = EdgeCompileConfig( _check_ir_validity=False, - _use_edge_ops=True, ) m = to_edge( export(model, example_inputs, strict=True), compile_config=compile_config