chapel/test/llvm/parallel_loop_access/generation_inside_loop.chpl at 2b7572e264c0b33a374541c054b4c5c272e7e6bb · chapel-lang/chapel · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
proc start_block() { return 1; }
proc if_branch() { return 2; }
proc while_loop() { return 3; }
proc while_body() { return 4; }
proc end_block() { return 5; }

//Check whether we generate parallel_loop_access metadata
//inside basic blocks in the loop like
// - non-parallel-loops inside loop
// - if branches
// - basic block at start of loop
// - basic block at the end of the loop

proc loop (ref A, B, n) {
  //CHECK-LABEL: void @loop
  foreach i in 1..n {
    // CHECK-LABEL: start_block
    start_block();
    // Make sure metadata is generated at start of loop
    // CHECK: !llvm.access.group ![[GROUP1:[0-9]+]]
    A[i] = 3*B[i];
    if(A[i] < B[i]) {
      // CHECK-LABEL: if_branch
      if_branch();
      // Make sure metadata is generated inside branches in loop
      // CHECK: !llvm.access.group ![[GROUP1]]
      A[i] = 3*B[i];
    }

    // CHECK: while_loop
    while_loop();
    while(A[i] < 100) // not a CForLoop so no metadata/loop annotation
    {
      // CHECK: while_body
      while_body();
      // Make sure metadata is generated in loops inside
      // CHECK: !llvm.access.group ![[GROUP1]]
      A[i] = A[i]+3;
    }
    // Make sure metadata is generated at the end of loop
    // CHECK-LABEL: end_block
    end_block();
    // CHECK: !llvm.access.group ![[GROUP1]]
    A[i] = 3*B[i];

    // CHECK: br i1
    // CHECK-SAME: !llvm.loop ![[LOOP1:[0-9]+]]
  }
}
// CHECK: ![[LOOP1]] = distinct !{![[LOOP1]], ![[PA1:[0-9]+]]
// CHECK: ![[PA1]] = !{!"llvm.loop.parallel_accesses",
// CHECK-SAME: ![[GROUP1]]
// CHECK-SAME: }

config const n = 1000;

var A : [1..n] int(32);
var B : [1..n] int(32);

loop(A, B, n);
writeln("Sum of A is ", + reduce A);