Skip to content

Commit 61a7a2d

Browse files
committed
Fix custom annotation with bias node
1 parent fba2d45 commit 61a7a2d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backends/qualcomm/quantizer/custom_annotation.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,10 @@ def annotate_conv2d(node: Node, quantization_config: QuantizationConfig) -> None
211211
weight = node.args[1]
212212
input_qspec_map[weight] = quantization_config.weight
213213

214+
if len(node.args) > 2 and isinstance(node.args[2], Node):
215+
bias = node.args[2]
216+
input_qspec_map[bias] = quantization_config.bias(node)
217+
214218
node.meta[Q_ANNOTATION_KEY] = QuantizationAnnotation(
215219
input_qspec_map=input_qspec_map,
216220
output_qspec=quantization_config.output_activation,

0 commit comments

Comments
 (0)