| // RUN: mlir-opt -convert-scf-to-openmp %s | FileCheck %s |
| |
| // CHECK-LABEL: @parallel |
| func @parallel(%arg0: index, %arg1: index, %arg2: index, |
| %arg3: index, %arg4: index, %arg5: index) { |
| // CHECK: omp.parallel { |
| // CHECK: "omp.wsloop"({{.*}}) ( { |
| scf.parallel (%i, %j) = (%arg0, %arg1) to (%arg2, %arg3) step (%arg4, %arg5) { |
| // CHECK: test.payload |
| "test.payload"(%i, %j) : (index, index) -> () |
| // CHECK: omp.yield |
| // CHECK: } |
| } |
| // CHECK: omp.terminator |
| // CHECK: } |
| return |
| } |
| |
| // CHECK-LABEL: @nested_loops |
| func @nested_loops(%arg0: index, %arg1: index, %arg2: index, |
| %arg3: index, %arg4: index, %arg5: index) { |
| // CHECK: omp.parallel { |
| // CHECK: "omp.wsloop"({{.*}}) ( { |
| // CHECK-NOT: omp.parallel |
| scf.parallel (%i) = (%arg0) to (%arg2) step (%arg4) { |
| // CHECK: "omp.wsloop"({{.*}}) ( { |
| scf.parallel (%j) = (%arg1) to (%arg3) step (%arg5) { |
| // CHECK: test.payload |
| "test.payload"(%i, %j) : (index, index) -> () |
| // CHECK: omp.yield |
| // CHECK: } |
| } |
| // CHECK: omp.yield |
| // CHECK: } |
| } |
| // CHECK: omp.terminator |
| // CHECK: } |
| return |
| } |
| |
| func @adjacent_loops(%arg0: index, %arg1: index, %arg2: index, |
| %arg3: index, %arg4: index, %arg5: index) { |
| // CHECK: omp.parallel { |
| // CHECK: "omp.wsloop"({{.*}}) ( { |
| scf.parallel (%i) = (%arg0) to (%arg2) step (%arg4) { |
| // CHECK: test.payload1 |
| "test.payload1"(%i) : (index) -> () |
| // CHECK: omp.yield |
| // CHECK: } |
| } |
| // CHECK: omp.terminator |
| // CHECK: } |
| |
| // CHECK: omp.parallel { |
| // CHECK: "omp.wsloop"({{.*}}) ( { |
| scf.parallel (%j) = (%arg1) to (%arg3) step (%arg5) { |
| // CHECK: test.payload2 |
| "test.payload2"(%j) : (index) -> () |
| // CHECK: omp.yield |
| // CHECK: } |
| } |
| // CHECK: omp.terminator |
| // CHECK: } |
| return |
| } |