From 5b0700bf1091b3dcff8354ae67caef67f620c79e Mon Sep 17 00:00:00 2001 From: Max Ren Date: Thu, 25 Jul 2024 11:07:21 -0700 Subject: [PATCH] some bug fixes (#4409) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/4409 Some bug fixes for to_edge_transform_and_lower Reviewed By: tarun292 Differential Revision: D60153719 fbshipit-source-id: 4b890c9a3cab26fce22ca64ddbc80dae68a57c35 --- exir/program/_program.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/exir/program/_program.py b/exir/program/_program.py index 3f963487fe..6fb9eca46f 100644 --- a/exir/program/_program.py +++ b/exir/program/_program.py @@ -899,7 +899,7 @@ def _gen_edge_manager_for_partitioners( # check on which ops need to be preserved and which ops need to be decomposed # Those which are truly preserved will be replaced with transformed ops ops_set_to_not_decompose_by_program[name] = ( - _replace_aten_ops_with_transformed_ops(name, program, partitioner) + _replace_aten_ops_with_transformed_ops(name, program, partitioner) or [] ) program = program.run_decompositions(_default_decomposition_table()) @@ -982,8 +982,8 @@ def _to_edge_transform_and_lower( if not isinstance(partitioner, dict) and partitioner is not None: partitioner = {"forward": partitioner} - else: - partitioner = {} + elif partitioner is None: + partitioner = {"forward": []} edge_manager = _gen_edge_manager_for_partitioners( partitioner, aten_programs, config, constant_methods