Skip to content

Commit 59474d2

Browse files
committed
Update comment
1 parent dc58c2c commit 59474d2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/tvm/relay/quantize/_annotate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,8 @@ def add_rewrite(ref_call, new_args, ctx):
193193
# quantize rhs to INPUT field if it is not Constant
194194
rhs_expr = attach_simulated_quantize(rhs_expr, QAnnotateKind.INPUT)
195195
if lhs_kind == QAnnotateKind.ACTIVATION and rhs_kind == QAnnotateKind.ACTIVATION:
196-
rhs_expr = attach_simulated_quantize(rhs_expr, QAnnotateKind.INPUT) # the other branch
196+
# quantize rhs to INPUT field if both lhs and rhs are ACTIVATION
197+
rhs_expr = attach_simulated_quantize(rhs_expr, QAnnotateKind.INPUT)
197198

198199
expr = _forward_op(ref_call, [lhs_expr, rhs_expr])
199200
return QAnnotateExpr(expr, QAnnotateKind.ACTIVATION)

0 commit comments

Comments
 (0)