[llvm] 2d1fea2 - [InstCombine] Add vectors-with-undef tests for and(logicalshift(1,X),1) --> zext(X == 0)

[Simon Pilgrim via llvm-commits]

Author: Simon Pilgrim
Date: 2020-10-19T11:10:31+01:00
New Revision: 2d1fea29231f1998e7123b3a1ce177284ae41cec

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

LOG: [InstCombine] Add vectors-with-undef tests for and(logicalshift(1,X),1) --> zext(X == 0)

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/and2.ll b/llvm/test/Transforms/InstCombine/and2.ll
index 74b120458bac..6cf71eeea7a7 100644
--- a/llvm/test/Transforms/InstCombine/and2.ll
+++ b/llvm/test/Transforms/InstCombine/and2.ll
@@ -135,6 +135,17 @@ define <2 x i8> @and1_shl1_is_cmp_eq_0_vec(<2 x i8> %x) {
   ret <2 x i8> %and
 }
 
+define <2 x i8> @and1_shl1_is_cmp_eq_0_vec_undef(<2 x i8> %x) {
+; CHECK-LABEL: @and1_shl1_is_cmp_eq_0_vec_undef(
+; CHECK-NEXT:    [[SH:%.*]] = shl <2 x i8> <i8 1, i8 undef>, [[X:%.*]]
+; CHECK-NEXT:    [[AND:%.*]] = and <2 x i8> [[SH]], <i8 1, i8 undef>
+; CHECK-NEXT:    ret <2 x i8> [[AND]]
+;
+  %sh = shl <2 x i8> <i8 1, i8 undef>, %x
+  %and = and <2 x i8> %sh, <i8 1, i8 undef>
+  ret <2 x i8> %and
+}
+
 ; (1 >> x) & 1 --> zext(x == 0)
 
 define i8 @and1_lshr1_is_cmp_eq_0(i8 %x) {
@@ -176,6 +187,17 @@ define <2 x i8> @and1_lshr1_is_cmp_eq_0_vec(<2 x i8> %x) {
   ret <2 x i8> %and
 }
 
+define <2 x i8> @and1_lshr1_is_cmp_eq_0_vec_undef(<2 x i8> %x) {
+; CHECK-LABEL: @and1_lshr1_is_cmp_eq_0_vec_undef(
+; CHECK-NEXT:    [[SH:%.*]] = lshr <2 x i8> <i8 1, i8 undef>, [[X:%.*]]
+; CHECK-NEXT:    [[AND:%.*]] = and <2 x i8> [[SH]], <i8 1, i8 undef>
+; CHECK-NEXT:    ret <2 x i8> [[AND]]
+;
+  %sh = lshr <2 x i8> <i8 1, i8 undef>, %x
+  %and = and <2 x i8> %sh, <i8 1, i8 undef>
+  ret <2 x i8> %and
+}
+
 ; The add in this test is unnecessary because the LSBs of the LHS are 0 and the 'and' only consumes bits from those LSBs. It doesn't matter what happens to the upper bits.
 define i32 @test11(i32 %a, i32 %b) {
 ; CHECK-LABEL: @test11(