We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2043e22 commit 89af1e0Copy full SHA for 89af1e0
extension/llm/export/partitioner_lib.py
@@ -191,5 +191,9 @@ def get_qnn_partitioner(
191
),
192
skip_node_id_set={},
193
skip_node_op_set=skip_node_op_set,
194
- skip_mutable_buffer=False,
+ # TODO: For fp flow, the behavior of mutable buffer is unexpected
195
+ # It seems that delegated mutable buffer is not removed from the output.
196
+ # When I trace back, I found the mutable buffer doesn't exist in original_program.state_dict.
197
+ # So, it doesn't be added into output_specs_to_delete.
198
+ skip_mutable_buffer=use_fp16,
199
)
0 commit comments