ca4dbc2718
This patch updates lowering from PFT to MLIR of workshare loops to follow the loop wrapper approach. Unit tests impacted by this change are also updated. As the last patch of the stack, this should compile and pass unit tests.
38 lines
1.2 KiB
Fortran
38 lines
1.2 KiB
Fortran
! This test checks lowering of OpenMP DO Directive(Worksharing) with
|
|
! simd schedule modifier.
|
|
|
|
! RUN: bbc -fopenmp -emit-hlfir %s -o - | FileCheck %s
|
|
|
|
program wsloop_dynamic
|
|
integer :: i
|
|
!CHECK-LABEL: func @_QQmain()
|
|
|
|
!$OMP PARALLEL
|
|
!CHECK: omp.parallel {
|
|
|
|
!$OMP DO SCHEDULE(simd: runtime)
|
|
!CHECK: %[[WS_LB:.*]] = arith.constant 1 : i32
|
|
!CHECK: %[[WS_UB:.*]] = arith.constant 9 : i32
|
|
!CHECK: %[[WS_STEP:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.wsloop schedule(runtime, simd) nowait {
|
|
!CHECK-NEXT: omp.loop_nest (%[[I:.*]]) : i32 = (%[[WS_LB]]) to (%[[WS_UB]]) inclusive step (%[[WS_STEP]]) {
|
|
!CHECK: fir.store %[[I]] to %[[STORE:.*]]#1 : !fir.ref<i32>
|
|
|
|
do i=1, 9
|
|
print*, i
|
|
!CHECK: %[[RTBEGIN:.*]] = fir.call @_FortranAioBeginExternalListOutput
|
|
!CHECK: %[[LOAD:.*]] = fir.load %[[STORE]]#0 : !fir.ref<i32>
|
|
!CHECK: fir.call @_FortranAioOutputInteger32(%[[RTBEGIN]], %[[LOAD]]) {{.*}}: (!fir.ref<i8>, i32) -> i1
|
|
!CHECK: fir.call @_FortranAioEndIoStatement(%[[RTBEGIN]]) {{.*}}: (!fir.ref<i8>) -> i32
|
|
end do
|
|
!CHECK: omp.yield
|
|
!CHECK: }
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
|
|
!$OMP END DO NOWAIT
|
|
!$OMP END PARALLEL
|
|
end
|