From c403280f6bf09062c905d5b65539c15894667a3f Mon Sep 17 00:00:00 2001 From: Johnson Zhang Date: Wed, 10 Feb 2021 09:31:09 +0800 Subject: [PATCH] [Relay][Quantization] Fix Bug Which Cause Negative Left Shift Op --- src/relay/quantize/realize.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/relay/quantize/realize.cc b/src/relay/quantize/realize.cc index 2716c6e65f65..d77ede3acbf9 100644 --- a/src/relay/quantize/realize.cc +++ b/src/relay/quantize/realize.cc @@ -165,7 +165,7 @@ Expr QuantizeRealize(const Call& ref_call, const Array& new_args, const Ob MakeConstantScalar(cfg->dtype_activation, static_cast(shift_nbit))); } else { data = LeftShift(data, - MakeConstantScalar(cfg->dtype_activation, static_cast(shift_nbit))); + MakeConstantScalar(cfg->dtype_activation, static_cast(-shift_nbit))); } data = Clip(data, clip_min_imm, clip_max_imm); return QRealizeIntExpr(data, dom_scale, n->dtype);