diff --git a/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll b/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll index 7845d314b596fb..9eb40596f94e36 100644 --- a/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll +++ b/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll @@ -533,3 +533,50 @@ define i1 @logical_or_icmp_extra_op(<4 x i32> %x, <4 x i32> %y, i1 %c) { %s7 = select i1 %s6, i1 true, i1 %d3 ret i1 %s7 } + +define i1 @logical_and_icmp_extra_args(<4 x i32> %x) { +; SSE-LABEL: @logical_and_icmp_extra_args( +; SSE-NEXT: [[TMP1:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 1 +; SSE-NEXT: [[TMP2:%.*]] = extractelement <4 x i32> [[X]], i32 0 +; SSE-NEXT: [[TMP3:%.*]] = insertelement <2 x i32> poison, i32 [[TMP1]], i32 0 +; SSE-NEXT: [[TMP4:%.*]] = insertelement <2 x i32> [[TMP3]], i32 [[TMP2]], i32 1 +; SSE-NEXT: [[TMP5:%.*]] = icmp slt <2 x i32> [[TMP4]], +; SSE-NEXT: [[TMP6:%.*]] = icmp sgt <4 x i32> [[X]], +; SSE-NEXT: [[TMP7:%.*]] = freeze <4 x i1> [[TMP6]] +; SSE-NEXT: [[TMP8:%.*]] = call i1 @llvm.vector.reduce.and.v4i1(<4 x i1> [[TMP7]]) +; SSE-NEXT: [[TMP9:%.*]] = extractelement <2 x i1> [[TMP5]], i32 0 +; SSE-NEXT: [[TMP10:%.*]] = extractelement <2 x i1> [[TMP5]], i32 1 +; SSE-NEXT: [[OP_RDX:%.*]] = select i1 [[TMP10]], i1 [[TMP9]], i1 false +; SSE-NEXT: [[OP_RDX1:%.*]] = select i1 [[TMP8]], i1 [[OP_RDX]], i1 false +; SSE-NEXT: ret i1 [[OP_RDX1]] +; +; AVX-LABEL: @logical_and_icmp_extra_args( +; AVX-NEXT: [[TMP1:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 1 +; AVX-NEXT: [[TMP2:%.*]] = extractelement <4 x i32> [[X]], i32 0 +; AVX-NEXT: [[C0:%.*]] = icmp slt i32 [[TMP2]], 42 +; AVX-NEXT: [[C1:%.*]] = icmp slt i32 [[TMP1]], 42 +; AVX-NEXT: [[TMP3:%.*]] = icmp sgt <4 x i32> [[X]], +; AVX-NEXT: [[TMP4:%.*]] = freeze <4 x i1> [[TMP3]] +; AVX-NEXT: [[TMP5:%.*]] = call i1 @llvm.vector.reduce.and.v4i1(<4 x i1> [[TMP4]]) +; AVX-NEXT: [[OP_RDX:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false +; AVX-NEXT: [[OP_RDX1:%.*]] = select i1 [[TMP5]], i1 [[OP_RDX]], i1 false +; AVX-NEXT: ret i1 [[OP_RDX1]] +; + %x0 = extractelement <4 x i32> %x, i32 0 + %x1 = extractelement <4 x i32> %x, i32 1 + %x2 = extractelement <4 x i32> %x, i32 2 + %x3 = extractelement <4 x i32> %x, i32 3 + %c0 = icmp slt i32 %x0, 42 + %c1 = icmp slt i32 %x1, 42 + %d0 = icmp sgt i32 %x0, 17 + %d1 = icmp sgt i32 %x1, 17 + %d2 = icmp sgt i32 %x2, 17 + %d3 = icmp sgt i32 %x3, 17 + %s1 = select i1 %d0, i1 %c0, i1 false ; <- d0, d1, d2, d3 gets reduced. + %s4 = select i1 %s1, i1 %c1, i1 false ; <- c0, c1 remain scalar logical and. + %s5 = select i1 %s4, i1 %d1, i1 false + %s6 = select i1 %s5, i1 %d2, i1 false + %s7 = select i1 %s6, i1 %d3, i1 false + ret i1 %s7 +} +