commit 4a4f35f32411abb73360247e53d59e1fea1ca7c8
author Sanjay Patel <spatel@rotateright.com> Mon Feb 12 19:39:21 2018 +0000
committer Sanjay Patel <spatel@rotateright.com> Mon Feb 12 19:39:21 2018 +0000
tree 94cff55bc16e12f81486d7a9cca0e4315fcf2c13
parent ee4257f676b81681c38cbb1ef7f3fed0ad6af4ce

[InstCombine] X / (X * Y) --> 1.0 / Y

This is similar to the instsimplify fold added with D42385 
( rL323716 )
...but this can't be in instsimplify because we're creating/morphing
a different instruction.

llvm-svn: 324927

llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp

diff --git a/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp b/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
index 3cd2403..b1cad6b 100644
--- a/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
+++ b/llvm/lib/Transforms/InstCombine/InstCombineMulDivRem.cpp
@@ -1523,6 +1523,16 @@
     return &I;
   }
 
+  // X / (X * Y) --> 1.0 / Y
+  // Reassociate to (X / X -> 1.0) is legal when NaNs are not allowed.
+  // We can ignore the possibility that X is infinity because INF/INF is NaN.
+  if (I.hasNoNaNs() && I.hasAllowReassoc() &&
+      match(Op1, m_c_FMul(m_Specific(Op0), m_Value(Y)))) {
+    I.setOperand(0, ConstantFP::get(I.getType(), 1.0));
+    I.setOperand(1, Y);
+    return &I;
+  }
+
   return nullptr;
 }
 

llvm/test/Transforms/InstCombine/fdiv.ll

diff --git a/llvm/test/Transforms/InstCombine/fdiv.ll b/llvm/test/Transforms/InstCombine/fdiv.ll
index 273a49c..b94cfcb 100644
--- a/llvm/test/Transforms/InstCombine/fdiv.ll
+++ b/llvm/test/Transforms/InstCombine/fdiv.ll
@@ -83,13 +83,11 @@
   ret float %div
 }
 
-; FIXME: 
 ; X / (X * Y) --> 1.0 / Y
 
 define float @div_factor(float %x, float %y) {
 ; CHECK-LABEL: @div_factor(
-; CHECK-NEXT:    [[M:%.*]] = fmul float [[X:%.*]], [[Y:%.*]]
-; CHECK-NEXT:    [[D:%.*]] = fdiv reassoc nnan float [[X]], [[M]]
+; CHECK-NEXT:    [[D:%.*]] = fdiv reassoc nnan float 1.000000e+00, [[Y:%.*]]
 ; CHECK-NEXT:    ret float [[D]]
 ;
   %m = fmul float %x, %y
@@ -110,14 +108,12 @@
   ret float %d
 }
 
-; FIXME:
 ; Commute, verify vector types, and show that we are not dropping extra FMF.
 ; X / (Y * X) --> 1.0 / Y
 
 define <2 x float> @div_factor_commute(<2 x float> %x, <2 x float> %y) {
 ; CHECK-LABEL: @div_factor_commute(
-; CHECK-NEXT:    [[M:%.*]] = fmul <2 x float> [[Y:%.*]], [[X:%.*]]
-; CHECK-NEXT:    [[D:%.*]] = fdiv reassoc nnan ninf nsz <2 x float> [[X]], [[M]]
+; CHECK-NEXT:    [[D:%.*]] = fdiv reassoc nnan ninf nsz <2 x float> <float 1.000000e+00, float 1.000000e+00>, [[Y:%.*]]
 ; CHECK-NEXT:    ret <2 x float> [[D]]
 ;
   %m = fmul <2 x float> %y, %x