xref: /llvm-project/llvm/test/Transforms/LoopUnroll/X86/partial.ll (revision ef992b60798b6cd2c50b25351bfc392e319896b7)
1; RUN: opt < %s -S -passes=loop-unroll -mcpu=nehalem | FileCheck %s
2; RUN: opt < %s -S -passes=loop-unroll -unroll-runtime=0 | FileCheck -check-prefix=CHECK-NOUNRL %s
3target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
4target triple = "x86_64-unknown-linux-gnu"
5
6define void @foo(ptr noalias nocapture readnone %ip, double %alpha, ptr noalias nocapture %a, ptr noalias nocapture readonly %b) #0 {
7entry:
8  br label %vector.body
9
10vector.body:                                      ; preds = %vector.body, %entry
11  %index = phi i64 [ 0, %entry ], [ %index.next, %vector.body ]
12  %0 = getelementptr inbounds double, ptr %b, i64 %index
13  %wide.load = load <2 x double>, ptr %0, align 8
14  %.sum9 = or i64 %index, 2
15  %1 = getelementptr double, ptr %b, i64 %.sum9
16  %wide.load8 = load <2 x double>, ptr %1, align 8
17  %2 = fadd <2 x double> %wide.load, <double 1.000000e+00, double 1.000000e+00>
18  %3 = fadd <2 x double> %wide.load8, <double 1.000000e+00, double 1.000000e+00>
19  %4 = getelementptr inbounds double, ptr %a, i64 %index
20  store <2 x double> %2, ptr %4, align 8
21  %.sum10 = or i64 %index, 2
22  %5 = getelementptr double, ptr %a, i64 %.sum10
23  store <2 x double> %3, ptr %5, align 8
24  %index.next = add i64 %index, 4
25  %6 = icmp eq i64 %index.next, 1600
26  br i1 %6, label %for.end, label %vector.body
27
28; FIXME: We should probably unroll this loop by a factor of 2, but the cost
29; model needs to be fixed to account for instructions likely to be folded
30; as part of an addressing mode.
31; CHECK-LABEL: @foo
32; CHECK-NOUNRL-LABEL: @foo
33
34for.end:                                          ; preds = %vector.body
35  ret void
36}
37
38define void @bar(ptr noalias nocapture readnone %ip, double %alpha, ptr noalias nocapture %a, ptr noalias nocapture readonly %b) #0 {
39entry:
40  br label %vector.body
41
42vector.body:                                      ; preds = %vector.body, %entry
43  %index = phi i64 [ 0, %entry ], [ %index.next, %vector.body ]
44  %v0 = getelementptr inbounds double, ptr %b, i64 %index
45  %wide.load = load <2 x double>, ptr %v0, align 8
46  %v4 = fadd <2 x double> %wide.load, <double 1.000000e+00, double 1.000000e+00>
47  %v5 = fmul <2 x double> %v4, <double 8.000000e+00, double 8.000000e+00>
48  %v6 = getelementptr inbounds double, ptr %a, i64 %index
49  store <2 x double> %v5, ptr %v6, align 8
50  %index.next = add i64 %index, 2
51  %v10 = icmp eq i64 %index.next, 1600
52  br i1 %v10, label %for.end, label %vector.body
53
54; FIXME: We should probably unroll this loop by a factor of 2, but the cost
55; model needs to first to fixed to account for instructions likely to be folded
56; as part of an addressing mode.
57
58; CHECK-LABEL: @bar
59; CHECK: fadd
60; CHECK-NEXT: fmul
61; CHECK: fadd
62; CHECK-NEXT: fmul
63
64; CHECK-NOUNRL-LABEL: @bar
65; CHECK-NOUNRL: fadd
66; CHECK-NOUNRL-NEXT: fmul
67; CHECK-NOUNRL-NOT: fadd
68
69for.end:                                          ; preds = %vector.body
70  ret void
71}
72
73define zeroext i16 @test1(ptr nocapture readonly %arr, i32 %n) #0 {
74entry:
75  %cmp25 = icmp eq i32 %n, 0
76  br i1 %cmp25, label %for.end, label %for.body
77
78for.body:                                         ; preds = %entry, %for.body
79  %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
80  %reduction.026 = phi i16 [ %add14, %for.body ], [ 0, %entry ]
81  %arrayidx = getelementptr inbounds i16, ptr %arr, i64 %indvars.iv
82  %0 = load i16, ptr %arrayidx, align 2
83  %mul = shl i16 %0, 1
84  %add = add i16 %mul, %reduction.026
85  %sext = mul i64 %indvars.iv, 12884901888
86  %idxprom3 = ashr exact i64 %sext, 32
87  %arrayidx4 = getelementptr inbounds i16, ptr %arr, i64 %idxprom3
88  %1 = load i16, ptr %arrayidx4, align 2
89  %mul2 = shl i16 %1, 1
90  %add7 = add i16 %add, %mul2
91  %sext28 = mul i64 %indvars.iv, 21474836480
92  %idxprom10 = ashr exact i64 %sext28, 32
93  %arrayidx11 = getelementptr inbounds i16, ptr %arr, i64 %idxprom10
94  %2 = load i16, ptr %arrayidx11, align 2
95  %mul3 = shl i16 %2, 1
96  %add14 = add i16 %add7, %mul3
97  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
98  %lftr.wideiv = trunc i64 %indvars.iv.next to i32
99  %exitcond = icmp eq i32 %lftr.wideiv, %n
100  br i1 %exitcond, label %for.end, label %for.body
101
102for.end:                                          ; preds = %for.body, %entry
103  %reduction.0.lcssa = phi i16 [ 0, %entry ], [ %add14, %for.body ]
104  ret i16 %reduction.0.lcssa
105
106; This loop is too large to be partially unrolled (size=16)
107
108; CHECK-LABEL: @test1
109; CHECK: br
110; CHECK: br
111; CHECK: br
112; CHECK: br
113; CHECK-NOT: br
114
115; CHECK-NOUNRL-LABEL: @test1
116; CHECK-NOUNRL: br
117; CHECK-NOUNRL: br
118; CHECK-NOUNRL: br
119; CHECK-NOUNRL: br
120; CHECK-NOUNRL-NOT: br
121}
122
123attributes #0 = { nounwind uwtable }
124
125