You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.8 KiB
66 lines
1.8 KiB
// 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
|
|
}
|