[Backport-0.7][Quantize] Fix Bug Which Cause Negative Left Shift Op (#…

…7433)
apache · Feb 17, 2021 · 56000bb · 56000bb
1 parent cdb00da
commit 56000bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/relay/quantize/realize.cc b/src/relay/quantize/realize.cc
@@ -165,7 +165,7 @@ Expr QuantizeRealize(const Call& ref_call, const Array<Expr>& new_args, const Ob
                           MakeConstantScalar(cfg->dtype_activation, static_cast<int>(shift_nbit)));
       } else {
         data = LeftShift(data,
-                         MakeConstantScalar(cfg->dtype_activation, static_cast<int>(shift_nbit)));
+                         MakeConstantScalar(cfg->dtype_activation, static_cast<int>(-shift_nbit)));
       }
       data = Clip(data, clip_min_imm, clip_max_imm);
       return QRealizeIntExpr(data, dom_scale, n->dtype);