8b951e64d9
Add support for lowering the schedule modifiers (simd, monotonic, non-monotonic) in worksharing loops. Note: This is part of upstreaming from the fir-dev branch of https://github.com/flang-compiler/f18-llvm-project. Reviewed By: peixin Differential Revision: https://reviews.llvm.org/D127311 Co-authored-by: Mats Petersson <mats.petersson@arm.com> Co-authored-by: Jean Perier <jperier@nvidia.com> Co-authored-by: Eric Schweitz <eschweitz@nvidia.com> Co-authored-by: V Donaldson <vdonaldson@nvidia.com>
35 lines
1.1 KiB
Fortran
35 lines
1.1 KiB
Fortran
! This test checks lowering of OpenMP DO Directive(Worksharing) with
|
|
! simd schedule modifier.
|
|
|
|
! RUN: bbc -fopenmp -emit-fir %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 for (%[[I:.*]]) : i32 = (%[[WS_LB]]) to (%[[WS_UB]]) inclusive step (%[[WS_STEP]])
|
|
!CHECK: fir.store %[[I]] to %[[STORE:.*]] : !fir.ref<i32>
|
|
|
|
do i=1, 9
|
|
print*, i
|
|
!CHECK: %[[RTBEGIN:.*]] = fir.call @_FortranAioBeginExternalListOutput
|
|
!CHECK: %[[LOAD:.*]] = fir.load %[[STORE]] : !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: }
|
|
|
|
!$OMP END DO NOWAIT
|
|
!$OMP END PARALLEL
|
|
end
|