Skip to content

Commit aed7e65

Browse files
cccclaifacebook-github-bot
authored andcommitted
forward fix for taking list of parititioners
Summary: forward fix for D55232842 Reviewed By: shoumikhin Differential Revision: D55432094
1 parent 08b644f commit aed7e65

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
@@ -710,17 +710,19 @@ def _export_llama(modelname, args) -> str: # noqa: C901
710710
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
711711
backend_options = generate_htp_compiler_spec(use_fp16=False)
712712
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
713-
partitioner = QnnPartitioner(
714-
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
715-
generate_qnn_executorch_compiler_spec(
716-
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`.
717-
soc_model=QcomChipset.SM8650, # default to SM8650
718-
backend_options=backend_options,
719-
debug=False,
720-
saver=False,
721-
),
722-
skip_node_id_set={},
723-
skip_node_op_set={},
713+
partitioners.append(
714+
QnnPartitioner(
715+
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
716+
generate_qnn_executorch_compiler_spec(
717+
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`.
718+
soc_model=QcomChipset.SM8650, # default to SM8650
719+
backend_options=backend_options,
720+
debug=False,
721+
saver=False,
722+
),
723+
skip_node_id_set={},
724+
skip_node_op_set={},
725+
)
724726
)
725727
# pyre-ignore: Undefined attribute [16]: Module `executorch.backends` has no attribute `qualcomm`
726728
_transform(builder_exported_to_edge.export_program())

0 commit comments

Comments
 (0)