1f734b0d17
This patch adds private/firstprivate support for sections construct. For
a source like the following:
```
!$omp sections private(x) firstprivate(y)
!$omp section
<block of code>
!$omp section
<block of code>
!$omp end sections
```
...privatization proceeds to privatize `x` and `y` accordingly
inside each of the generated `omp.section` operations.
Reviewed By: peixin
Differential Revision: https://reviews.llvm.org/D131463
111 lines
4.9 KiB
Fortran
111 lines
4.9 KiB
Fortran
! This test checks the lowering of OpenMP sections construct with several clauses present
|
|
|
|
! RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s
|
|
|
|
!CHECK: func @_QQmain() {
|
|
!CHECK: %[[COUNT:.*]] = fir.address_of(@_QFEcount) : !fir.ref<i32>
|
|
!CHECK: %[[ETA:.*]] = fir.alloca f32 {bindc_name = "eta", uniq_name = "_QFEeta"}
|
|
!CHECK: %[[CONST_1:.*]] = arith.constant 1 : i32
|
|
!CHECK: omp.sections allocate(%[[CONST_1]] : i32 -> %0 : !fir.ref<i32>) {
|
|
!CHECK: omp.section {
|
|
!CHECK: %[[PRIVATE_ETA:.*]] = fir.alloca f32 {bindc_name = "eta", pinned, uniq_name = "_QFEeta"}
|
|
!CHECK: %[[PRIVATE_DOUBLE_COUNT:.*]] = fir.alloca i32 {bindc_name = "double_count", pinned, uniq_name = "_QFEdouble_count"}
|
|
!CHECK: %[[const:.*]] = arith.constant 5 : i32
|
|
!CHECK: fir.store %[[const]] to %[[COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[temp_count:.*]] = fir.load %[[COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[temp_double_count:.*]] = fir.load %[[PRIVATE_DOUBLE_COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[result:.*]] = arith.muli %[[temp_count]], %[[temp_double_count]] : i32
|
|
!CHECK: {{.*}} = fir.convert %[[result]] : (i32) -> f32
|
|
!CHECK: fir.store {{.*}} to %[[PRIVATE_ETA]] : !fir.ref<f32>
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.section {
|
|
!CHECK: %[[PRIVATE_ETA:.*]] = fir.alloca f32 {bindc_name = "eta", pinned, uniq_name = "_QFEeta"}
|
|
!CHECK: %[[PRIVATE_DOUBLE_COUNT:.*]] = fir.alloca i32 {bindc_name = "double_count", pinned, uniq_name = "_QFEdouble_count"}
|
|
!CHECK: %[[temp:.*]] = fir.load %[[PRIVATE_DOUBLE_COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[const:.*]] = arith.constant 1 : i32
|
|
!CHECK: %[[result:.*]] = arith.addi %[[temp]], %[[const]] : i32
|
|
!CHECK: fir.store %[[result]] to %[[PRIVATE_DOUBLE_COUNT]] : !fir.ref<i32>
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.section {
|
|
!CHECK: %[[PRIVATE_ETA:.*]] = fir.alloca f32 {bindc_name = "eta", pinned, uniq_name = "_QFEeta"}
|
|
!CHECK: %[[PRIVATE_DOUBLE_COUNT:.*]] = fir.alloca i32 {bindc_name = "double_count", pinned, uniq_name = "_QFEdouble_count"}
|
|
!CHECK: %[[temp:.*]] = fir.load %[[PRIVATE_ETA]] : !fir.ref<f32>
|
|
!CHECK: %[[const:.*]] = arith.constant 7.000000e+00 : f32
|
|
!CHECK: %[[result:.*]] = arith.subf %[[temp]], %[[const]] : f32
|
|
!CHECK: fir.store %[[result]] to %[[PRIVATE_ETA]] : !fir.ref<f32>
|
|
!CHECK: {{.*}} = fir.load %[[COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[temp_count:.*]] = fir.convert {{.*}} : (i32) -> f32
|
|
!CHECK: %[[temp_eta:.*]] = fir.load %[[PRIVATE_ETA]] : !fir.ref<f32>
|
|
!CHECK: {{.*}} = arith.mulf %[[temp_count]], %[[temp_eta]] : f32
|
|
!CHECK: %[[result:.*]] = fir.convert {{.*}} : (f32) -> i32
|
|
!CHECK: fir.store %[[result]] to %[[COUNT]] : !fir.ref<i32>
|
|
!CHECK: {{.*}} = fir.load %[[COUNT]] : !fir.ref<i32>
|
|
!CHECK: %[[temp_count:.*]] = fir.convert {{.*}} : (i32) -> f32
|
|
!CHECK: %[[temp_eta:.*]] = fir.load %[[PRIVATE_ETA]] : !fir.ref<f32>
|
|
!CHECK: {{.*}} = arith.subf %[[temp_count]], %[[temp_eta]] : f32
|
|
!CHECK: %[[result:.*]] = fir.convert {{.*}} : (f32) -> i32
|
|
!CHECK: fir.store %[[result]] to %[[PRIVATE_DOUBLE_COUNT]] : !fir.ref<i32>
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.sections nowait {
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: return
|
|
!CHECK: }
|
|
|
|
program sample
|
|
use omp_lib
|
|
integer :: count = 0, double_count = 1
|
|
!$omp sections private (eta, double_count) allocate(omp_high_bw_mem_alloc: count)
|
|
!$omp section
|
|
count = 1 + 4
|
|
eta = count * double_count
|
|
!$omp section
|
|
double_count = double_count + 1
|
|
!$omp section
|
|
eta = eta - 7
|
|
count = count * eta
|
|
double_count = count - eta
|
|
!$omp end sections
|
|
|
|
!$omp sections
|
|
!$omp end sections nowait
|
|
end program sample
|
|
|
|
!CHECK: func @_QPfirstprivate(%[[ARG:.*]]: !fir.ref<f32> {fir.bindc_name = "alpha"}) {
|
|
!CHECK: omp.sections {
|
|
!CHECK: omp.section {
|
|
!CHECK: %[[PRIVATE_ALPHA:.*]] = fir.alloca f32 {bindc_name = "alpha", pinned, uniq_name = "_QFfirstprivateEalpha"}
|
|
!CHECK: %[[temp:.*]] = fir.load %[[ARG]] : !fir.ref<f32>
|
|
!CHECK: fir.store %[[temp]] to %[[PRIVATE_ALPHA]] : !fir.ref<f32>
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.sections {
|
|
!CHECK: omp.section {
|
|
!CHECK: %[[PRIVATE_VAR:.*]] = fir.load %[[ARG]] : !fir.ref<f32>
|
|
!CHECK: %[[CONSTANT:.*]] = arith.constant 5.000000e+00 : f32
|
|
!CHECK: %[[PRIVATE_VAR_2:.*]] = arith.mulf %[[PRIVATE_VAR]], %[[CONSTANT]] : f32
|
|
!CHECK: fir.store %[[PRIVATE_VAR_2]] to %[[ARG]] : !fir.ref<f32>
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: omp.terminator
|
|
!CHECK: }
|
|
!CHECK: return
|
|
!CHECK: }
|
|
|
|
subroutine firstprivate(alpha)
|
|
real :: alpha
|
|
!$omp sections firstprivate(alpha)
|
|
!$omp end sections
|
|
|
|
!$omp sections
|
|
alpha = alpha * 5
|
|
!$omp end sections
|
|
end subroutine
|