llvm/test/Transforms/LoopVectorize/icmp-uniforms.ll

   1 ; REQUIRES: asserts
   2 ; RUN: opt < %s -passes=loop-vectorize,instcombine -force-vector-width=4 -force-vector-interleave=1 -debug-only=loop-vectorize -disable-output -print-after=instcombine 2>&1 | FileCheck %s
   3
   4 target datalayout = "e-m:e-i64:64-i128:128-n32:64-S128"
   5
   6 ; CHECK-LABEL: more_than_one_use
   7 ;
   8 ; PR30627. Check that a compare instruction with more than one use is not
   9 ; recognized as uniform and is vectorized.
  10 ;
  11 ; CHECK-NOT: Found uniform instruction: %cond = icmp slt i64 %i.next, %n
  12 ; CHECK:     vector.body
  13 ; CHECK:       %[[I:.+]] = add nuw nsw <4 x i64> %vec.ind, <i64 1, i64 1, i64 1, i64 1>
  14 ; CHECK:       icmp slt <4 x i64> %[[I]], %broadcast.splat
  15 ; CHECK:       br i1 {{.*}}, label %middle.block, label %vector.body
  16 ;
  17 define i32 @more_than_one_use(ptr %a, i64 %n) {
  18 entry:
  19   br label %for.body
  20
  21 for.body:
  22   %i = phi i64 [ %i.next, %for.body ], [ 0, %entry ]
  23   %r = phi i32 [ %tmp3, %for.body ], [ 0, %entry ]
  24   %i.next = add nuw nsw i64 %i, 1
  25   %cond = icmp slt i64 %i.next, %n
  26   %tmp0 = select i1 %cond, i64 %i.next, i64 0
  27   %tmp1 = getelementptr inbounds i32, ptr %a, i64 %tmp0
  28   %tmp2 = load i32, ptr %tmp1, align 8
  29   %tmp3 = add i32 %r, %tmp2
  30   br i1 %cond, label %for.body, label %for.end
  31
  32 for.end:
  33   %tmp4 = phi i32 [ %tmp3, %for.body ]
  34   ret i32 %tmp4
  35 }
  36
  37 ; Check for crash exposed by D76992.
  38 ; CHECK-LABEL: 'test'
  39 ; CHECK:      VPlan 'Initial VPlan for VF={4},UF>=1' {
  40 ; CHECK-NEXT: Live-in vp<[[VEC_TC:%.+]]> = vector-trip-count
  41 ; CHECK-NEXT: Live-in vp<[[BTC:%.+]]> = backedge-taken count
  42 ; CHECK-NEXT: Live-in ir<14> = original trip-count
  43 ; CHECK-EMPTY:
  44 ; CHECK-NEXT: vector.ph:
  45 ; CHECK-NEXT: Successor(s): vector loop
  46 ; CHECK-EMPTY:
  47 ; CHECK-NEXT: <x1> vector loop: {
  48 ; CHECK-NEXT: vector.body:
  49 ; CHECK-NEXT:   EMIT vp<[[CAN_IV:%.+]]> = CANONICAL-INDUCTION
  50 ; CHECK-NEXT:   WIDEN-INDUCTION %iv = phi 0, %iv.next, ir<1>
  51 ; CHECK-NEXT:   EMIT vp<[[COND:%.+]]> = icmp ule ir<%iv>, vp<[[BTC]]>
  52 ; CHECK-NEXT:   WIDEN ir<%cond0> = icmp ult ir<%iv>, ir<13>
  53 ; CHECK-NEXT:   WIDEN-SELECT ir<%s> = select ir<%cond0>, ir<10>, ir<20>
  54 ; CHECK-NEXT: Successor(s): pred.store
  55 ; CHECK-EMPTY:
  56 ; CHECK-NEXT:  <xVFxUF> pred.store: {
  57 ; CHECK-NEXT:    pred.store.entry:
  58 ; CHECK-NEXT:      BRANCH-ON-MASK vp<[[COND]]>
  59 ; CHECK-NEXT:    Successor(s): pred.store.if, pred.store.continue
  60 ; CHECK-EMPTY:
  61 ; CHECK-NEXT:    pred.store.if:
  62 ; CHECK-NEXT:      vp<[[STEPS:%.+]]> = SCALAR-STEPS vp<[[CAN_IV]]>, ir<1>
  63 ; CHECK-NEXT:      REPLICATE ir<%gep> = getelementptr inbounds ir<%ptr>, vp<[[STEPS]]>
  64 ; CHECK-NEXT:      REPLICATE store ir<%s>, ir<%gep>
  65 ; CHECK-NEXT:    Successor(s): pred.store.continue
  66 ; CHECK-EMPTY:
  67 ; CHECK-NEXT:    pred.store.continue:
  68 ; CHECK-NEXT:    No successors
  69 ; CHECK-NEXT:  }
  70 ; CHECK-NEXT:  Successor(s): loop.0
  71 ; CHECK-EMPTY:
  72 ; CHECK-NEXT: loop.0:
  73 ; CHECK-NEXT:   EMIT vp<[[CAN_IV_NEXT:%.+]]> = VF * UF + vp<[[CAN_IV]]>
  74 ; CHECK-NEXT:   EMIT branch-on-count vp<[[CAN_IV_NEXT]]>, vp<[[VEC_TC]]>
  75 ; CHECK-NEXT: No successor
  76 ; CHECK-NEXT: }
  77 define void @test(ptr %ptr) {
  78 entry:
  79   br label %loop
  80
  81 loop:                       ; preds = %loop, %entry
  82   %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
  83   %cond0 = icmp ult i64 %iv, 13
  84   %s = select i1 %cond0, i32 10, i32 20
  85   %gep = getelementptr inbounds i32, ptr %ptr, i64 %iv
  86   store i32 %s, ptr %gep
  87   %iv.next = add nuw nsw i64 %iv, 1
  88   %exitcond = icmp eq i64 %iv.next, 14
  89   br i1 %exitcond, label %exit, label %loop
  90
  91 exit:
  92   ret void
  93 }