test/Transforms/LoopVectorize/hoist-loads.ll

   1 ; RUN: opt -loop-vectorize -force-vector-width=2 -force-vector-unroll=1 -S < %s | FileCheck %s
   2
   3 target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
   4
   5 @A = common global [1024 x float] zeroinitializer, align 16
   6 @B = common global [1024 x float] zeroinitializer, align 16
   7
   8 ; Make sure we can vectorize in the presence of hoistable conditional loads.
   9 ; CHECK-LABEL: @hoist_cond_load(
  10 ; CHECK: load <2 x float>
  11
  12 define void @hoist_cond_load() {
  13 entry:
  14   br label %for.body
  15 for.body:
  16   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %if.end9 ]
  17   %arrayidx = getelementptr inbounds [1024 x float]* @A, i64 0, i64 %indvars.iv
  18   %arrayidx2 = getelementptr inbounds [1024 x float]* @B, i64 0, i64 %indvars.iv
  19   %0 = load float* %arrayidx2, align 4
  20   %cmp3 = fcmp oeq float %0, 0.000000e+00
  21   br i1 %cmp3, label %if.end9, label %if.else
  22
  23 if.else:
  24   %1 = load float* %arrayidx, align 4
  25   br label %if.end9
  26
  27 if.end9:
  28   %tmp.0 = phi float [ %1, %if.else ], [ 0.000000e+00, %for.body ]
  29   store float %tmp.0, float* %arrayidx, align 4
  30   %indvars.iv.next = add i64 %indvars.iv, 1
  31   %lftr.wideiv = trunc i64 %indvars.iv.next to i32
  32   %exitcond = icmp ne i32 %lftr.wideiv, 1024
  33   br i1 %exitcond, label %for.body, label %for.end
  34
  35 for.end:
  36   ret void
  37 }
  38
  39 ; However, we can't hoist loads whose address we have not seen unconditionally
  40 ; accessed.
  41 ; CHECK-LABEL: @dont_hoist_cond_load(
  42 ; CHECK-NOT: load <2 x float>
  43
  44 define void @dont_hoist_cond_load() {
  45 entry:
  46   br label %for.body
  47 for.body:
  48   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %if.end9 ]
  49   %arrayidx = getelementptr inbounds [1024 x float]* @A, i64 0, i64 %indvars.iv
  50   %arrayidx2 = getelementptr inbounds [1024 x float]* @B, i64 0, i64 %indvars.iv
  51   %0 = load float* %arrayidx2, align 4
  52   %cmp3 = fcmp oeq float %0, 0.000000e+00
  53   br i1 %cmp3, label %if.end9, label %if.else
  54
  55 if.else:
  56   %1 = load float* %arrayidx, align 4
  57   br label %if.end9
  58
  59 if.end9:
  60   %tmp.0 = phi float [ %1, %if.else ], [ 0.000000e+00, %for.body ]
  61   store float %tmp.0, float* %arrayidx2, align 4
  62   %indvars.iv.next = add i64 %indvars.iv, 1
  63   %lftr.wideiv = trunc i64 %indvars.iv.next to i32
  64   %exitcond = icmp ne i32 %lftr.wideiv, 1024
  65   br i1 %exitcond, label %for.body, label %for.end
  66
  67 for.end:
  68   ret void
  69 }