D3348 - [BUG] "Rotate Loop" pass kills "llvm.vectorizer.enable" metadata
[oota-llvm.git] / test / Transforms / LoopVectorize / vect.omp.persistence.ll
1 ; RUN: opt < %s -O2 -force-vector-unroll=2 -force-vector-width=4 -debug-only=loop-vectorize -stats -S 2>&1 | FileCheck %s
2
3 ; Loop from "rotated"
4 ; CHECK: LV: Loop hints: force=enabled
5 ; Loop from "nonrotated"
6 ; CHECK: LV: Loop hints: force=enabled
7 ; No more loops in the module
8 ; CHECK-NOT: LV: Loop hints: force=
9 ; In total only 1 loop should be rotated.
10 ; CHECK: 1 loop-rotate
11
12 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"
13 target triple = "x86_64-unknown-linux-gnu"
14
15 ; See http://reviews.llvm.org/D3348 for details.
16
17 ;
18 ; Test #1
19 ;
20 ; Ensure that "llvm.vectorizer.enable" metadata was not lost prior to LoopVectorize pass.
21 ; In past LoopRotate was clearing that metadata.
22 ;
23 ; The source C code is:
24 ; void rotated(float *a, int size)
25 ; {
26 ;   int t = 0;
27 ;   #pragma omp simd
28 ;   for (int i = 0; i < size; ++i) {
29 ;     a[i] = a[i-5] * a[i+2];
30 ;     ++t;
31 ;   }
32 ;}
33
34 define void @rotated(float* nocapture %a, i64 %size) {
35 entry:
36   %cmp1 = icmp sgt i64 %size, 0
37   br i1 %cmp1, label %for.header, label %for.end
38
39 for.header:
40   %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
41   %cmp2 = icmp sgt i64 %indvars.iv, %size
42   br i1 %cmp2, label %for.end, label %for.body
43
44 for.body:
45
46   %0 = add nsw i64 %indvars.iv, -5
47   %arrayidx = getelementptr inbounds float* %a, i64 %0
48   %1 = load float* %arrayidx, align 4, !llvm.mem.parallel_loop_access !1
49   %2 = add nsw i64 %indvars.iv, 2
50   %arrayidx2 = getelementptr inbounds float* %a, i64 %2
51   %3 = load float* %arrayidx2, align 4, !llvm.mem.parallel_loop_access !1
52   %mul = fmul float %1, %3
53   %arrayidx4 = getelementptr inbounds float* %a, i64 %indvars.iv
54   store float %mul, float* %arrayidx4, align 4, !llvm.mem.parallel_loop_access !1
55
56   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
57   br label %for.header, !llvm.loop !1
58
59 for.end:
60   ret void
61 }
62
63 !1 = metadata !{metadata !1, metadata !2}
64 !2 = metadata !{metadata !"llvm.vectorizer.enable", i1 true}
65
66 ;
67 ; Test #2
68 ;
69 ; Ensure that "llvm.vectorizer.enable" metadata was not lost even
70 ; if loop was not rotated (see http://reviews.llvm.org/D3348#comment-4).
71 ;
72 define i32 @nonrotated(i32 %a) {
73 entry:
74   br label %loop_cond
75 loop_cond:
76   %indx = phi i32 [ 1, %entry ], [ %inc, %loop_inc ]
77   %cmp = icmp ne i32 %indx, %a
78   br i1 %cmp, label %return, label %loop_inc
79 loop_inc:
80   %inc = add i32 %indx, 1
81   br label %loop_cond, !llvm.loop !3
82 return:
83   ret i32 0
84 }
85
86 !3 = metadata !{metadata !3, metadata !4}
87 !4 = metadata !{metadata !"llvm.vectorizer.enable", i1 true}