1*b3cba9beSMel Chen; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5 2*b3cba9beSMel Chen; RUN: opt -passes=loop-vectorize -force-vector-interleave=1 -force-vector-width=4 -S < %s | FileCheck %s --check-prefix=CHECK 3*b3cba9beSMel Chen 4*b3cba9beSMel Chendefine i32 @select_icmp_switch(i32 %n, i32 %case, ptr %a, ptr %b) { 5*b3cba9beSMel Chen; CHECK-LABEL: define i32 @select_icmp_switch( 6*b3cba9beSMel Chen; CHECK-SAME: i32 [[N:%.*]], i32 [[CASE:%.*]], ptr [[A:%.*]], ptr [[B:%.*]]) { 7*b3cba9beSMel Chen; CHECK-NEXT: [[ENTRY:.*]]: 8*b3cba9beSMel Chen; CHECK-NEXT: [[CMP_SGT:%.*]] = icmp sgt i32 [[N]], 0 9*b3cba9beSMel Chen; CHECK-NEXT: br i1 [[CMP_SGT]], label %[[FOR_BODY_PREHEADER:.*]], label %[[FOR_END:.*]] 10*b3cba9beSMel Chen; CHECK: [[FOR_BODY_PREHEADER]]: 11*b3cba9beSMel Chen; CHECK-NEXT: [[WIDE_TRIP_COUNT:%.*]] = zext i32 [[N]] to i64 12*b3cba9beSMel Chen; CHECK-NEXT: br label %[[FOR_BODY:.*]] 13*b3cba9beSMel Chen; CHECK: [[FOR_BODY]]: 14*b3cba9beSMel Chen; CHECK-NEXT: [[INDVARS:%.*]] = phi i64 [ 0, %[[FOR_BODY_PREHEADER]] ], [ [[INDVARS_NEXT:%.*]], %[[FOR_INC:.*]] ] 15*b3cba9beSMel Chen; CHECK-NEXT: [[RDX_PHI:%.*]] = phi i32 [ 0, %[[FOR_BODY_PREHEADER]] ], [ [[RDX_PHI_NEXT:%.*]], %[[FOR_INC]] ] 16*b3cba9beSMel Chen; CHECK-NEXT: switch i32 [[CASE]], label %[[SW_BB0:.*]] [ 17*b3cba9beSMel Chen; CHECK-NEXT: i32 0, label %[[SW_BB0]] 18*b3cba9beSMel Chen; CHECK-NEXT: i32 1, label %[[SW_BB1:.*]] 19*b3cba9beSMel Chen; CHECK-NEXT: ] 20*b3cba9beSMel Chen; CHECK: [[SW_BB0]]: 21*b3cba9beSMel Chen; CHECK-NEXT: [[A_ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[A]], i64 [[INDVARS]] 22*b3cba9beSMel Chen; CHECK-NEXT: [[A_VALUE:%.*]] = load i8, ptr [[A_ARRAYIDX]], align 1 23*b3cba9beSMel Chen; CHECK-NEXT: [[CMP_A:%.*]] = icmp eq i8 [[A_VALUE]], -1 24*b3cba9beSMel Chen; CHECK-NEXT: [[TRUNC_BB0:%.*]] = trunc i64 [[INDVARS]] to i32 25*b3cba9beSMel Chen; CHECK-NEXT: [[SELECT_BB0:%.*]] = select i1 [[CMP_A]], i32 [[RDX_PHI]], i32 [[TRUNC_BB0]] 26*b3cba9beSMel Chen; CHECK-NEXT: br label %[[FOR_INC]] 27*b3cba9beSMel Chen; CHECK: [[SW_BB1]]: 28*b3cba9beSMel Chen; CHECK-NEXT: [[B_ARRAYIDX:%.*]] = getelementptr inbounds i8, ptr [[B]], i64 [[INDVARS]] 29*b3cba9beSMel Chen; CHECK-NEXT: [[B_VALUE:%.*]] = load i8, ptr [[B_ARRAYIDX]], align 1 30*b3cba9beSMel Chen; CHECK-NEXT: [[CMP_B:%.*]] = icmp eq i8 [[B_VALUE]], -1 31*b3cba9beSMel Chen; CHECK-NEXT: [[TRUNC_BB1:%.*]] = trunc i64 [[INDVARS]] to i32 32*b3cba9beSMel Chen; CHECK-NEXT: [[SELECT_BB1:%.*]] = select i1 [[CMP_B]], i32 [[RDX_PHI]], i32 [[TRUNC_BB1]] 33*b3cba9beSMel Chen; CHECK-NEXT: br label %[[FOR_INC]] 34*b3cba9beSMel Chen; CHECK: [[FOR_INC]]: 35*b3cba9beSMel Chen; CHECK-NEXT: [[RDX_PHI_NEXT]] = phi i32 [ [[SELECT_BB0]], %[[SW_BB0]] ], [ [[SELECT_BB1]], %[[SW_BB1]] ] 36*b3cba9beSMel Chen; CHECK-NEXT: [[INDVARS_NEXT]] = add nuw nsw i64 [[INDVARS]], 1 37*b3cba9beSMel Chen; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i64 [[INDVARS_NEXT]], [[WIDE_TRIP_COUNT]] 38*b3cba9beSMel Chen; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label %[[FOR_END_LOOPEXIT:.*]], label %[[FOR_BODY]] 39*b3cba9beSMel Chen; CHECK: [[FOR_END_LOOPEXIT]]: 40*b3cba9beSMel Chen; CHECK-NEXT: [[RDX_PHI_NEXT_LCSSA:%.*]] = phi i32 [ [[RDX_PHI_NEXT]], %[[FOR_INC]] ] 41*b3cba9beSMel Chen; CHECK-NEXT: br label %[[FOR_END]] 42*b3cba9beSMel Chen; CHECK: [[FOR_END]]: 43*b3cba9beSMel Chen; CHECK-NEXT: [[SELECT_LCSSA:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[RDX_PHI_NEXT_LCSSA]], %[[FOR_END_LOOPEXIT]] ] 44*b3cba9beSMel Chen; CHECK-NEXT: ret i32 [[SELECT_LCSSA]] 45*b3cba9beSMel Chen; 46*b3cba9beSMel Chenentry: 47*b3cba9beSMel Chen %cmp.sgt = icmp sgt i32 %n, 0 48*b3cba9beSMel Chen br i1 %cmp.sgt, label %for.body.preheader, label %for.end 49*b3cba9beSMel Chen 50*b3cba9beSMel Chenfor.body.preheader: 51*b3cba9beSMel Chen %wide.trip.count = zext i32 %n to i64 52*b3cba9beSMel Chen br label %for.body 53*b3cba9beSMel Chen 54*b3cba9beSMel Chenfor.body: 55*b3cba9beSMel Chen %indvars = phi i64 [ 0, %for.body.preheader ], [ %indvars.next, %for.inc ] 56*b3cba9beSMel Chen %rdx.phi = phi i32 [ 0, %for.body.preheader ], [ %rdx.phi.next, %for.inc ] 57*b3cba9beSMel Chen switch i32 %case, label %sw.bb0 [ 58*b3cba9beSMel Chen i32 0, label %sw.bb0 59*b3cba9beSMel Chen i32 1, label %sw.bb1 60*b3cba9beSMel Chen ] 61*b3cba9beSMel Chen 62*b3cba9beSMel Chensw.bb0: 63*b3cba9beSMel Chen %a.arrayidx = getelementptr inbounds i8, ptr %a, i64 %indvars 64*b3cba9beSMel Chen %a.value = load i8, ptr %a.arrayidx, align 1 65*b3cba9beSMel Chen %cmp.a = icmp eq i8 %a.value, -1 66*b3cba9beSMel Chen %trunc.bb0 = trunc i64 %indvars to i32 67*b3cba9beSMel Chen %select.bb0 = select i1 %cmp.a, i32 %rdx.phi, i32 %trunc.bb0 68*b3cba9beSMel Chen br label %for.inc 69*b3cba9beSMel Chen 70*b3cba9beSMel Chensw.bb1: 71*b3cba9beSMel Chen %b.arrayidx = getelementptr inbounds i8, ptr %b, i64 %indvars 72*b3cba9beSMel Chen %b.value = load i8, ptr %b.arrayidx, align 1 73*b3cba9beSMel Chen %cmp.b = icmp eq i8 %b.value, -1 74*b3cba9beSMel Chen %trunc.bb1 = trunc i64 %indvars to i32 75*b3cba9beSMel Chen %select.bb1 = select i1 %cmp.b, i32 %rdx.phi, i32 %trunc.bb1 76*b3cba9beSMel Chen br label %for.inc 77*b3cba9beSMel Chen 78*b3cba9beSMel Chenfor.inc: 79*b3cba9beSMel Chen %rdx.phi.next = phi i32 [ %select.bb0, %sw.bb0 ], [ %select.bb1, %sw.bb1 ] 80*b3cba9beSMel Chen %indvars.next = add nuw nsw i64 %indvars, 1 81*b3cba9beSMel Chen %exitcond.not = icmp eq i64 %indvars.next, %wide.trip.count 82*b3cba9beSMel Chen br i1 %exitcond.not, label %for.end, label %for.body 83*b3cba9beSMel Chen 84*b3cba9beSMel Chenfor.end: 85*b3cba9beSMel Chen %select.lcssa = phi i32 [ %rdx.phi.next, %for.inc ], [ 0, %entry ] 86*b3cba9beSMel Chen ret i32 %select.lcssa 87*b3cba9beSMel Chen} 88