xref: /llvm-project/polly/test/IstAstInfo/reduction_modulo_schedule_multiple_dimensions_5.ll (revision e1f056f692d869708c1898d9d65a69ac5584a0ed)
1; RUN: opt %loadNPMPolly '-passes=polly-import-jscop,print<polly-ast>' -polly-ast-detect-parallel -disable-output < %s | FileCheck %s
2;
3; Verify that only the outer dimension needs privatization
4;
5; CHECK:    #pragma known-parallel reduction
6; CHECK:    for (int c1 = 0; c1 <= 1023; c1 += 1) {
7; CHECK:      if ((c1 + 1) % 2 == 0) {
8; CHECK-NOT:    #pragma simd reduction
9; CHECK:        #pragma simd
10; CHECK:        for (int c3 = -2 * n + 1; c3 <= 0; c3 += 1)
11; CHECK:          Stmt_for_body3(-c3, c1);
12; CHECK:      } else {
13; CHECK-NOT:    #pragma simd reduction
14; CHECK:        #pragma simd
15; CHECK:        for (int c3 = 0; c3 < 2 * n; c3 += 1)
16; CHECK:          Stmt_for_body3(c3, c1);
17; CHECK:      }
18; CHECK:    }
19;
20;    void rmsmd5(int *A, long n) {
21;      for (long i = 0; i < 2 * n; i++)
22;        for (long j = 0; j < 1024; j++)
23;          A[i] += i;
24;    }
25;
26target datalayout = "e-m:e-p:32:32-i64:64-v128:64:128-n32-S64"
27
28define void @rmsmd5(ptr %A, i32 %n) {
29entry:
30  br label %for.cond
31
32for.cond:                                         ; preds = %for.inc4, %entry
33  %i.0 = phi i32 [ 0, %entry ], [ %inc5, %for.inc4 ]
34  %mul = shl nsw i32 %n, 1
35  %cmp = icmp slt i32 %i.0, %mul
36  br i1 %cmp, label %for.body, label %for.end6
37
38for.body:                                         ; preds = %for.cond
39  br label %for.cond1
40
41for.cond1:                                        ; preds = %for.inc, %for.body
42  %j.0 = phi i32 [ 0, %for.body ], [ %inc, %for.inc ]
43  %exitcond = icmp ne i32 %j.0, 1024
44  br i1 %exitcond, label %for.body3, label %for.end
45
46for.body3:                                        ; preds = %for.cond1
47  %arrayidx = getelementptr inbounds i32, ptr %A, i32 %i.0
48  %tmp = load i32, ptr %arrayidx, align 4
49  %add = add nsw i32 %tmp, %i.0
50  store i32 %add, ptr %arrayidx, align 4
51  br label %for.inc
52
53for.inc:                                          ; preds = %for.body3
54  %inc = add nsw i32 %j.0, 1
55  br label %for.cond1
56
57for.end:                                          ; preds = %for.cond1
58  br label %for.inc4
59
60for.inc4:                                         ; preds = %for.end
61  %inc5 = add nsw i32 %i.0, 1
62  br label %for.cond
63
64for.end6:                                         ; preds = %for.cond
65  ret void
66}
67