Skip to content

Commit 7f5e423

Browse files
author
shubhagr-quic
authored
Name change fix in QNN Config Data format file (#435)
Updated generate_data_format_config API to disable original (framework) name sanitization on AIC backend for QNN Compilation path. Signed-off-by: shubhagr-quic <[email protected]>
1 parent 1e5dbe1 commit 7f5e423

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

QEfficient/utils/generate_qnn_network_specialization_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -173,10 +173,10 @@ def generate_data_format_config(
173173

174174
for input in onnx_model.graph.input:
175175
if "past_key" in input.name or "past_value" in input.name:
176-
kv_nodes.append((input.name).replace(".", "_"))
176+
kv_nodes.append(input.name)
177177
for output in onnx_model.graph.output:
178178
if "past_key" in output.name or "past_value" in output.name:
179-
kv_nodes.append((output.name).replace(".", "_"))
179+
kv_nodes.append(output.name)
180180
kv_overrides = {}
181181

182182
kv_overrides["graphs"] = [

0 commit comments

Comments
 (0)