xref: /llvm-project/mlir/test/Conversion/SCFToOpenMP/scf-to-openmp.mlir (revision 0a17bdfc361400cb511368f2edfc68c0d11e1974) 
1// RUN: mlir-opt -convert-scf-to-openmp='num-threads=4' %s | FileCheck %s
2
3// CHECK-LABEL: @parallel
4func.func @parallel(%arg0: index, %arg1: index, %arg2: index,
5                    %arg3: index, %arg4: index, %arg5: index) {
6  // CHECK: %[[FOUR:.+]] = llvm.mlir.constant(4 : i32) : i32
7  // CHECK: omp.parallel num_threads(%[[FOUR]] : i32) {
8  // CHECK: omp.wsloop {
9  // CHECK: omp.loop_nest (%[[LVAR1:.*]], %[[LVAR2:.*]]) : index = (%arg0, %arg1) to (%arg2, %arg3) step (%arg4, %arg5) {
10  // CHECK: memref.alloca_scope
11  scf.parallel (%i, %j) = (%arg0, %arg1) to (%arg2, %arg3) step (%arg4, %arg5) {
12    // CHECK: "test.payload"(%[[LVAR1]], %[[LVAR2]]) : (index, index) -> ()
13    "test.payload"(%i, %j) : (index, index) -> ()
14    // CHECK:   omp.yield
15    // CHECK: }
16  }
17  // CHECK:   }
18  // CHECK:   omp.terminator
19  // CHECK: }
20  return
21}
22
23// CHECK-LABEL: @nested_loops
24func.func @nested_loops(%arg0: index, %arg1: index, %arg2: index,
25                   %arg3: index, %arg4: index, %arg5: index) {
26  // CHECK: %[[FOUR:.+]] = llvm.mlir.constant(4 : i32) : i32
27  // CHECK: omp.parallel num_threads(%[[FOUR]] : i32) {
28  // CHECK: omp.wsloop {
29  // CHECK: omp.loop_nest (%[[LVAR_OUT1:.*]]) : index = (%arg0) to (%arg2) step (%arg4) {
30  // CHECK: memref.alloca_scope
31  scf.parallel (%i) = (%arg0) to (%arg2) step (%arg4) {
32    // CHECK: omp.parallel
33    // CHECK: omp.wsloop {
34    // CHECK: omp.loop_nest (%[[LVAR_IN1:.*]]) : index = (%arg1) to (%arg3) step (%arg5) {
35    // CHECK: memref.alloca_scope
36    scf.parallel (%j) = (%arg1) to (%arg3) step (%arg5) {
37      // CHECK: "test.payload"(%[[LVAR_OUT1]], %[[LVAR_IN1]]) : (index, index) -> ()
38      "test.payload"(%i, %j) : (index, index) -> ()
39      // CHECK: }
40    }
41    // CHECK:     omp.yield
42    // CHECK:   }
43    // CHECK: }
44  }
45  // CHECK:   }
46  // CHECK:   omp.terminator
47  // CHECK: }
48  return
49}
50
51// CHECK-LABEL: @adjacent_loops
52func.func @adjacent_loops(%arg0: index, %arg1: index, %arg2: index,
53                     %arg3: index, %arg4: index, %arg5: index) {
54  // CHECK: %[[FOUR:.+]] = llvm.mlir.constant(4 : i32) : i32
55  // CHECK: omp.parallel num_threads(%[[FOUR]] : i32) {
56  // CHECK: omp.wsloop {
57  // CHECK: omp.loop_nest (%[[LVAR_AL1:.*]]) : index = (%arg0) to (%arg2) step (%arg4) {
58  // CHECK: memref.alloca_scope
59  scf.parallel (%i) = (%arg0) to (%arg2) step (%arg4) {
60    // CHECK: "test.payload1"(%[[LVAR_AL1]]) : (index) -> ()
61    "test.payload1"(%i) : (index) -> ()
62    // CHECK:   omp.yield
63    // CHECK: }
64  }
65  // CHECK:   }
66  // CHECK:   omp.terminator
67  // CHECK: }
68
69  // CHECK: %[[FOUR:.+]] = llvm.mlir.constant(4 : i32) : i32
70  // CHECK: omp.parallel num_threads(%[[FOUR]] : i32) {
71  // CHECK: omp.wsloop {
72  // CHECK: omp.loop_nest (%[[LVAR_AL2:.*]]) : index = (%arg1) to (%arg3) step (%arg5) {
73  // CHECK: memref.alloca_scope
74  scf.parallel (%j) = (%arg1) to (%arg3) step (%arg5) {
75    // CHECK: "test.payload2"(%[[LVAR_AL2]]) : (index) -> ()
76    "test.payload2"(%j) : (index) -> ()
77    // CHECK:   omp.yield
78    // CHECK: }
79  }
80  // CHECK:   }
81  // CHECK:   omp.terminator
82  // CHECK: }
83  return
84}
85