flang/test/Lower/OpenACC/acc-atomic-write.f90

   1 ! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s
   2
   3 ! This test checks the lowering of atomic write
   4
   5 !CHECK: func @_QQmain() attributes {fir.bindc_name = "acc_atomic_write_test"} {
   6 !CHECK: %[[VAR_X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFEx"}
   7 !CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[VAR_X]] {uniq_name = "_QFEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
   8 !CHECK: %[[VAR_Y:.*]] = fir.alloca i32 {bindc_name = "y", uniq_name = "_QFEy"}
   9 !CHECK: %[[Y_DECL:.*]]:2 = hlfir.declare %[[VAR_Y]] {uniq_name = "_QFEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  10 !CHECK: %[[CONST_7:.*]] = arith.constant 7 : i32
  11 !CHECK: {{.*}} = fir.load %[[Y_DECL]]#0 : !fir.ref<i32>
  12 !CHECK: %[[VAR_7y:.*]] = arith.muli %[[CONST_7]], {{.*}} : i32
  13 !CHECK: acc.atomic.write %[[X_DECL]]#1 = %[[VAR_7y]] : !fir.ref<i32>, i32
  14 !CHECK: return
  15 !CHECK: }
  16
  17 program acc_atomic_write_test
  18     integer :: x, y
  19
  20     !$acc atomic write
  21         x = 7 * y
  22
  23 end program acc_atomic_write_test
  24
  25 ! Test lowering atomic read for pointer variables.
  26
  27 !CHECK-LABEL: func.func @_QPatomic_write_pointer() {
  28 !CHECK: %[[X:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "x", uniq_name = "_QFatomic_write_pointerEx"}
  29 !CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFatomic_write_pointerEx"} : (!fir.ref<!fir.box<!fir.ptr<i32>>>) -> (!fir.ref<!fir.box<!fir.ptr<i32>>>, !fir.ref<!fir.box<!fir.ptr<i32>>>)
  30 !CHECK: %[[C1:.*]] = arith.constant 1 : i32
  31 !CHECK: %[[LOAD_X:.*]] = fir.load %[[X_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>
  32 !CHECK: %[[BOX_ADDR_X:.*]] = fir.box_addr %[[LOAD_X]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>
  33 !CHECK: acc.atomic.write %[[BOX_ADDR_X]] = %[[C1]] : !fir.ptr<i32>, i32
  34 !CHECK: %[[C2:.*]] = arith.constant 2 : i32
  35 !CHECK: %[[LOAD_X:.*]] = fir.load %[[X_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>
  36 !CHECK: %[[BOX_ADDR_X:.*]] = fir.box_addr %[[LOAD_X]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>
  37 !CHECK: hlfir.assign %[[C2]] to %[[BOX_ADDR_X]] : i32, !fir.ptr<i32>
  38
  39 subroutine atomic_write_pointer()
  40   integer, pointer :: x
  41
  42   !$acc atomic write
  43     x = 1
  44
  45   x = 2
  46 end subroutine
  47
  48 !CHECK-LABEL: func.func @_QPatomic_write_typed_assign
  49 !CHECK: %[[R2:.*]] = fir.alloca f32 {bindc_name = "r2", uniq_name = "{{.*}}r2"}
  50 !CHECK: %[[R2_DECL:.*]]:2 = hlfir.declare %[[R2]] {uniq_name = "_QFatomic_write_typed_assignEr2"} : (!fir.ref<f32>) -> (!fir.ref<f32>, !fir.ref<f32>)
  51 !CHECK: %[[CST:.*]] = arith.constant 0.000000e+00 : f32
  52 !CHECK: acc.atomic.write %[[R2_DECL]]#1 = %[[CST]]   : !fir.ref<f32>, f32
  53
  54 subroutine atomic_write_typed_assign
  55   real :: r2
  56   !$acc atomic write
  57   r2 = 0
  58 end subroutine