llvm-project/flang/test/Lower/OpenMP/parallel-sections.f90
Shraiysh Vaishay 1fe4b968c5 [flang][OpenMP] Added parallel sections translation
This patch adds translation for parallel sections from PFT to MLIR.

Reviewed By: kiranchandramohan, NimishMishra

Differential Revision: https://reviews.llvm.org/D122464
2022-03-29 11:16:38 +05:30

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 allocate(%[[allocator]] : i32 -> %{{.*}} : !fir.ref<i32>) {
!OMPDialect: omp.sections {
!$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