xref: /llvm-project/polly/test/ScopInfo/switch-7.ll (revision e1f056f692d869708c1898d9d65a69ac5584a0ed)
1; RUN: opt %loadNPMPolly '-passes=print<polly-function-scops>' -disable-output < %s 2>&1 | FileCheck %s
2; RUN: opt %loadNPMPolly '-passes=print<polly-ast>' -disable-output < %s 2>&1 | FileCheck %s --check-prefix=AST
3;
4;    void f(int *A, int c, int N) {
5;      switch (c) {
6;      case -1: {
7;        for (int j = N; j > 0; j--)
8;          A[j] += A[j - 1];
9;        break;
10;      }
11;      case 1: {
12;        for (int j = 1; j <= N; j++)
13;          A[j] += A[j - 1];
14;        break;
15;      }
16;      }
17;    }
18
19; CHECK:      Statements {
20; CHECK-NEXT:     Stmt_for_body
21; CHECK-NEXT:         Domain :=
22; CHECK-NEXT:             [c, N] -> { Stmt_for_body[i0] : c = -1 and 0 <= i0 < N };
23; CHECK-NEXT:         Schedule :=
24; CHECK-NEXT:             [c, N] -> { Stmt_for_body[i0] -> [1, i0] };
25; CHECK-NEXT:         ReadAccess :=    [Reduction Type: NONE] [Scalar: 0]
26; CHECK-NEXT:             [c, N] -> { Stmt_for_body[i0] -> MemRef_A[-1 + N - i0] };
27; CHECK-NEXT:         ReadAccess :=    [Reduction Type: NONE] [Scalar: 0]
28; CHECK-NEXT:             [c, N] -> { Stmt_for_body[i0] -> MemRef_A[N - i0] };
29; CHECK-NEXT:         MustWriteAccess :=    [Reduction Type: NONE] [Scalar: 0]
30; CHECK-NEXT:             [c, N] -> { Stmt_for_body[i0] -> MemRef_A[N - i0] };
31; CHECK-NEXT:     Stmt_for_body_7
32; CHECK-NEXT:         Domain :=
33; CHECK-NEXT:             [c, N] -> { Stmt_for_body_7[i0] : c = 1 and 0 <= i0 < N };
34; CHECK-NEXT:         Schedule :=
35; CHECK-NEXT:             [c, N] -> { Stmt_for_body_7[i0] -> [0, i0] };
36; CHECK-NEXT:         ReadAccess :=    [Reduction Type: NONE] [Scalar: 0]
37; CHECK-NEXT:             [c, N] -> { Stmt_for_body_7[i0] -> MemRef_A[i0] };
38; CHECK-NEXT:         ReadAccess :=    [Reduction Type: NONE] [Scalar: 0]
39; CHECK-NEXT:             [c, N] -> { Stmt_for_body_7[i0] -> MemRef_A[1 + i0] };
40; CHECK-NEXT:         MustWriteAccess :=    [Reduction Type: NONE] [Scalar: 0]
41; CHECK-NEXT:             [c, N] -> { Stmt_for_body_7[i0] -> MemRef_A[1 + i0] };
42; CHECK-NEXT: }
43
44; AST:      if (1)
45;
46; AST:          if (c == 1) {
47; AST-NEXT:       for (int c0 = 0; c0 < N; c0 += 1)
48; AST-NEXT:         Stmt_for_body_7(c0);
49; AST-NEXT:     } else if (c == -1)
50; AST-NEXT:       for (int c0 = 0; c0 < N; c0 += 1)
51; AST-NEXT:         Stmt_for_body(c0);
52;
53; AST:      else
54; AST-NEXT:     {  /* original code */ }
55
56target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
57
58define void @f(ptr %A, i32 %c, i32 %N) {
59entry:
60  br label %entry.split
61
62entry.split:
63  switch i32 %c, label %sw.epilog [
64    i32 -1, label %sw.bb
65    i32 1, label %sw.bb.3
66  ]
67
68sw.bb:                                            ; preds = %entry
69  %tmp = sext i32 %N to i64
70  br label %for.cond
71
72for.cond:                                         ; preds = %for.inc, %sw.bb
73  %indvars.iv = phi i64 [ %indvars.iv.next, %for.inc ], [ %tmp, %sw.bb ]
74  %j.0 = phi i32 [ %N, %sw.bb ], [ %dec, %for.inc ]
75  %cmp = icmp sgt i64 %indvars.iv, 0
76  br i1 %cmp, label %for.body, label %for.end
77
78for.body:                                         ; preds = %for.cond
79  %sub = add nsw i32 %j.0, -1
80  %idxprom = sext i32 %sub to i64
81  %arrayidx = getelementptr inbounds i32, ptr %A, i64 %idxprom
82  %tmp6 = load i32, ptr %arrayidx, align 4
83  %arrayidx2 = getelementptr inbounds i32, ptr %A, i64 %indvars.iv
84  %tmp7 = load i32, ptr %arrayidx2, align 4
85  %add = add nsw i32 %tmp7, %tmp6
86  store i32 %add, ptr %arrayidx2, align 4
87  br label %for.inc
88
89for.inc:                                          ; preds = %for.body
90  %dec = add nsw i32 %j.0, -1
91  %indvars.iv.next = add nsw i64 %indvars.iv, -1
92  br label %for.cond
93
94for.end:                                          ; preds = %for.cond
95  br label %sw.epilog
96
97sw.bb.3:                                          ; preds = %entry
98  %tmp8 = sext i32 %N to i64
99  br label %for.cond.5
100
101for.cond.5:                                       ; preds = %for.inc.14, %sw.bb.3
102  %indvars.iv3 = phi i64 [ %indvars.iv.next4, %for.inc.14 ], [ 1, %sw.bb.3 ]
103  %cmp6 = icmp sgt i64 %indvars.iv3, %tmp8
104  br i1 %cmp6, label %for.end.15, label %for.body.7
105
106for.body.7:                                       ; preds = %for.cond.5
107  %tmp9 = add nsw i64 %indvars.iv3, -1
108  %arrayidx10 = getelementptr inbounds i32, ptr %A, i64 %tmp9
109  %tmp10 = load i32, ptr %arrayidx10, align 4
110  %arrayidx12 = getelementptr inbounds i32, ptr %A, i64 %indvars.iv3
111  %tmp11 = load i32, ptr %arrayidx12, align 4
112  %add13 = add nsw i32 %tmp11, %tmp10
113  store i32 %add13, ptr %arrayidx12, align 4
114  br label %for.inc.14
115
116for.inc.14:                                       ; preds = %for.body.7
117  %indvars.iv.next4 = add nuw nsw i64 %indvars.iv3, 1
118  br label %for.cond.5
119
120for.end.15:                                       ; preds = %for.cond.5
121  br label %sw.epilog
122
123sw.epilog:                                        ; preds = %for.end.15, %for.end, %entry
124  ret void
125}
126