[llvm] b05f1d9 - [InstCombine] Add pre-commit tests for PR57328. NFC.

[Yingwei Zheng via llvm-commits]

Author: Yingwei Zheng
Date: 2023-09-24T23:00:50+08:00
New Revision: b05f1d93469fbd645118739798e51f21a9e410de

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

LOG: [InstCombine] Add pre-commit tests for PR57328. NFC.

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/ispow2.ll b/llvm/test/Transforms/InstCombine/ispow2.ll
index 1fa0d4c4af054b3..bbd693b11b388ad 100644
--- a/llvm/test/Transforms/InstCombine/ispow2.ll
+++ b/llvm/test/Transforms/InstCombine/ispow2.ll
@@ -175,6 +175,7 @@ define i1 @is_pow2or0_negate_op_extra_use2(i32 %x) {
 
 declare i32 @llvm.ctpop.i32(i32)
 declare <2 x i8> @llvm.ctpop.v2i8(<2 x i8>)
+declare void @llvm.assume(i1)
 
 ; (X != 0) && (ctpop(X) u< 2) --> ctpop(X) == 1
 
@@ -191,6 +192,22 @@ define i1 @is_pow2_ctpop(i32 %x) {
   ret i1 %r
 }
 
+; tests from PR57328
+define i1 @is_pow2_non_zero(i32 %x) {
+; CHECK-LABEL: @is_pow2_non_zero(
+; CHECK-NEXT:    [[NOTZERO:%.*]] = icmp ne i32 [[X:%.*]], 0
+; CHECK-NEXT:    call void @llvm.assume(i1 [[NOTZERO]])
+; CHECK-NEXT:    [[T0:%.*]] = tail call i32 @llvm.ctpop.i32(i32 [[X]]), !range [[RNG0]]
+; CHECK-NEXT:    [[CMP:%.*]] = icmp ult i32 [[T0]], 2
+; CHECK-NEXT:    ret i1 [[CMP]]
+;
+  %notzero = icmp ne i32 %x, 0
+  call void @llvm.assume(i1 %notzero)
+  %t0 = tail call i32 @llvm.ctpop.i32(i32 %x)
+  %cmp = icmp ult i32 %t0, 2
+  ret i1 %cmp
+}
+
 define i1 @is_pow2_ctpop_logical(i32 %x) {
 ; CHECK-LABEL: @is_pow2_ctpop_logical(
 ; CHECK-NEXT:    [[T0:%.*]] = tail call i32 @llvm.ctpop.i32(i32 [[X:%.*]]), !range [[RNG0]]