[llvm] fbca27b - [InstCombine] add tests for fdiv of exp/exp2; NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2021-02-20T16:02:58-05:00
New Revision: fbca27bf2967aff047bb6c2842e0de344b67cf7b

URL: https://github.com/llvm/llvm-project/commit/fbca27bf2967aff047bb6c2842e0de344b67cf7b
DIFF: https://github.com/llvm/llvm-project/commit/fbca27bf2967aff047bb6c2842e0de344b67cf7b.diff

LOG: [InstCombine] add tests for fdiv of exp/exp2; NFC

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/fdiv.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/fdiv.ll b/llvm/test/Transforms/InstCombine/fdiv.ll
index 52bf746dd369..3a7e87a0ef6d 100644
--- a/llvm/test/Transforms/InstCombine/fdiv.ll
+++ b/llvm/test/Transforms/InstCombine/fdiv.ll
@@ -4,6 +4,10 @@
 declare float @llvm.fabs.f32(float) nounwind readnone
 declare float @llvm.pow.f32(float, float) nounwind readnone
 declare <2 x half> @llvm.pow.v2f16(<2 x half>, <2 x half>) nounwind readnone
+declare float @llvm.exp.f32(float) nounwind readnone
+declare <2 x half> @llvm.exp.v2f16(<2 x half>) nounwind readnone
+declare float @llvm.exp2.f32(float) nounwind readnone
+declare <2 x half> @llvm.exp2.v2f16(<2 x half>) nounwind readnone
 
 define float @exact_inverse(float %x) {
 ; CHECK-LABEL: @exact_inverse(
@@ -724,3 +728,133 @@ define <2 x half> @pow_recip(<2 x half> %x, <2 x half> %y) {
   %r = fdiv reassoc arcp ninf <2 x half> <half 1.0, half 1.0>, %p
   ret <2 x half> %r
 }
+
+define float @exp_divisor(float %y, float %z) {
+; CHECK-LABEL: @exp_divisor(
+; CHECK-NEXT:    [[P:%.*]] = call float @llvm.exp.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call float @llvm.exp.f32(float %y)
+  %r = fdiv reassoc arcp float %z, %p
+  ret float %r
+}
+
+; Negative test - don't create an extra exp
+
+define float @exp_divisor_extra_use(float %y, float %z) {
+; CHECK-LABEL: @exp_divisor_extra_use(
+; CHECK-NEXT:    [[P:%.*]] = call float @llvm.exp.f32(float [[Y:%.*]])
+; CHECK-NEXT:    call void @use_f32(float [[P]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call float @llvm.exp.f32(float %y)
+  call void @use_f32(float %p)
+  %r = fdiv reassoc arcp float %z, %p
+  ret float %r
+}
+
+; Negative test - must have reassoc+arcp
+
+define float @exp_divisor_not_enough_fmf(float %y, float %z) {
+; CHECK-LABEL: @exp_divisor_not_enough_fmf(
+; CHECK-NEXT:    [[P:%.*]] = call fast float @llvm.exp.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call fast float @llvm.exp.f32(float %y)
+  %r = fdiv reassoc float %z, %p
+  ret float %r
+}
+
+; Negative test - must have reassoc+arcp
+
+define float @exp_divisor_not_enough_fmf2(float %y, float %z) {
+; CHECK-LABEL: @exp_divisor_not_enough_fmf2(
+; CHECK-NEXT:    [[P:%.*]] = call fast float @llvm.exp.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call fast float @llvm.exp.f32(float %y)
+  %r = fdiv arcp float %z, %p
+  ret float %r
+}
+
+; Special-case - reciprocal does not require extra fmul
+
+define <2 x half> @exp_recip(<2 x half> %x, <2 x half> %y) {
+; CHECK-LABEL: @exp_recip(
+; CHECK-NEXT:    [[P:%.*]] = call <2 x half> @llvm.exp.v2f16(<2 x half> [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc ninf arcp <2 x half> <half 0xH3C00, half 0xH3C00>, [[P]]
+; CHECK-NEXT:    ret <2 x half> [[R]]
+;
+  %p = call <2 x half> @llvm.exp.v2f16(<2 x half> %y)
+  %r = fdiv reassoc arcp ninf <2 x half> <half 1.0, half 1.0>, %p
+  ret <2 x half> %r
+}
+
+define float @exp2_divisor(float %y, float %z) {
+; CHECK-LABEL: @exp2_divisor(
+; CHECK-NEXT:    [[P:%.*]] = call float @llvm.exp2.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call float @llvm.exp2.f32(float %y)
+  %r = fdiv reassoc arcp float %z, %p
+  ret float %r
+}
+
+; Negative test - don't create an extra exp
+
+define float @exp2_divisor_extra_use(float %y, float %z) {
+; CHECK-LABEL: @exp2_divisor_extra_use(
+; CHECK-NEXT:    [[P:%.*]] = call float @llvm.exp2.f32(float [[Y:%.*]])
+; CHECK-NEXT:    call void @use_f32(float [[P]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call float @llvm.exp2.f32(float %y)
+  call void @use_f32(float %p)
+  %r = fdiv reassoc arcp float %z, %p
+  ret float %r
+}
+
+; Negative test - must have reassoc+arcp
+
+define float @exp2_divisor_not_enough_fmf(float %y, float %z) {
+; CHECK-LABEL: @exp2_divisor_not_enough_fmf(
+; CHECK-NEXT:    [[P:%.*]] = call fast float @llvm.exp2.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call fast float @llvm.exp2.f32(float %y)
+  %r = fdiv reassoc float %z, %p
+  ret float %r
+}
+
+; Negative test - must have reassoc+arcp
+
+define float @exp2_divisor_not_enough_fmf2(float %y, float %z) {
+; CHECK-LABEL: @exp2_divisor_not_enough_fmf2(
+; CHECK-NEXT:    [[P:%.*]] = call fast float @llvm.exp2.f32(float [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv arcp float [[Z:%.*]], [[P]]
+; CHECK-NEXT:    ret float [[R]]
+;
+  %p = call fast float @llvm.exp2.f32(float %y)
+  %r = fdiv arcp float %z, %p
+  ret float %r
+}
+
+; Special-case - reciprocal does not require extra fmul
+
+define <2 x half> @exp2_recip(<2 x half> %x, <2 x half> %y) {
+; CHECK-LABEL: @exp2_recip(
+; CHECK-NEXT:    [[P:%.*]] = call <2 x half> @llvm.exp2.v2f16(<2 x half> [[Y:%.*]])
+; CHECK-NEXT:    [[R:%.*]] = fdiv reassoc ninf arcp <2 x half> <half 0xH3C00, half 0xH3C00>, [[P]]
+; CHECK-NEXT:    ret <2 x half> [[R]]
+;
+  %p = call <2 x half> @llvm.exp2.v2f16(<2 x half> %y)
+  %r = fdiv reassoc arcp ninf <2 x half> <half 1.0, half 1.0>, %p
+  ret <2 x half> %r
+}