flang/test/Lower/OpenMP/master.f90

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