Add directional rounding version to LLVM libc and use that in APFloat.

Author: lntue

libc/src/__support/math/expf.h

@@ -109,6 +109,15 @@ static constexpr float expf(float x) {
   return static_cast<float>(exp_hi * exp_mid * exp_lo);
 }
 
+// Directional rounding version of expf.
+LIBC_INLINE static float expf(float x, int rounding_mode) {
+  int current_rounding_mode = fputil::get_round();
+  fputil::set_round(rounding_mode);
+  float result = expf(x);
+  fputil::set_round(current_rounding_mode);
+  return result;
+}
+
 } // namespace math
 
 } // namespace LIBC_NAMESPACE_DECL

llvm/lib/Support/APFloat.cpp

@@ -5607,28 +5607,27 @@ float APFloat::convertToFloat() const {
 }
 
 #ifdef LLVM_INTEGRATE_LIBC
-APFloat exp(const APFloat &X, RoundingMode rounding_mode) {
+static constexpr int getFEnvRoundingMode(llvm::RoundingMode rm) {
+  switch (rm) {
+  case APFloat::rmTowardPositive:
+    return FE_UPWARD;
+  case APFloat::rmTowardNegative:
+    return FE_DOWNWARD;
+  case APFloat::rmTowardZero:
+    return FE_TOWARDZERO;
+  default:
+    // TODO: fix rmNearestTiesToAway for platform without FE_TONEARESTFROMZERO.
+    return FE_TONEAREST;
+  };
+}
+
+APFloat exp(const APFloat &X,
+            RoundingMode rounding_mode = APFloat::rmNearestTiesToEven) {
   assert((&X.getSemantics() == (const llvm::fltSemantics *)&semIEEEsingle) &&
          "Float semantics is not IEEEsingle");
   if (&X.getSemantics() == (const llvm::fltSemantics *)&semIEEEsingle) {
-    int current_rounding_mode = fegetround();
-    switch (rounding_mode) {
-    case APFloat::rmNearestTiesToEven:
-      fesetround(FE_TONEAREST);
-      break;
-    case APFloat::rmTowardPositive:
-      fesetround(FE_UPWARD);
-      break;
-    case APFloat::rmTowardNegative:
-      fesetround(FE_DOWNWARD);
-      break;
-    case APFloat::rmTowardZero:
-      fesetround(FE_TOWARDZERO);
-      break;
-    default:
-    }
-    float result = LIBC_NAMESPACE::shared::expf(X.convertToFloat());
-    fesetround(current_rounding_mode);
+    float result = LIBC_NAMESPACE::shared::expf(
+        X.convertToFloat(), getFEnvRoundingMode(rounding_mode));
     return APFloat(result);
   }
   llvm_unreachable("Unexpected semantics");