4202d69d9e
When parallel is used in a combined construct, then use a separate function to create the parallel operation. It handles the parallel specific clauses and leaves the rest for handling at the inner operations. Reviewed By: peixin, shraiysh Differential Revision: https://reviews.llvm.org/D125465 Co-authored-by: Sourabh Singh Tomar <SourabhSingh.Tomar@amd.com> Co-authored-by: Eric Schweitz <eschweitz@nvidia.com> Co-authored-by: Valentin Clement <clementval@gmail.com> Co-authored-by: Nimish Mishra <neelam.nimish@gmail.com>
58 lines
2.1 KiB
Fortran
58 lines
2.1 KiB
Fortran
!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s --check-prefixes="FIRDialect,OMPDialect"
|
|
!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | fir-opt --cfg-conversion | fir-opt --fir-to-llvm-ir | FileCheck %s --check-prefixes="OMPDialect,LLVMDialect"
|
|
|
|
!===============================================================================
|
|
! Parallel sections construct
|
|
!===============================================================================
|
|
|
|
!FIRDialect: func @_QPomp_parallel_sections
|
|
subroutine omp_parallel_sections(x, y)
|
|
integer, intent(inout) :: x, y
|
|
!OMPDialect: omp.parallel {
|
|
!OMPDialect: omp.sections {
|
|
!$omp parallel sections
|
|
!OMPDialect: omp.section {
|
|
!$omp section
|
|
!FIRDialect: fir.load
|
|
!FIRDialect: arith.addi
|
|
!FIRDialect: fir.store
|
|
x = x + 12
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.section {
|
|
!$omp section
|
|
!FIRDialect: fir.load
|
|
!FIRDialect: arith.subi
|
|
!FIRDialect: fir.store
|
|
y = y - 5
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.terminator
|
|
!$omp end parallel sections
|
|
end subroutine omp_parallel_sections
|
|
|
|
!===============================================================================
|
|
! Parallel sections construct with allocate clause
|
|
!===============================================================================
|
|
|
|
!FIRDialect: func @_QPomp_parallel_sections
|
|
subroutine omp_parallel_sections_allocate(x, y)
|
|
use omp_lib
|
|
integer, intent(inout) :: x, y
|
|
!FIRDialect: %[[allocator:.*]] = arith.constant 1 : i32
|
|
!LLVMDialect: %[[allocator:.*]] = llvm.mlir.constant(1 : i32) : i32
|
|
!OMPDialect: omp.parallel {
|
|
!OMPDialect: omp.sections allocate(%[[allocator]] : i32 -> %{{.*}} : !fir.ref<i32>) {
|
|
!$omp parallel sections allocate(omp_high_bw_mem_alloc: x)
|
|
!OMPDialect: omp.section {
|
|
!$omp section
|
|
x = x + 12
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.section {
|
|
!$omp section
|
|
y = y + 5
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.terminator
|
|
!OMPDialect: omp.terminator
|
|
!$omp end parallel sections
|
|
end subroutine omp_parallel_sections_allocate
|