From 759bcd4b4b6f70fcec045b24fb6874aaca989549 Mon Sep 17 00:00:00 2001
From: Aldy Hernandez <aldyh@redhat.com>
Date: Sun, 13 Nov 2022 18:39:59 +0100
Subject: [PATCH] [range-ops] Implement sqrt.

gcc/ChangeLog:

	* fold-const-call.cc (do_mpfr_arg1): Remove static.
	* gimple-range-op.cc (class cfn_sqrt): New.
	(gimple_range_op_handler::maybe_builtin_call): Add sqrt case.
	* realmpfr.h (do_mpfr_arg1): Add extern.

gcc/testsuite/ChangeLog:

	* gcc.dg/tree-ssa/vrp124.c: New test.
---
 gcc/fold-const-call.cc                 |  2 +-
 gcc/gimple-range-op.cc                 | 56 ++++++++++++++++++++++++++
 gcc/realmpfr.h                         |  4 ++
 gcc/testsuite/gcc.dg/tree-ssa/vrp124.c | 16 ++++++++
 4 files changed, 77 insertions(+), 1 deletion(-)
 create mode 100644 gcc/testsuite/gcc.dg/tree-ssa/vrp124.c

diff --git a/gcc/fold-const-call.cc b/gcc/fold-const-call.cc
index 8ceed8f02f9..5c6b852acdc 100644
--- a/gcc/fold-const-call.cc
+++ b/gcc/fold-const-call.cc
@@ -118,7 +118,7 @@ do_mpfr_ckconv (real_value *result, mpfr_srcptr m, bool inexact,
    in format FORMAT, given that FUNC is the MPFR implementation of f.
    Return true on success.  */
 
-static bool
+bool
 do_mpfr_arg1 (real_value *result,
 	      int (*func) (mpfr_ptr, mpfr_srcptr, mpfr_rnd_t),
 	      const real_value *arg, const real_format *format)
diff --git a/gcc/gimple-range-op.cc b/gcc/gimple-range-op.cc
index 7764166d5fb..f1f2f098305 100644
--- a/gcc/gimple-range-op.cc
+++ b/gcc/gimple-range-op.cc
@@ -43,6 +43,8 @@ along with GCC; see the file COPYING3.  If not see
 #include "range.h"
 #include "value-query.h"
 #include "gimple-range.h"
+#include "fold-const-call.h"
+#include "realmpfr.h"
 
 // Given stmt S, fill VEC, up to VEC_SIZE elements, with relevant ssa-names
 // on the statement.  For efficiency, it is an error to not pass in enough
@@ -301,6 +303,54 @@ public:
   }
 } op_cfn_constant_p;
 
+// Implement range operator for SQRT.
+class cfn_sqrt : public range_operator_float
+{
+  using range_operator_float::fold_range;
+private:
+  void rv_fold (REAL_VALUE_TYPE &lb, REAL_VALUE_TYPE &ub, bool &maybe_nan,
+		tree type,
+		const REAL_VALUE_TYPE &lh_lb,
+		const REAL_VALUE_TYPE &lh_ub,
+		const REAL_VALUE_TYPE &,
+		const REAL_VALUE_TYPE &,
+		relation_kind) const final override
+  {
+    // All negatives.
+    if (real_compare (LT_EXPR, &lh_ub, &dconst0))
+      {
+	real_nan (&lb, "", 0, TYPE_MODE (type));
+	ub = lb;
+	maybe_nan = true;
+	return;
+      }
+    const real_format *format = REAL_MODE_FORMAT (TYPE_MODE (type));
+    // All positives.
+    if (real_compare (GE_EXPR, &lh_lb, &dconst0))
+      {
+	// ?? Handle +INF manually since do_mpfr_arg1 does not.
+	if (real_isinf (&lh_lb, 0))
+	  lb = lh_lb;
+	else if (!do_mpfr_arg1 (&lb, mpfr_sqrt, &lh_lb, format))
+	  {
+	    lb = dconst0;
+	    lb.sign = 1;
+	  }
+	maybe_nan = false;
+      }
+    // Both positives and negatives.
+    else
+      {
+	// Range is [-0.0, sqrt(lh_ub)] +-NAN.
+	lb = dconst0;
+	lb.sign = 1;
+	maybe_nan = true;
+      }
+    if (!do_mpfr_arg1 (&ub, mpfr_sqrt, &lh_ub, format))
+      ub = dconstinf;
+  }
+} fop_cfn_sqrt;
+
 // Implement range operator for CFN_BUILT_IN_SIGNBIT.
 class cfn_signbit : public range_operator_float
 {
@@ -907,6 +957,12 @@ gimple_range_op_handler::maybe_builtin_call ()
       m_int = &op_cfn_parity;
       break;
 
+    CASE_CFN_SQRT_ALL:
+      m_valid = true;
+      m_op1 = gimple_call_arg (call, 0);
+      m_float = &fop_cfn_sqrt;
+      break;
+
     default:
       break;
     }
diff --git a/gcc/realmpfr.h b/gcc/realmpfr.h
index edc08385fe8..807dd2308d2 100644
--- a/gcc/realmpfr.h
+++ b/gcc/realmpfr.h
@@ -32,4 +32,8 @@ extern void real_from_mpfr (REAL_VALUE_TYPE *, mpfr_srcptr,
 			    const real_format *, mpfr_rnd_t);
 extern void mpfr_from_real (mpfr_ptr, const REAL_VALUE_TYPE *, mpfr_rnd_t);
 
+extern bool do_mpfr_arg1 (real_value *result,
+			  int (*func) (mpfr_ptr, mpfr_srcptr, mpfr_rnd_t),
+			  const real_value *arg, const real_format *format);
+
 #endif /* ! GCC_REALGMP_H */
diff --git a/gcc/testsuite/gcc.dg/tree-ssa/vrp124.c b/gcc/testsuite/gcc.dg/tree-ssa/vrp124.c
new file mode 100644
index 00000000000..ef72d660153
--- /dev/null
+++ b/gcc/testsuite/gcc.dg/tree-ssa/vrp124.c
@@ -0,0 +1,16 @@
+// { dg-do compile }
+// { dg-options "-O2 -fdump-tree-evrp -fdisable-tree-ethread" }
+
+void link_error ();
+
+void foo (float f)
+{
+  float z = __builtin_sqrt (f);
+  if (!__builtin_isnan (z))
+    {
+      if (z < 0.0)
+	link_error ();
+    }
+}
+
+// { dg-final { scan-tree-dump-not "link_error" "evrp" } }
-- 
2.38.1