[llvm] 5cbff43 - [InstCombine] Test for trunc to i1 in foldLogOpOfMaskedICmps.

[Andreas Jonson via llvm-commits]

Author: Andreas Jonson
Date: 2025-02-26T11:46:11+01:00
New Revision: 5cbff437fadd4c2983fb73e727c82044ae269a6f

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

LOG: [InstCombine] Test for trunc to i1 in foldLogOpOfMaskedICmps.

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/onehot_merge.ll b/llvm/test/Transforms/InstCombine/onehot_merge.ll
index 3b7314d36eaaa..e60edc7315d46 100644
--- a/llvm/test/Transforms/InstCombine/onehot_merge.ll
+++ b/llvm/test/Transforms/InstCombine/onehot_merge.ll
@@ -1159,3 +1159,39 @@ define i1 @two_types_of_bittest(i8 %x, i8 %c) {
   %ret = and i1 %icmp1, %icmp2
   ret i1 %ret
 }
+
+define i1 @trunc_bittest_and_icmp_bittest(i8 %x, i8 %c) {
+; CHECK-LABEL: @trunc_bittest_and_icmp_bittest(
+; CHECK-NEXT:    [[T0:%.*]] = shl nuw i8 1, [[C:%.*]]
+; CHECK-NEXT:    [[TRUNC:%.*]] = trunc i8 [[X:%.*]] to i1
+; CHECK-NEXT:    [[AND:%.*]] = and i8 [[X]], [[T0]]
+; CHECK-NEXT:    [[ICMP2:%.*]] = icmp ne i8 [[AND]], 0
+; CHECK-NEXT:    [[RET:%.*]] = and i1 [[ICMP2]], [[TRUNC]]
+; CHECK-NEXT:    ret i1 [[RET]]
+;
+  %t0 = shl i8 1, %c
+  %trunc = trunc i8 %x to i1
+  %and = and i8 %x, %t0
+  %icmp2 = icmp ne i8 %and, 0
+  %ret = and i1 %trunc, %icmp2
+  ret i1 %ret
+}
+
+define i1 @trunc_bittest_or_icmp_bittest(i8 %x, i8 %c) {
+; CHECK-LABEL: @trunc_bittest_or_icmp_bittest(
+; CHECK-NEXT:    [[T0:%.*]] = shl nuw i8 1, [[C:%.*]]
+; CHECK-NEXT:    [[TRUNC:%.*]] = trunc i8 [[X:%.*]] to i1
+; CHECK-NEXT:    [[AND:%.*]] = and i8 [[X]], [[T0]]
+; CHECK-NEXT:    [[ICMP2:%.*]] = icmp eq i8 [[AND]], 0
+; CHECK-NEXT:    [[NOT:%.*]] = xor i1 [[TRUNC]], true
+; CHECK-NEXT:    [[RET:%.*]] = or i1 [[ICMP2]], [[NOT]]
+; CHECK-NEXT:    ret i1 [[RET]]
+;
+  %t0 = shl i8 1, %c
+  %trunc = trunc i8 %x to i1
+  %and = and i8 %x, %t0
+  %icmp2 = icmp eq i8 %and, 0
+  %not = xor i1 %trunc, true
+  %ret = or i1 %not, %icmp2
+  ret i1 %ret
+}