Skip to content

Commit 7a9c453

Browse files
cccclaifacebook-github-bot
authored andcommitted
forward fix for taking list of parititioners (#2713)
Summary: Pull Request resolved: #2713 forward fix for D55232842 Reviewed By: shoumikhin Differential Revision: D55432094 fbshipit-source-id: 5478b8c347cea140dae2ae3a457574aa7f0b1e5e
1 parent 609f71e commit 7a9c453

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

examples/models/llama2/export_llama_lib.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -717,17 +717,19 @@ def _export_llama(modelname, args) -> str: # noqa: C901
717717
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
718718
backend_options = generate_htp_compiler_spec(use_fp16=False)
719719
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
720-
partitioner = QnnPartitioner(
721-
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
722-
generate_qnn_executorch_compiler_spec(
723-
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`.
724-
soc_model=QcomChipset.SM8650, # default to SM8650
725-
backend_options=backend_options,
726-
debug=False,
727-
saver=False,
728-
),
729-
skip_node_id_set={},
730-
skip_node_op_set={},
720+
partitioners.append(
721+
QnnPartitioner(
722+
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
723+
generate_qnn_executorch_compiler_spec(
724+
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`.
725+
soc_model=QcomChipset.SM8650, # default to SM8650
726+
backend_options=backend_options,
727+
debug=False,
728+
saver=False,
729+
),
730+
skip_node_id_set={},
731+
skip_node_op_set={},
732+
)
731733
)
732734
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
733735
_transform(builder_exported_to_edge.export_program())

0 commit comments

Comments
 (0)