Skip to content

Commit

Permalink
some bug fixes (#4409)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #4409

Some bug fixes for to_edge_transform_and_lower

Reviewed By: tarun292

Differential Revision: D60153719

fbshipit-source-id: 4b890c9a3cab26fce22ca64ddbc80dae68a57c35
  • Loading branch information
mcr229 authored and facebook-github-bot committed Jul 25, 2024
1 parent dbf7d6e commit 5b0700b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions exir/program/_program.py
Original file line number Diff line number Diff line change
Expand Up @@ -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())

Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 5b0700b

Please sign in to comment.