Fix the equality check of two floating point values

PiperOrigin-RevId: 266022088

1 files changed, 5 insertions, 3 deletions

diff --git a/mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp b/mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp
index 2667da98242..eeb69b77102 100644
--- a/mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp
+++ b/mlir/lib/Dialect/QuantOps/Utils/FakeQuantSupport.cpp
@@ -71,10 +71,12 @@ mlir::quant::fakeQuantAttrsToType(Location loc, unsigned numBits, double rmin,
             nullptr);
   }
 
-  // Special case where min/max is a point. Must be 0.
-  if (rmin == rmax) {
+  // Special case where min/max is close enough. The tensor contents are all
+  // 0.0s, so the scale is set to 1.0 and the tensor can be quantized to zero
+  // points and dequantized to 0.0.
+  if (std::fabs(rmax - rmin) < std::numeric_limits<double>::epsilon()) {
     return UniformQuantizedType::getChecked(flags, storageType, expressedType,
-                                            0.0, 0, qmin, qmax, loc);
+                                            1.0, 0, qmin, qmax, loc);
   }
 
   // Determine the scale.