Sanjay Patel | b653de1 | 2014-09-10 17:58:16 +0000 | [diff] [blame] | 1 | ; RUN: opt < %s -loop-vectorize -force-vector-interleave=1 -force-vector-width=4 -dce -instcombine -S | FileCheck %s |
Nadav Rotem | 1cfef3e | 2012-11-09 07:09:44 +0000 | [diff] [blame] | 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 | target triple = "x86_64-apple-macosx10.9.0" |
| 5 | |
| 6 | ; Make sure we vectorize this loop: |
| 7 | ; int foo(float *a, float *b, int n) { |
| 8 | ; for (int i=0; i<n; ++i) |
| 9 | ; a[i] = b[i] * 3; |
| 10 | ; } |
| 11 | |
Benjamin Kramer | d455ed8 | 2013-01-19 13:57:58 +0000 | [diff] [blame] | 12 | ;CHECK: for.body.preheader: |
| 13 | ;CHECK: br i1 %cmp.zero, label %middle.block, label %vector.memcheck |
| 14 | ;CHECK: vector.memcheck: |
Arnold Schwaighofer | b252c11 | 2013-06-24 12:09:15 +0000 | [diff] [blame] | 15 | ;CHECK: br i1 %memcheck.conflict, label %middle.block, label %vector.ph |
Nadav Rotem | 1cfef3e | 2012-11-09 07:09:44 +0000 | [diff] [blame] | 16 | ;CHECK: load <4 x float> |
| 17 | define i32 @foo(float* nocapture %a, float* nocapture %b, i32 %n) nounwind uwtable ssp { |
| 18 | entry: |
| 19 | %cmp6 = icmp sgt i32 %n, 0 |
| 20 | br i1 %cmp6, label %for.body, label %for.end |
| 21 | |
| 22 | for.body: ; preds = %entry, %for.body |
| 23 | %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ] |
David Blaikie | 79e6c74 | 2015-02-27 19:29:02 +0000 | [diff] [blame^] | 24 | %arrayidx = getelementptr inbounds float, float* %b, i64 %indvars.iv |
Manman Ren | 1a5ff28 | 2013-04-30 17:52:57 +0000 | [diff] [blame] | 25 | %0 = load float* %arrayidx, align 4 |
Nadav Rotem | 1cfef3e | 2012-11-09 07:09:44 +0000 | [diff] [blame] | 26 | %mul = fmul float %0, 3.000000e+00 |
David Blaikie | 79e6c74 | 2015-02-27 19:29:02 +0000 | [diff] [blame^] | 27 | %arrayidx2 = getelementptr inbounds float, float* %a, i64 %indvars.iv |
Manman Ren | 1a5ff28 | 2013-04-30 17:52:57 +0000 | [diff] [blame] | 28 | store float %mul, float* %arrayidx2, align 4 |
Nadav Rotem | 1cfef3e | 2012-11-09 07:09:44 +0000 | [diff] [blame] | 29 | %indvars.iv.next = add i64 %indvars.iv, 1 |
| 30 | %lftr.wideiv = trunc i64 %indvars.iv.next to i32 |
| 31 | %exitcond = icmp eq i32 %lftr.wideiv, %n |
| 32 | br i1 %exitcond, label %for.end, label %for.body |
| 33 | |
| 34 | for.end: ; preds = %for.body, %entry |
| 35 | ret i32 undef |
| 36 | } |
Arnold Schwaighofer | 70a4665 | 2013-11-01 03:05:07 +0000 | [diff] [blame] | 37 | |
| 38 | ; Make sure that we try to vectorize loops with a runtime check if the |
| 39 | ; dependency check fails. |
| 40 | |
| 41 | ; CHECK-LABEL: test_runtime_check |
| 42 | ; CHECK: <4 x float> |
| 43 | define void @test_runtime_check(float* %a, float %b, i64 %offset, i64 %offset2, i64 %n) { |
| 44 | entry: |
| 45 | br label %for.body |
| 46 | |
| 47 | for.body: |
| 48 | %iv = phi i64 [ 0, %entry ], [ %iv.next, %for.body ] |
| 49 | %ind.sum = add i64 %iv, %offset |
David Blaikie | 79e6c74 | 2015-02-27 19:29:02 +0000 | [diff] [blame^] | 50 | %arr.idx = getelementptr inbounds float, float* %a, i64 %ind.sum |
Arnold Schwaighofer | 70a4665 | 2013-11-01 03:05:07 +0000 | [diff] [blame] | 51 | %l1 = load float* %arr.idx, align 4 |
| 52 | %ind.sum2 = add i64 %iv, %offset2 |
David Blaikie | 79e6c74 | 2015-02-27 19:29:02 +0000 | [diff] [blame^] | 53 | %arr.idx2 = getelementptr inbounds float, float* %a, i64 %ind.sum2 |
Arnold Schwaighofer | 70a4665 | 2013-11-01 03:05:07 +0000 | [diff] [blame] | 54 | %l2 = load float* %arr.idx2, align 4 |
| 55 | %m = fmul fast float %b, %l2 |
| 56 | %ad = fadd fast float %l1, %m |
| 57 | store float %ad, float* %arr.idx, align 4 |
| 58 | %iv.next = add nuw nsw i64 %iv, 1 |
| 59 | %exitcond = icmp eq i64 %iv.next, %n |
| 60 | br i1 %exitcond, label %loopexit, label %for.body |
| 61 | |
| 62 | loopexit: |
| 63 | ret void |
| 64 | } |