diff --git a/mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp b/mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp --- a/mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp +++ b/mlir/lib/Dialect/Quant/Utils/FakeQuantSupport.cpp @@ -35,6 +35,15 @@ qmin = 0; qmax = 65535; } + } else if (numBits <= 32) { + storageType = IntegerType::get(32, ctx); + if (isSigned) { + qmin = std::numeric_limits::min(); + qmax = std::numeric_limits::max(); + } else { + qmin = std::numeric_limits::min(); + qmax = std::numeric_limits::max(); + } } else { return true; }