[flang] Handle subnormals while folding SCALE, SET_EXPONENT, & FRACTION

[klausler]

Summary

The implementations of folding for the intrinsic functions SCALE and SET_EXPONENT
do not cope correctly with numbers in the subnormal range. Fix SCALE,
then modify SET_EXPONENT to be a special case of SCALE.

flang/include/flang/Evaluate/real.h

public:
   Real SPACING() const;
   Real SET_EXPONENT(int) const;
   Real FRACTION() const;
 
   // SCALE(); also known as IEEE_SCALB and (in IEEE-754 '08) ScaleB.
   template <typename INT>
   ValueWithRealFlags<Real> SCALE(const INT &by,
       Rounding rounding = TargetCharacteristics::defaultRounding) const {
-    auto expo{exponentBias + by.ToInt64()};
+    // Normalize a fraction with just its LSB set and then multiply.
+    // (Set the LSB, not the MSB, in case the scale factor needs to
+    //  be subnormal.)
+    auto adjust{exponentBias + binaryPrecision - 1};
+    auto expo{adjust + by.ToInt64()};
     if (IsZero()) {
       expo = exponentBias; // ignore by, don't overflow
     } else if (by > INT{maxExponent}) {
-      expo = maxExponent;
-    } else if (by < INT{-exponentBias}) {
+      expo = maxExponent + binaryPrecision - 1;
+    } else if (by < INT{-adjust}) {
       expo = -1;
     }
     Real twoPow;
     RealFlags flags{
-        twoPow.Normalize(false, static_cast<int>(expo), Fraction::MASKL(1))};
+        twoPow.Normalize(false, static_cast<int>(expo), Fraction::MASKR(1))};
     ValueWithRealFlags<Real> result{Multiply(twoPow, rounding)};
     result.flags |= flags;
     return result;
   }
 
   constexpr Real FlushSubnormalToZero() const {
     if (IsSubnormal()) {
       return Real{};

flang/lib/Evaluate/real.cpp

 Real<W, P> Real<W, P>::SET_EXPONENT(int expo) const {
   if (IsNotANumber()) {
     return *this;
   } else if (IsInfinite()) {
     return NotANumber();
   } else if (IsZero()) {
     return *this;
   } else {
-    Real result;
-    result.Normalize(IsNegative(), expo + exponentBias - 1, GetFraction());
-    return result;
+    return SCALE(Integer<32>(expo - UnbiasedExponent() - 1)).value;
   }
 }
 
 // 16.9.171
 template <typename W, int P> Real<W, P> Real<W, P>::FRACTION() const {
   return SET_EXPONENT(0);
 }
 
 template class Real<Integer<16>, 11>;
 template class Real<Integer<16>, 8>;
 template class Real<Integer<32>, 24>;
 template class Real<Integer<64>, 53>;
 template class Real<Integer<80>, 64>;
 template class Real<Integer<128>, 113>;
 } // namespace Fortran::evaluate::value

flang/test/Evaluate/folding07.f90

module m
   logical, parameter :: test_set_exponent_nan = transfer(set_exponent(nan, 0), 0) == nanInt
   real,    parameter :: set_expo_inf = set_exponent(inf, 0)
   integer, parameter :: set_expo_inf_int = transfer(set_expo_inf, 0)
   logical, parameter :: test_set_exponent_inf = shiftr(set_expo_inf_int, 23) == 255 .and. &
                                                 shiftl(set_expo_inf_int, 9) /= 0 ! NaN
   logical, parameter :: test_set_exponent_0 = set_exponent(1., 0) == 0.5
   logical, parameter :: test_set_exponent_1 = set_exponent(1., 1) == 1.
   logical, parameter :: test_set_exponent_2 = set_exponent(1., 2) == 2.
+  logical, parameter :: test_set_exponent_min = set_exponent(1., -149) == 1.40129846432481707092372958328991613128026194187651577175706828388979108268586060148663818836212158203125e-45_4
 
 end module