Skip to content

Commit 6886974

Browse files
authored
Qualcomm AI Engine Direct - Fix the bug in rms_norm builder (#10250)
As the title
1 parent 1edf692 commit 6886974

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/qualcomm/builders/op_rms_norm.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,9 @@ def define_node(
8181
{}, # kwargs
8282
)
8383
if quant_attrs := node.meta.get(QCOM_QUANT_ATTRS):
84+
quant_attrs = quant_attrs.copy()
85+
quant_attrs[QCOM_ZERO_POINT] = 0
8486
bias_node.meta[QCOM_QUANT_ATTRS] = quant_attrs
85-
bias_node.meta[QCOM_QUANT_ATTRS][QCOM_ZERO_POINT] = 0
8687
bias_tensor_wrapper = self.define_tensor(
8788
bias_node,
8889
node,

0 commit comments

Comments
 (0)