flang/test/Lower/OpenACC/acc-atomic-update-array.f90

   1 ! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s
   2
   3 subroutine atomic_update_array1(r, n, x)
   4   implicit none
   5   integer :: n
   6   real :: r(n), x
   7   integer :: i
   8
   9   !$acc data copy(r)
  10
  11   !$acc parallel loop
  12   do i = 1, n
  13     !$acc atomic update
  14     r(i) = r(i) + x
  15     !$acc end atomic
  16   end do
  17
  18   !$acc end data
  19 end subroutine
  20
  21 ! CHECK-LABEL: func.func @_QPatomic_update_array1(
  22 ! CHECK-SAME: %[[ARG0:.*]]: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "r"}, %[[ARG1:.*]]: !fir.ref<i32> {fir.bindc_name = "n"}, %[[ARG2:.*]]: !fir.ref<f32> {fir.bindc_name = "x"}) {
  23 ! CHECK: %[[DECL_ARG2:.*]]:2 = hlfir.declare %[[ARG2]] {uniq_name = "_QFatomic_update_array1Ex"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  24 ! CHECK: %[[DECL_ARG0:.*]]:2 = hlfir.declare %[[ARG0]](%{{.*}}) {uniq_name = "_QFatomic_update_array1Er"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>)
  25 ! CHECK: %[[ARRAY_REF:.*]] = hlfir.designate %[[DECL_ARG0]]#0 (%{{.*}})  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  26 ! CHECK: %[[LOAD_X:.*]] = fir.load %[[DECL_ARG2]]#0 : !fir.ref<f32>
  27 ! CHECK: acc.atomic.update %[[ARRAY_REF]] : !fir.ref<f32> {
  28 ! CHECK: ^bb0(%[[ARG:.*]]: f32):
  29 ! CHECK:   %[[ATOMIC:.*]] = arith.addf %[[ARG]], %[[LOAD_X]] fastmath<contract> : f32
  30 ! CHECK:   acc.yield %[[ATOMIC]] : f32
  31 ! CHECK: }
  32
  33
  34 subroutine atomic_read_array1(r, n, x)
  35   implicit none
  36   integer :: n
  37   real :: r(n), x
  38
  39   !$acc atomic read
  40   x = r(n)
  41 end subroutine
  42
  43 ! CHECK-LABEL: func.func @_QPatomic_read_array1(
  44 ! CHECK-SAME: %[[ARG0:.*]]: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "r"}, %[[ARG1:.*]]: !fir.ref<i32> {fir.bindc_name = "n"}, %[[ARG2:.*]]: !fir.ref<f32> {fir.bindc_name = "x"}) {
  45 ! CHECK: %[[DECL_X:.*]]:2 = hlfir.declare %[[ARG2]] {uniq_name = "_QFatomic_read_array1Ex"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  46 ! CHECK: %[[DECL_R:.*]]:2 = hlfir.declare %[[ARG0]](%{{.*}}) {uniq_name = "_QFatomic_read_array1Er"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>)
  47 ! CHECK: %[[DES:.*]] = hlfir.designate %[[DECL_R]]#0 (%{{.*}})  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  48 ! CHECK: acc.atomic.read %[[DECL_X]]#1 = %[[DES]] : !fir.ref<f32>, f32
  49
  50 subroutine atomic_write_array1(r, n, x)
  51   implicit none
  52   integer :: n
  53   real :: r(n), x
  54
  55   !$acc atomic write
  56   x = r(n)
  57 end subroutine
  58
  59 ! CHECK-LABEL: func.func @_QPatomic_write_array1(
  60 ! CHECK-SAME: %[[ARG0:.*]]: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "r"}, %[[ARG1:.*]]: !fir.ref<i32> {fir.bindc_name = "n"}, %[[ARG2:.*]]: !fir.ref<f32> {fir.bindc_name = "x"}) {
  61 ! CHECK: %[[DECL_X:.*]]:2 = hlfir.declare %[[ARG2]] {uniq_name = "_QFatomic_write_array1Ex"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  62 ! CHECK: %[[DECL_R:.*]]:2 = hlfir.declare %[[ARG0]](%{{.*}}) {uniq_name = "_QFatomic_write_array1Er"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>)
  63 ! CHECK: %[[DES:.*]] = hlfir.designate %[[DECL_R]]#0 (%{{.*}})  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  64 ! CHECK: %[[LOAD:.*]] = fir.load %[[DES]] : !fir.ref<f32>
  65 ! CHECK: acc.atomic.write %[[DECL_X]]#1 = %[[LOAD]] : !fir.ref<f32>, f32
  66
  67 subroutine atomic_capture_array1(r, n, x, y)
  68   implicit none
  69   integer :: n, i
  70   real :: r(n), x, y
  71
  72   !$acc atomic capture
  73   r(i) = r(i) + x
  74   y = r(i)
  75   !$acc end atomic
  76 end subroutine
  77
  78 ! CHECK-LABEL: func.func @_QPatomic_capture_array1(
  79 ! CHECK-SAME: %[[ARG0:.*]]: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "r"}, %[[ARG1:.*]]: !fir.ref<i32> {fir.bindc_name = "n"}, %[[ARG2:.*]]: !fir.ref<f32> {fir.bindc_name = "x"}, %[[ARG3:.*]]: !fir.ref<f32> {fir.bindc_name = "y"}) {
  80 ! CHECK: %[[DECL_X:.*]]:2 = hlfir.declare %[[ARG2]] {uniq_name = "_QFatomic_capture_array1Ex"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  81 ! CHECK: %[[DECL_Y:.*]]:2 = hlfir.declare %[[ARG3]] {uniq_name = "_QFatomic_capture_array1Ey"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  82 ! CHECK: %[[DECL_R:.*]]:2 = hlfir.declare %[[ARG0]](%{{.*}}) {uniq_name = "_QFatomic_capture_array1Er"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>) -> (!fir.box<!fir.array<?xf32>>, !fir.ref<!fir.array<?xf32>>)
  83 ! CHECK: %[[R_I:.*]] = hlfir.designate %[[DECL_R]]#0 (%{{.*}})  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  84 ! CHECK: %[[LOAD:.*]] = fir.load %[[DECL_X]]#0 : !fir.ref<f32>
  85 ! CHECK: acc.atomic.capture {
  86 ! CHECK:   acc.atomic.update %[[R_I]] : !fir.ref<f32> {
  87 ! CHECK:   ^bb0(%[[ARG:.*]]: f32):
  88 ! CHECK:     %[[ADD:.*]] = arith.addf %[[ARG]], %[[LOAD]] fastmath<contract> : f32
  89 ! CHECK:     acc.yield %[[ADD]] : f32
  90 ! CHECK:   }
  91 ! CHECK:   acc.atomic.read %[[DECL_Y]]#1 = %[[R_I]] : !fir.ref<f32>, f32
  92 ! CHECK: }