[llvm] 6d970e8 - [InstCombine] Add PR51784 test cases

[Simon Pilgrim via llvm-commits]

Author: Simon Pilgrim
Date: 2021-09-13T08:36:43+01:00
New Revision: 6d970e83fac56cefc18d352a760cddb1c6d29e09

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

LOG: [InstCombine] Add PR51784 test cases

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/ctpop-cttz.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/ctpop-cttz.ll b/llvm/test/Transforms/InstCombine/ctpop-cttz.ll
index 88675a4973e7b..d7cefe62affaa 100644
--- a/llvm/test/Transforms/InstCombine/ctpop-cttz.ll
+++ b/llvm/test/Transforms/InstCombine/ctpop-cttz.ll
@@ -89,3 +89,50 @@ define i32 @ctpop2_multiuse(i32 %0) {
   %6 = add i32 %5, %4
   ret i32 %6
 }
+
+; PR51784
+; __builtin_popcount((i & -i) - 1) -> __builtin_cttz(i, false)
+define i32 @ctpop3(i32 %0) {
+; CHECK-LABEL: @ctpop3(
+; CHECK-NEXT:    [[TMP2:%.*]] = sub i32 0, [[TMP0:%.*]]
+; CHECK-NEXT:    [[TMP3:%.*]] = and i32 [[TMP2]], [[TMP0]]
+; CHECK-NEXT:    [[TMP4:%.*]] = add i32 [[TMP3]], -1
+; CHECK-NEXT:    [[TMP5:%.*]] = tail call i32 @llvm.ctpop.i32(i32 [[TMP4]]), !range [[RNG0]]
+; CHECK-NEXT:    ret i32 [[TMP5]]
+;
+  %2 = sub i32 0, %0
+  %3 = and i32 %2, %0
+  %4 = add i32 %3, -1
+  %5 = tail call i32 @llvm.ctpop.i32(i32 %4)
+  ret i32 %5
+}
+
+define <2 x i32> @ctpop3v(<2 x i32> %0) {
+; CHECK-LABEL: @ctpop3v(
+; CHECK-NEXT:    [[TMP2:%.*]] = sub <2 x i32> zeroinitializer, [[TMP0:%.*]]
+; CHECK-NEXT:    [[TMP3:%.*]] = and <2 x i32> [[TMP2]], [[TMP0]]
+; CHECK-NEXT:    [[TMP4:%.*]] = add <2 x i32> [[TMP3]], <i32 -1, i32 -1>
+; CHECK-NEXT:    [[TMP5:%.*]] = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> [[TMP4]])
+; CHECK-NEXT:    ret <2 x i32> [[TMP5]]
+;
+  %2 = sub <2 x i32> zeroinitializer, %0
+  %3 = and <2 x i32> %2, %0
+  %4 = add <2 x i32> %3, <i32 -1, i32 -1>
+  %5 = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %4)
+  ret <2 x i32> %5
+}
+
+define <2 x i32> @ctpop3v_undef(<2 x i32> %0) {
+; CHECK-LABEL: @ctpop3v_undef(
+; CHECK-NEXT:    [[TMP2:%.*]] = sub <2 x i32> zeroinitializer, [[TMP0:%.*]]
+; CHECK-NEXT:    [[TMP3:%.*]] = and <2 x i32> [[TMP2]], [[TMP0]]
+; CHECK-NEXT:    [[TMP4:%.*]] = add <2 x i32> [[TMP3]], <i32 -1, i32 undef>
+; CHECK-NEXT:    [[TMP5:%.*]] = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> [[TMP4]])
+; CHECK-NEXT:    ret <2 x i32> [[TMP5]]
+;
+  %2 = sub <2 x i32> zeroinitializer, %0
+  %3 = and <2 x i32> %2, %0
+  %4 = add <2 x i32> %3, <i32 -1, i32 undef>
+  %5 = tail call <2 x i32> @llvm.ctpop.v2i32(<2 x i32> %4)
+  ret <2 x i32> %5
+}