Commit b508b4ba authored by Hans Wennborg's avatar Hans Wennborg
Browse files

Merging r371221 and r371224:

------------------------------------------------------------------------
r371221 | spatel | 2019-09-06 18:10:18 +0200 (Fri, 06 Sep 2019) | 3 lines

[SimplifyLibCalls] handle pow(x,-0.0) before it can assert (PR43233)

https://bugs.llvm.org/show_bug.cgi?id=43233
------------------------------------------------------------------------

------------------------------------------------------------------------
r371224 | jfb | 2019-09-06 18:26:59 +0200 (Fri, 06 Sep 2019) | 39 lines

[InstCombine] pow(x, +/- 0.0) -> 1.0

Summary:
This isn't an important optimization at all... We're already doing:
  pow(x, 0.0) -> 1.0
My patch merely teaches instcombine that -0.0 does the same.

However, doing this fixes an AMAZING bug! Compile this program:

  extern "C" double pow(double, double);
  double boom(double base) {
    return pow(base, -0.0);
  }

With:
  clang++ ~/Desktop/fast-math.cpp -ffast-math -O2 -S

And clang will crash with a signal. Wow, fast math is so fast it ICEs the
compiler! Arguably, the generated math is infinitely fast.

What's actually happening is that we recurse infinitely in getPow. In debug we
hit its assertion:
  assert(Exp != 0 && "Incorrect exponent 0 not handled");

We avoid this entire mess if we instead recognize that an exponent of positive
and negative zero yield 1.0.

A separate commit, r371221, fixed the same problem. This only contains the added
tests.

<rdar://problem/54598300>

Reviewers: scanon

Subscribers: hiraditya, jkorous, dexonsmith, ributzka, llvm-commits

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D67248
------------------------------------------------------------------------

llvm-svn: 371381
parent 1c21c197
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -1480,8 +1480,8 @@ Value *LibCallSimplifier::optimizePow(CallInst *Pow, IRBuilder<> &B) {
  if (match(Expo, m_SpecificFP(-1.0)))
    return B.CreateFDiv(ConstantFP::get(Ty, 1.0), Base, "reciprocal");

  // pow(x, 0.0) -> 1.0
  if (match(Expo, m_SpecificFP(0.0)))
  // pow(x, +/-0.0) -> 1.0
  if (match(Expo, m_AnyZeroFP()))
    return ConstantFP::get(Ty, 1.0);

  // pow(x, 1.0) -> x
+60 −0
Original line number Diff line number Diff line
; RUN: opt < %s -instcombine -S | FileCheck %s

; CHECK-LABEL: @zero(
; CHECK-NEXT:  ret double 1.000000e+00
define double @zero(double %value) {
  %res = call double @llvm.pow.f64(double %value, double 0.000000e+00)
  ret double %res
}

; CHECK-LABEL: @minus_zero(
; CHECK-NEXT:  ret double 1.000000e+00
define double @minus_zero(double %value) {
  %res = call double @llvm.pow.f64(double %value, double -0.000000e+00)
  ret double %res
}

; CHECK-LABEL: @fast_zero(
; CHECK-NEXT:  ret double 1.000000e+00
define double @fast_zero(double %value) {
  %res = call fast double @llvm.pow.f64(double %value, double 0.000000e+00)
  ret double %res
}

; CHECK-LABEL: @fast_minus_zero(
; CHECK-NEXT:  ret double 1.000000e+00
define double @fast_minus_zero(double %value) {
  %res = call fast double @llvm.pow.f64(double %value, double -0.000000e+00)
  ret double %res
}

; CHECK-LABEL: @vec_zero(
; CHECK-NEXT:  ret <2 x double> <double 1.000000e+00, double 1.000000e+00>
define <2 x double> @vec_zero(<2 x double> %value) {
  %res = call <2 x double> @llvm.pow.v2f64(<2 x double> %value, <2 x double> <double 0.000000e+00, double 0.000000e+00>)
  ret <2 x double> %res
}

; CHECK-LABEL: @vec_minus_zero(
; CHECK-NEXT:  ret <2 x double> <double 1.000000e+00, double 1.000000e+00>
define <2 x double> @vec_minus_zero(<2 x double> %value) {
  %res = call <2 x double> @llvm.pow.v2f64(<2 x double> %value, <2 x double> <double -0.000000e+00, double -0.000000e+00>)
  ret <2 x double> %res
}

; CHECK-LABEL: @vec_fast_zero(
; CHECK-NEXT:  ret <2 x double> <double 1.000000e+00, double 1.000000e+00>
define <2 x double> @vec_fast_zero(<2 x double> %value) {
  %res = call fast <2 x double> @llvm.pow.v2f64(<2 x double> %value, <2 x double> <double 0.000000e+00, double 0.000000e+00>)
  ret <2 x double> %res
}

; CHECK-LABEL: @vec_fast_minus_zero(
; CHECK-NEXT:  ret <2 x double> <double 1.000000e+00, double 1.000000e+00>
define <2 x double> @vec_fast_minus_zero(<2 x double> %value) {
  %res = call fast <2 x double> @llvm.pow.v2f64(<2 x double> %value, <2 x double> <double -0.000000e+00, double -0.000000e+00>)
  ret <2 x double> %res
}

declare double @llvm.pow.f64(double, double)
declare <2 x double> @llvm.pow.v2f64(<2 x double>, <2 x double>)
+10 −0
Original line number Diff line number Diff line
@@ -223,3 +223,13 @@ define <4 x float> @test_simplify_3_5(<4 x float> %x) {
  ret <4 x float> %1
}

; Make sure that -0.0 exponent is always simplified.

define double @PR43233(double %x) {
; CHECK-LABEL: @PR43233(
; CHECK-NEXT:    ret double 1.000000e+00
;
  %r = call fast double @llvm.pow.f64(double %x, double -0.0)
  ret double %r
}