flang/test/Lower/OpenMP/master.f90

   1 !RUN: %flang_fc1 -emit-hlfir -fopenmp %s -o - | FileCheck %s
   2
   3 !===============================================================================
   4 ! parallel construct with function call which has master construct internally
   5 !===============================================================================
   6 !CHECK-LABEL: func @_QPomp_master
   7 subroutine omp_master()
   8
   9 !CHECK: omp.master  {
  10 !$omp master
  11
  12     !CHECK: fir.call @_QPmaster() {{.*}}: () -> ()
  13     call master()
  14
  15 !CHECK: omp.terminator
  16 !$omp end master
  17
  18 end subroutine omp_master
  19
  20 !CHECK-LABEL: func @_QPparallel_function_master
  21 subroutine parallel_function_master()
  22
  23 !CHECK: omp.parallel {
  24 !$omp parallel
  25
  26     !CHECK: fir.call @_QPfoo() {{.*}}: () -> ()
  27     call foo()
  28
  29 !CHECK: omp.terminator
  30 !$omp end parallel
  31
  32 end subroutine parallel_function_master
  33
  34 !===============================================================================
  35 ! master construct nested inside parallel construct
  36 !===============================================================================
  37
  38 !CHECK-LABEL: func @_QPomp_parallel_master
  39 subroutine omp_parallel_master()
  40
  41 !CHECK: omp.parallel {
  42 !$omp parallel
  43     !CHECK: fir.call @_QPparallel() {{.*}}: () -> ()
  44     call parallel()
  45
  46 !CHECK: omp.master {
  47 !$omp master
  48
  49     !CHECK: fir.call @_QPparallel_master() {{.*}}: () -> ()
  50     call parallel_master()
  51
  52 !CHECK: omp.terminator
  53 !$omp end master
  54
  55 !CHECK: omp.terminator
  56 !$omp end parallel
  57
  58 end subroutine omp_parallel_master
  59
  60 !===============================================================================
  61 ! master construct nested inside parallel construct with conditional flow
  62 !===============================================================================
  63
  64 !CHECK-LABEL: func @_QPomp_master_parallel
  65 subroutine omp_master_parallel()
  66     integer :: alpha, beta, gama
  67     alpha = 4
  68     beta = 5
  69     gama = 6
  70
  71 !CHECK: omp.master {
  72 !$omp master
  73
  74     !CHECK: %{{.*}} = fir.load %{{.*}}
  75     !CHECK: %{{.*}} = fir.load %{{.*}}
  76     !CHECK: %[[RESULT:.*]] = arith.cmpi sge, %{{.*}}, %{{.*}}
  77     !CHECK: fir.if %[[RESULT]] {
  78     if (alpha .ge. gama) then
  79
  80 !CHECK: omp.parallel {
  81 !$omp parallel
  82         !CHECK: fir.call @_QPinside_if_parallel() {{.*}}: () -> ()
  83         call inside_if_parallel()
  84
  85 !CHECK: omp.terminator
  86 !$omp end parallel
  87
  88         !CHECK: %{{.*}} = fir.load %{{.*}}
  89         !CHECK: %{{.*}} = fir.load %{{.*}}
  90         !CHECK: %{{.*}} = arith.addi %{{.*}}, %{{.*}}
  91         !CHECK: hlfir.assign %{{.*}} to %{{.*}}#0 : i32, !fir.ref<i32>
  92         beta = alpha + gama
  93     end if
  94     !CHECK: else
  95
  96 !CHECK: omp.terminator
  97 !$omp end master
  98
  99 end subroutine omp_master_parallel