e7e5541616
Add a new pass that lowers an `omp.workshare` with its binding `omp.workshare.loop_wrapper` loop nests into other OpenMP constructs that can be lowered to LLVM. More specifically, in order to preserve the sequential execution semantics of the code contained, it wraps portions that needs to be executed on a single thread in `omp.single` blocks, converts code that must be parallelized into `omp.wsloop` nests and inserts the appropriate synchronization.
50 lines
1.5 KiB
MLIR
50 lines
1.5 KiB
MLIR
// RUN: fir-opt --split-input-file --lower-workshare --allow-unregistered-dialect %s | FileCheck %s
|
|
|
|
// Checks that the omp.workshare.loop_wrapper binds to the correct omp.workshare
|
|
|
|
func.func @wsfunc() {
|
|
%c1 = arith.constant 1 : index
|
|
%c42 = arith.constant 42 : index
|
|
omp.parallel {
|
|
omp.workshare nowait {
|
|
omp.parallel {
|
|
omp.workshare nowait {
|
|
omp.workshare.loop_wrapper {
|
|
omp.loop_nest (%arg1) : index = (%c1) to (%c42) inclusive step (%c1) {
|
|
"test.test2"() : () -> ()
|
|
omp.yield
|
|
}
|
|
}
|
|
omp.terminator
|
|
}
|
|
omp.terminator
|
|
}
|
|
omp.terminator
|
|
}
|
|
omp.terminator
|
|
}
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @wsfunc() {
|
|
// CHECK: %[[VAL_0:.*]] = arith.constant 1 : index
|
|
// CHECK: %[[VAL_1:.*]] = arith.constant 42 : index
|
|
// CHECK: omp.parallel {
|
|
// CHECK: omp.single nowait {
|
|
// CHECK: omp.parallel {
|
|
// CHECK: omp.wsloop nowait {
|
|
// CHECK: omp.loop_nest (%[[VAL_2:.*]]) : index = (%[[VAL_0]]) to (%[[VAL_1]]) inclusive step (%[[VAL_0]]) {
|
|
// CHECK: "test.test2"() : () -> ()
|
|
// CHECK: omp.yield
|
|
// CHECK: }
|
|
// CHECK: }
|
|
// CHECK: omp.terminator
|
|
// CHECK: }
|
|
// CHECK: omp.terminator
|
|
// CHECK: }
|
|
// CHECK: omp.terminator
|
|
// CHECK: }
|
|
// CHECK: return
|
|
// CHECK: }
|
|
|