[InstCombine] add tests for pow reassociation; NFC
authorSanjay Patel <spatel@rotateright.com>
Fri, 13 Jan 2023 17:30:15 +0000 (12:30 -0500)
committerSanjay Patel <spatel@rotateright.com>
Fri, 13 Jan 2023 18:26:10 +0000 (13:26 -0500)
llvm/test/Transforms/InstCombine/fmul-pow.ll

index a14b834..e522640 100644 (file)
@@ -64,6 +64,53 @@ define double @pow_ab_pow_cb_reassoc(double %a, double %b, double %c) {
   ret double %mul
 }
 
+define double @pow_ab_pow_cb_reassoc_use1(double %a, double %b, double %c) {
+; CHECK-LABEL: @pow_ab_pow_cb_reassoc_use1(
+; CHECK-NEXT:    [[AB:%.*]] = call double @llvm.pow.f64(double [[A:%.*]], double [[B:%.*]])
+; CHECK-NEXT:    [[CB:%.*]] = call double @llvm.pow.f64(double [[C:%.*]], double [[B]])
+; CHECK-NEXT:    [[MUL:%.*]] = fmul reassoc double [[AB]], [[CB]]
+; CHECK-NEXT:    call void @use(double [[AB]])
+; CHECK-NEXT:    ret double [[MUL]]
+;
+  %ab = call double @llvm.pow.f64(double %a, double %b)
+  %cb = call double @llvm.pow.f64(double %c, double %b)
+  %mul = fmul reassoc double %ab, %cb
+  call void @use(double %ab)
+  ret double %mul
+}
+
+define double @pow_ab_pow_cb_reassoc_use2(double %a, double %b, double %c) {
+; CHECK-LABEL: @pow_ab_pow_cb_reassoc_use2(
+; CHECK-NEXT:    [[AB:%.*]] = call double @llvm.pow.f64(double [[A:%.*]], double [[B:%.*]])
+; CHECK-NEXT:    [[CB:%.*]] = call double @llvm.pow.f64(double [[C:%.*]], double [[B]])
+; CHECK-NEXT:    [[MUL:%.*]] = fmul reassoc double [[AB]], [[CB]]
+; CHECK-NEXT:    call void @use(double [[CB]])
+; CHECK-NEXT:    ret double [[MUL]]
+;
+  %ab = call double @llvm.pow.f64(double %a, double %b)
+  %cb = call double @llvm.pow.f64(double %c, double %b)
+  %mul = fmul reassoc double %ab, %cb
+  call void @use(double %cb)
+  ret double %mul
+}
+
+define double @pow_ab_pow_cb_reassoc_use3(double %a, double %b, double %c) {
+; CHECK-LABEL: @pow_ab_pow_cb_reassoc_use3(
+; CHECK-NEXT:    [[AB:%.*]] = call double @llvm.pow.f64(double [[A:%.*]], double [[B:%.*]])
+; CHECK-NEXT:    [[CB:%.*]] = call double @llvm.pow.f64(double [[C:%.*]], double [[B]])
+; CHECK-NEXT:    [[MUL:%.*]] = fmul reassoc double [[AB]], [[CB]]
+; CHECK-NEXT:    call void @use(double [[AB]])
+; CHECK-NEXT:    call void @use(double [[CB]])
+; CHECK-NEXT:    ret double [[MUL]]
+;
+  %ab = call double @llvm.pow.f64(double %a, double %b)
+  %cb = call double @llvm.pow.f64(double %c, double %b)
+  %mul = fmul reassoc double %ab, %cb
+  call void @use(double %ab)
+  call void @use(double %cb)
+  ret double %mul
+}
+
 define double @pow_ab_pow_ac(double %a, double %b, double %c) {
 ; CHECK-LABEL: @pow_ab_pow_ac(
 ; CHECK-NEXT:    [[TMP1:%.*]] = call double @llvm.pow.f64(double [[A:%.*]], double [[B:%.*]])
@@ -80,8 +127,8 @@ define double @pow_ab_pow_ac(double %a, double %b, double %c) {
 define double @pow_ab_x_pow_ac_reassoc(double %a, double %b, double %c) {
 ; CHECK-LABEL: @pow_ab_x_pow_ac_reassoc(
 ; CHECK-NEXT:    [[TMP1:%.*]] = fadd reassoc double [[C:%.*]], [[B:%.*]]
-; CHECK-NEXT:    [[TMP2:%.*]] = call reassoc double @llvm.pow.f64(double [[A:%.*]], double [[TMP1]])
-; CHECK-NEXT:    ret double [[TMP2]]
+; CHECK-NEXT:    [[MUL:%.*]] = call reassoc double @llvm.pow.f64(double [[A:%.*]], double [[TMP1]])
+; CHECK-NEXT:    ret double [[MUL]]
 ;
   %1 = call double @llvm.pow.f64(double %a, double %b)
   %2 = call double @llvm.pow.f64(double %a, double %c)
@@ -92,8 +139,8 @@ define double @pow_ab_x_pow_ac_reassoc(double %a, double %b, double %c) {
 define double @pow_ab_reassoc(double %a, double %b) {
 ; CHECK-LABEL: @pow_ab_reassoc(
 ; CHECK-NEXT:    [[TMP1:%.*]] = fadd reassoc double [[B:%.*]], [[B]]
-; CHECK-NEXT:    [[TMP2:%.*]] = call reassoc double @llvm.pow.f64(double [[A:%.*]], double [[TMP1]])
-; CHECK-NEXT:    ret double [[TMP2]]
+; CHECK-NEXT:    [[MUL:%.*]] = call reassoc double @llvm.pow.f64(double [[A:%.*]], double [[TMP1]])
+; CHECK-NEXT:    ret double [[MUL]]
 ;
   %1 = call double @llvm.pow.f64(double %a, double %b)
   %mul = fmul reassoc double %1, %1
@@ -117,9 +164,9 @@ define double @pow_ab_x_pow_ac_reassoc_extra_use(double %a, double %b, double %c
 ; CHECK-LABEL: @pow_ab_x_pow_ac_reassoc_extra_use(
 ; CHECK-NEXT:    [[TMP1:%.*]] = call double @llvm.pow.f64(double [[A:%.*]], double [[B:%.*]])
 ; CHECK-NEXT:    [[TMP2:%.*]] = fadd reassoc double [[B]], [[C:%.*]]
-; CHECK-NEXT:    [[TMP3:%.*]] = call reassoc double @llvm.pow.f64(double [[A]], double [[TMP2]])
+; CHECK-NEXT:    [[MUL:%.*]] = call reassoc double @llvm.pow.f64(double [[A]], double [[TMP2]])
 ; CHECK-NEXT:    call void @use(double [[TMP1]])
-; CHECK-NEXT:    ret double [[TMP3]]
+; CHECK-NEXT:    ret double [[MUL]]
 ;
   %1 = call double @llvm.pow.f64(double %a, double %b)
   %2 = call double @llvm.pow.f64(double %a, double %c)