[llvm] 83ad4cb - [InstCombine] Add tests for folding `(icmp eq/ne (zext i1) (sext i1))`; NFC

[Noah Goldstein via llvm-commits]

Author: Noah Goldstein
Date: 2023-07-12T17:17:52-05:00
New Revision: 83ad4cb61f4d8f36be07e7d9d89529921f0c1d2f

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

LOG: [InstCombine] Add tests for folding `(icmp eq/ne (zext i1) (sext i1))`; NFC

Differential Revision: https://reviews.llvm.org/D154573

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/icmp-ext-ext.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/icmp-ext-ext.ll b/llvm/test/Transforms/InstCombine/icmp-ext-ext.ll
index 42472bb45dc024..c350358f488edb 100644
--- a/llvm/test/Transforms/InstCombine/icmp-ext-ext.ll
+++ b/llvm/test/Transforms/InstCombine/icmp-ext-ext.ll
@@ -380,3 +380,43 @@ define i1 @sext_zext_uge_known_nonneg_op0_wide(i16 %x, i8 %y) {
   %c = icmp uge i32 %a, %b
   ret i1 %c
 }
+
+
+define i1 @zext_eq_sext(i1 %a, i1 %b) {
+; CHECK-LABEL: @zext_eq_sext(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i1 [[A:%.*]] to i32
+; CHECK-NEXT:    [[CONV3_NEG:%.*]] = sext i1 [[B:%.*]] to i32
+; CHECK-NEXT:    [[TOBOOL4:%.*]] = icmp eq i32 [[CONV]], [[CONV3_NEG]]
+; CHECK-NEXT:    ret i1 [[TOBOOL4]]
+;
+  %conv = zext i1 %a to i32
+  %conv3.neg = sext i1 %b to i32
+  %tobool4 = icmp eq i32 %conv, %conv3.neg
+  ret i1 %tobool4
+}
+
+define i1 @zext_eq_sext_fail_not_i1(i1 %a, i8 %b) {
+; CHECK-LABEL: @zext_eq_sext_fail_not_i1(
+; CHECK-NEXT:    [[CONV:%.*]] = zext i1 [[A:%.*]] to i32
+; CHECK-NEXT:    [[CONV3_NEG:%.*]] = sext i8 [[B:%.*]] to i32
+; CHECK-NEXT:    [[TOBOOL4:%.*]] = icmp eq i32 [[CONV]], [[CONV3_NEG]]
+; CHECK-NEXT:    ret i1 [[TOBOOL4]]
+;
+  %conv = zext i1 %a to i32
+  %conv3.neg = sext i8 %b to i32
+  %tobool4 = icmp eq i32 %conv, %conv3.neg
+  ret i1 %tobool4
+}
+
+define <2 x i1> @zext_ne_sext(<2 x i1> %a, <2 x i1> %b) {
+; CHECK-LABEL: @zext_ne_sext(
+; CHECK-NEXT:    [[CONV:%.*]] = zext <2 x i1> [[A:%.*]] to <2 x i8>
+; CHECK-NEXT:    [[CONV3_NEG:%.*]] = sext <2 x i1> [[B:%.*]] to <2 x i8>
+; CHECK-NEXT:    [[TOBOOL4:%.*]] = icmp ne <2 x i8> [[CONV3_NEG]], [[CONV]]
+; CHECK-NEXT:    ret <2 x i1> [[TOBOOL4]]
+;
+  %conv = zext <2 x i1> %a to <2 x i8>
+  %conv3.neg = sext <2 x i1> %b to <2 x i8>
+  %tobool4 = icmp ne <2 x i8> %conv3.neg, %conv
+  ret <2 x i1> %tobool4
+}