flang/test/Lower/OpenMP/copyin.f90

   1 ! This test checks lowering of `COPYIN` clause.
   2 ! RUN: bbc -fopenmp -emit-fir %s -o - | FileCheck %s
   3 ! RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s
   4
   5 ! CHECK-LABEL: func.func @_QPcopyin_scalar_array() {
   6 ! CHECK:         %[[VAL_0:.*]] = fir.address_of(@_QFcopyin_scalar_arrayEx1) : !fir.ref<i32>
   7 ! CHECK:         %[[VAL_1:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<i32> -> !fir.ref<i32>
   8 ! CHECK:         %[[VAL_2:.*]] = fir.address_of(@_QFcopyin_scalar_arrayEx2) : !fir.ref<!fir.array<10xi64>>
   9 ! CHECK:         %[[VAL_3:.*]] = arith.constant 10 : index
  10 ! CHECK:         %[[VAL_4:.*]] = omp.threadprivate %[[VAL_2]] : !fir.ref<!fir.array<10xi64>> -> !fir.ref<!fir.array<10xi64>>
  11 ! CHECK:         omp.parallel   {
  12 ! CHECK:           %[[VAL_5:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<i32> -> !fir.ref<i32>
  13 ! CHECK:           %[[VAL_6:.*]] = fir.load %[[VAL_1]] : !fir.ref<i32>
  14 ! CHECK:           fir.store %[[VAL_6]] to %[[VAL_5]] : !fir.ref<i32>
  15 ! CHECK:           %[[VAL_7:.*]] = omp.threadprivate %[[VAL_2]] : !fir.ref<!fir.array<10xi64>> -> !fir.ref<!fir.array<10xi64>>
  16 ! CHECK:           %[[VAL_8:.*]] = fir.shape %[[VAL_3]] : (index) -> !fir.shape<1>
  17 ! CHECK:           %[[VAL_9:.*]] = fir.array_load %[[VAL_7]](%[[VAL_8]]) : (!fir.ref<!fir.array<10xi64>>, !fir.shape<1>) -> !fir.array<10xi64>
  18 ! CHECK:           %[[VAL_10:.*]] = fir.shape %[[VAL_3]] : (index) -> !fir.shape<1>
  19 ! CHECK:           %[[VAL_11:.*]] = fir.array_load %[[VAL_4]](%[[VAL_10]]) : (!fir.ref<!fir.array<10xi64>>, !fir.shape<1>) -> !fir.array<10xi64>
  20 ! CHECK:           %[[VAL_12:.*]] = arith.constant 1 : index
  21 ! CHECK:           %[[VAL_13:.*]] = arith.constant 0 : index
  22 ! CHECK:           %[[VAL_14:.*]] = arith.subi %[[VAL_3]], %[[VAL_12]] : index
  23 ! CHECK:           %[[VAL_15:.*]] = fir.do_loop %[[VAL_16:.*]] = %[[VAL_13]] to %[[VAL_14]] step %[[VAL_12]] unordered iter_args(%[[VAL_17:.*]] = %[[VAL_9]]) -> (!fir.array<10xi64>) {
  24 ! CHECK:             %[[VAL_18:.*]] = fir.array_fetch %[[VAL_11]], %[[VAL_16]] : (!fir.array<10xi64>, index) -> i64
  25 ! CHECK:             %[[VAL_19:.*]] = fir.array_update %[[VAL_17]], %[[VAL_18]], %[[VAL_16]] : (!fir.array<10xi64>, i64, index) -> !fir.array<10xi64>
  26 ! CHECK:             fir.result %[[VAL_19]] : !fir.array<10xi64>
  27 ! CHECK:           }
  28 ! CHECK:           fir.array_merge_store %[[VAL_9]], %[[VAL_20:.*]] to %[[VAL_7]] : !fir.array<10xi64>, !fir.array<10xi64>, !fir.ref<!fir.array<10xi64>>
  29 ! CHECK:           omp.barrier
  30 ! CHECK:           fir.call @_QPsub1(%[[VAL_5]], %[[VAL_7]]) {{.*}}: (!fir.ref<i32>, !fir.ref<!fir.array<10xi64>>) -> ()
  31 ! CHECK:           omp.terminator
  32 ! CHECK:         }
  33 ! CHECK:         return
  34 ! CHECK:       }
  35
  36 subroutine copyin_scalar_array()
  37   integer(kind=4), save :: x1
  38   integer(kind=8), save :: x2(10)
  39   !$omp threadprivate(x1, x2)
  40
  41   !$omp parallel copyin(x1) copyin(x2)
  42     call sub1(x1, x2)
  43   !$omp end parallel
  44
  45 end
  46
  47 ! CHECK-LABEL: func.func @_QPcopyin_char_chararray() {
  48 ! CHECK:         %[[VAL_0:.*]] = fir.address_of(@_QFcopyin_char_chararrayEx3) : !fir.ref<!fir.char<1,5>>
  49 ! CHECK:         %[[VAL_1:.*]] = arith.constant 5 : index
  50 ! CHECK:         %[[VAL_2:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<!fir.char<1,5>> -> !fir.ref<!fir.char<1,5>>
  51 ! CHECK:         %[[VAL_3:.*]] = fir.address_of(@_QFcopyin_char_chararrayEx4) : !fir.ref<!fir.array<10x!fir.char<1,5>>>
  52 ! CHECK:         %[[VAL_4:.*]] = arith.constant 5 : index
  53 ! CHECK:         %[[VAL_5:.*]] = arith.constant 10 : index
  54 ! CHECK:         %[[VAL_6:.*]] = omp.threadprivate %[[VAL_3]] : !fir.ref<!fir.array<10x!fir.char<1,5>>> -> !fir.ref<!fir.array<10x!fir.char<1,5>>>
  55 ! CHECK:         omp.parallel   {
  56 ! CHECK:           %[[VAL_7:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<!fir.char<1,5>> -> !fir.ref<!fir.char<1,5>>
  57 ! CHECK:           %[[VAL_8:.*]] = arith.constant 1 : i64
  58 ! CHECK:           %[[VAL_9:.*]] = fir.convert %[[VAL_1]] : (index) -> i64
  59 ! CHECK:           %[[VAL_10:.*]] = arith.muli %[[VAL_8]], %[[VAL_9]] : i64
  60 ! CHECK:           %[[VAL_11:.*]] = arith.constant false
  61 ! CHECK:           %[[VAL_12:.*]] = fir.convert %[[VAL_7]] : (!fir.ref<!fir.char<1,5>>) -> !fir.ref<i8>
  62 ! CHECK:           %[[VAL_13:.*]] = fir.convert %[[VAL_2]] : (!fir.ref<!fir.char<1,5>>) -> !fir.ref<i8>
  63 ! CHECK:           fir.call @llvm.memmove.p0.p0.i64(%[[VAL_12]], %[[VAL_13]], %[[VAL_10]], %[[VAL_11]]) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64, i1) -> ()
  64 ! CHECK:           %[[VAL_14:.*]] = omp.threadprivate %[[VAL_3]] : !fir.ref<!fir.array<10x!fir.char<1,5>>> -> !fir.ref<!fir.array<10x!fir.char<1,5>>>
  65 ! CHECK:           %[[VAL_15:.*]] = fir.shape %[[VAL_5]] : (index) -> !fir.shape<1>
  66 ! CHECK:           %[[VAL_16:.*]] = fir.array_load %[[VAL_14]](%[[VAL_15]]) : (!fir.ref<!fir.array<10x!fir.char<1,5>>>, !fir.shape<1>) -> !fir.array<10x!fir.char<1,5>>
  67 ! CHECK:           %[[VAL_17:.*]] = fir.shape %[[VAL_5]] : (index) -> !fir.shape<1>
  68 ! CHECK:           %[[VAL_18:.*]] = fir.array_load %[[VAL_6]](%[[VAL_17]]) : (!fir.ref<!fir.array<10x!fir.char<1,5>>>, !fir.shape<1>) -> !fir.array<10x!fir.char<1,5>>
  69 ! CHECK:           %[[VAL_19:.*]] = arith.constant 1 : index
  70 ! CHECK:           %[[VAL_20:.*]] = arith.constant 0 : index
  71 ! CHECK:           %[[VAL_21:.*]] = arith.subi %[[VAL_5]], %[[VAL_19]] : index
  72 ! CHECK:           %[[VAL_22:.*]] = fir.do_loop %[[VAL_23:.*]] = %[[VAL_20]] to %[[VAL_21]] step %[[VAL_19]] unordered iter_args(%[[VAL_24:.*]] = %[[VAL_16]]) -> (!fir.array<10x!fir.char<1,5>>) {
  73 ! CHECK:             %[[VAL_25:.*]] = fir.array_access %[[VAL_18]], %[[VAL_23]] : (!fir.array<10x!fir.char<1,5>>, index) -> !fir.ref<!fir.char<1,5>>
  74 ! CHECK:             %[[VAL_26:.*]] = fir.array_access %[[VAL_24]], %[[VAL_23]] : (!fir.array<10x!fir.char<1,5>>, index) -> !fir.ref<!fir.char<1,5>>
  75 ! CHECK:             %[[VAL_27:.*]] = arith.constant 5 : index
  76 ! CHECK:             %[[VAL_28:.*]] = arith.constant 1 : i64
  77 ! CHECK:             %[[VAL_29:.*]] = fir.convert %[[VAL_27]] : (index) -> i64
  78 ! CHECK:             %[[VAL_30:.*]] = arith.muli %[[VAL_28]], %[[VAL_29]] : i64
  79 ! CHECK:             %[[VAL_31:.*]] = arith.constant false
  80 ! CHECK:             %[[VAL_32:.*]] = fir.convert %[[VAL_26]] : (!fir.ref<!fir.char<1,5>>) -> !fir.ref<i8>
  81 ! CHECK:             %[[VAL_33:.*]] = fir.convert %[[VAL_25]] : (!fir.ref<!fir.char<1,5>>) -> !fir.ref<i8>
  82 ! CHECK:             fir.call @llvm.memmove.p0.p0.i64(%[[VAL_32]], %[[VAL_33]], %[[VAL_30]], %[[VAL_31]]) {{.*}}: (!fir.ref<i8>, !fir.ref<i8>, i64, i1) -> ()
  83 ! CHECK:             %[[VAL_34:.*]] = fir.array_amend %[[VAL_24]], %[[VAL_26]] : (!fir.array<10x!fir.char<1,5>>, !fir.ref<!fir.char<1,5>>) -> !fir.array<10x!fir.char<1,5>>
  84 ! CHECK:             fir.result %[[VAL_34]] : !fir.array<10x!fir.char<1,5>>
  85 ! CHECK:           }
  86 ! CHECK:           fir.array_merge_store %[[VAL_16]], %[[VAL_35:.*]] to %[[VAL_14]] : !fir.array<10x!fir.char<1,5>>, !fir.array<10x!fir.char<1,5>>, !fir.ref<!fir.array<10x!fir.char<1,5>>>
  87 ! CHECK:           omp.barrier
  88 ! CHECK:           %[[VAL_36:.*]] = fir.convert %[[VAL_7]] : (!fir.ref<!fir.char<1,5>>) -> !fir.ref<!fir.char<1,?>>
  89 ! CHECK:           %[[VAL_37:.*]] = fir.emboxchar %[[VAL_36]], %[[VAL_1]] : (!fir.ref<!fir.char<1,?>>, index) -> !fir.boxchar<1>
  90 ! CHECK:           %[[VAL_38:.*]] = fir.convert %[[VAL_14]] : (!fir.ref<!fir.array<10x!fir.char<1,5>>>) -> !fir.ref<!fir.char<1,?>>
  91 ! CHECK:           %[[VAL_39:.*]] = fir.emboxchar %[[VAL_38]], %[[VAL_4]] : (!fir.ref<!fir.char<1,?>>, index) -> !fir.boxchar<1>
  92 ! CHECK:           fir.call @_QPsub2(%[[VAL_37]], %[[VAL_39]]) {{.*}}: (!fir.boxchar<1>, !fir.boxchar<1>) -> ()
  93 ! CHECK:           omp.terminator
  94 ! CHECK:         }
  95 ! CHECK:         return
  96 ! CHECK:       }
  97
  98 subroutine copyin_char_chararray()
  99   character(5), save :: x3, x4(10)
 100   !$omp threadprivate(x3, x4)
 101
 102   !$omp parallel copyin(x3) copyin(x4)
 103     call sub2(x3, x4)
 104   !$omp end parallel
 105
 106 end
 107
 108 ! CHECK-LABEL: func.func @_QPcopyin_derived_type() {
 109 ! CHECK:         %[[VAL_0:.*]] = fir.address_of(@_QFcopyin_derived_typeEx5) : !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>
 110 ! CHECK:         %[[VAL_1:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>> -> !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>
 111 ! CHECK:         omp.parallel   {
 112 ! CHECK:           %[[VAL_2:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>> -> !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>
 113 ! CHECK:           %[[VAL_3:.*]] = fir.load %[[VAL_1]] : !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>
 114 ! CHECK:           fir.store %[[VAL_3]] to %[[VAL_2]] : !fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>
 115 ! CHECK:           omp.barrier
 116 ! CHECK:           fir.call @_QPsub3(%[[VAL_2]]) {{.*}}: (!fir.ref<!fir.type<_QFcopyin_derived_typeTmy_type{t_i:i32,t_arr:!fir.array<5xi32>}>>) -> ()
 117 ! CHECK:           omp.terminator
 118 ! CHECK:         }
 119 ! CHECK:         return
 120 ! CHECK:       }
 121
 122 subroutine copyin_derived_type()
 123   type my_type
 124     integer :: t_i
 125     integer :: t_arr(5)
 126   end type my_type
 127   type(my_type), save :: x5
 128   !$omp threadprivate(x5)
 129
 130   !$omp parallel copyin(x5)
 131     call sub3(x5)
 132   !$omp end parallel
 133
 134 end
 135
 136 ! CHECK-LABEL: func.func @_QPcombined_parallel_worksharing_loop() {
 137 ! CHECK:         %[[VAL_0:.*]] = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFcombined_parallel_worksharing_loopEi"}
 138 ! CHECK:         %[[VAL_1:.*]] = fir.address_of(@_QFcombined_parallel_worksharing_loopEx6) : !fir.ref<i32>
 139 ! CHECK:         %[[VAL_2:.*]] = omp.threadprivate %[[VAL_1]] : !fir.ref<i32> -> !fir.ref<i32>
 140 ! CHECK:         omp.parallel   {
 141 ! CHECK:           %[[VAL_3:.*]] = fir.alloca i32 {adapt.valuebyref, pinned}
 142 ! CHECK:           %[[VAL_4:.*]] = omp.threadprivate %[[VAL_1]] : !fir.ref<i32> -> !fir.ref<i32>
 143 ! CHECK:           %[[VAL_5:.*]] = fir.load %[[VAL_2]] : !fir.ref<i32>
 144 ! CHECK:           fir.store %[[VAL_5]] to %[[VAL_4]] : !fir.ref<i32>
 145 ! CHECK:           omp.barrier
 146 ! CHECK:           %[[VAL_6:.*]] = arith.constant 1 : i32
 147 ! CHECK:           %[[VAL_7:.*]] = fir.load %[[VAL_4]] : !fir.ref<i32>
 148 ! CHECK:           %[[VAL_8:.*]] = arith.constant 1 : i32
 149 ! CHECK:           omp.wsloop   for  (%[[VAL_9:.*]]) : i32 = (%[[VAL_6]]) to (%[[VAL_7]]) inclusive step (%[[VAL_8]]) {
 150 ! CHECK:             fir.store %[[VAL_9]] to %[[VAL_3]] : !fir.ref<i32>
 151 ! CHECK:             fir.call @_QPsub4(%[[VAL_4]]) {{.*}}: (!fir.ref<i32>) -> ()
 152 ! CHECK:             omp.yield
 153 ! CHECK:           }
 154 ! CHECK:           omp.terminator
 155 ! CHECK:         }
 156 ! CHECK:         return
 157 ! CHECK:       }
 158
 159 subroutine combined_parallel_worksharing_loop()
 160   integer, save :: x6
 161   !$omp threadprivate(x6)
 162
 163   !$omp parallel do copyin(x6)
 164     do i=1, x6
 165       call sub4(x6)
 166     end do
 167   !$omp end parallel do
 168
 169 end
 170
 171 ! CHECK-LABEL: func.func @_QPcombined_parallel_sections() {
 172 ! CHECK:         %[[VAL_0:.*]] = fir.address_of(@_QFcombined_parallel_sectionsEx7) : !fir.ref<i32>
 173 ! CHECK:         %[[VAL_1:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<i32> -> !fir.ref<i32>
 174 ! CHECK:         omp.parallel   {
 175 ! CHECK:           %[[VAL_2:.*]] = omp.threadprivate %[[VAL_0]] : !fir.ref<i32> -> !fir.ref<i32>
 176 ! CHECK:           %[[VAL_3:.*]] = fir.load %[[VAL_1]] : !fir.ref<i32>
 177 ! CHECK:           fir.store %[[VAL_3]] to %[[VAL_2]] : !fir.ref<i32>
 178 ! CHECK:           omp.barrier
 179 ! CHECK:           omp.sections   {
 180 ! CHECK:             omp.section {
 181 ! CHECK:               fir.call @_QPsub5(%[[VAL_2]]) {{.*}}: (!fir.ref<i32>) -> ()
 182 ! CHECK:               omp.terminator
 183 ! CHECK:             }
 184 ! CHECK:             omp.section {
 185 ! CHECK:               fir.call @_QPsub6(%[[VAL_2]]) {{.*}}: (!fir.ref<i32>) -> ()
 186 ! CHECK:               omp.terminator
 187 ! CHECK:             }
 188 ! CHECK:             omp.terminator
 189 ! CHECK:           }
 190 ! CHECK:           omp.terminator
 191 ! CHECK:         }
 192 ! CHECK:         return
 193 ! CHECK:       }
 194
 195 subroutine combined_parallel_sections()
 196   integer, save :: x7
 197   !$omp threadprivate(x7)
 198
 199   !$omp parallel sections copyin(x7)
 200     !$omp section
 201       call sub5(x7)
 202     !$omp section
 203       call sub6(x7)
 204   !$omp end parallel sections
 205
 206 end