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

   1 ! This test checks lowering of atomic and atomic update constructs
   2 ! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s
   3 ! RUN: %flang_fc1 -fopenacc -emit-hlfir %s -o - | FileCheck %s
   4
   5 program acc_atomic_update_test
   6     integer :: x, y, z
   7     integer, pointer :: a, b
   8     integer, target :: c, d
   9     integer(1) :: i1
  10
  11     a=>c
  12     b=>d
  13
  14 !CHECK: %[[A:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "a", uniq_name = "_QFEa"}
  15 !CHECK: %[[A_DECL:.*]]:2 = hlfir.declare %[[A]] {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFEa"} : (!fir.ref<!fir.box<!fir.ptr<i32>>>) -> (!fir.ref<!fir.box<!fir.ptr<i32>>>, !fir.ref<!fir.box<!fir.ptr<i32>>>)
  16 !CHECK: %[[B:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "b", uniq_name = "_QFEb"}
  17 !CHECK: %[[B_DECL:.*]]:2 = hlfir.declare %[[B]] {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFEb"} : (!fir.ref<!fir.box<!fir.ptr<i32>>>) -> (!fir.ref<!fir.box<!fir.ptr<i32>>>, !fir.ref<!fir.box<!fir.ptr<i32>>>)
  18 !CHECK: %[[C_ADDR:.*]] = fir.address_of(@_QFEc) : !fir.ref<i32>
  19 !CHECK: %[[D_ADDR:.*]] = fir.address_of(@_QFEd) : !fir.ref<i32>
  20 !CHECK: %[[I1:.*]] = fir.alloca i8 {bindc_name = "i1", uniq_name = "_QFEi1"}
  21 !CHECK: %[[I1_DECL:.*]]:2 = hlfir.declare %[[I1]] {uniq_name = "_QFEi1"} : (!fir.ref<i8>) -> (!fir.ref<i8>, !fir.ref<i8>)
  22 !CHECK: %[[X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFEx"}
  23 !CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] {uniq_name = "_QFEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  24 !CHECK: %[[Y:.*]] = fir.alloca i32 {bindc_name = "y", uniq_name = "_QFEy"}
  25 !CHECK: %[[Y_DECL:.*]]:2 = hlfir.declare %[[Y]] {uniq_name = "_QFEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  26 !CHECK: %[[Z:.*]] = fir.alloca i32 {bindc_name = "z", uniq_name = "_QFEz"}
  27 !CHECK: %[[Z_DECL:.*]]:2 = hlfir.declare %[[Z]] {uniq_name = "_QFEz"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  28 !CHECK: %[[LOAD_A:.*]] = fir.load %[[A_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>
  29 !CHECK: %[[BOX_ADDR_A:.*]] = fir.box_addr %[[LOAD_A]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>
  30 !CHECK: %[[LOAD_B:.*]] = fir.load %[[B_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>
  31 !CHECK: %[[BOX_ADDR_B:.*]] = fir.box_addr %[[LOAD_B]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>
  32 !CHECK: %[[LOAD_BOX_ADDR_B:.*]] = fir.load %[[BOX_ADDR_B]] : !fir.ptr<i32>
  33 !CHECK: acc.atomic.update %[[BOX_ADDR_A]] : !fir.ptr<i32> {
  34 !CHECK: ^bb0(%[[ARG0:.*]]: i32):
  35 !CHECK:   %[[ADD:.*]] = arith.addi %[[ARG0]], %[[LOAD_BOX_ADDR_B]] : i32
  36 !CHECK:   acc.yield %[[ADD]] : i32
  37 !CHECK: }
  38
  39     !$acc atomic update
  40         a = a + b
  41
  42 !CHECK: {{.*}} = arith.constant 1 : i32
  43 !CHECK: acc.atomic.update %[[Y_DECL]]#1 : !fir.ref<i32> {
  44 !CHECK:  ^bb0(%[[ARG:.*]]: i32):
  45 !CHECK:    %[[RESULT:.*]] = arith.addi %[[ARG]], {{.*}} : i32
  46 !CHECK:    acc.yield %[[RESULT]] : i32
  47 !CHECK:  }
  48 !CHECK:  %[[LOADED_X:.*]] = fir.load %[[X_DECL]]#0 : !fir.ref<i32>
  49 !CHECK:  acc.atomic.update %[[Z_DECL]]#1 : !fir.ref<i32> {
  50 !CHECK:  ^bb0(%[[ARG:.*]]: i32):
  51 !CHECK:    %[[RESULT:.*]] = arith.muli %[[LOADED_X]], %[[ARG]] : i32
  52 !CHECK:    acc.yield %[[RESULT]] : i32
  53 !CHECK:  }
  54     !$acc atomic
  55         y = y + 1
  56     !$acc atomic update
  57         z = x * z
  58
  59 !CHECK:  %[[C1_VAL:.*]] = arith.constant 1 : i32
  60 !CHECK:  acc.atomic.update %[[I1_DECL]]#1 : !fir.ref<i8> {
  61 !CHECK:  ^bb0(%[[VAL:.*]]: i8):
  62 !CHECK:    %[[CVT_VAL:.*]] = fir.convert %[[VAL]] : (i8) -> i32
  63 !CHECK:    %[[ADD_VAL:.*]] = arith.addi %[[CVT_VAL]], %[[C1_VAL]] : i32
  64 !CHECK:    %[[UPDATED_VAL:.*]] = fir.convert %[[ADD_VAL]] : (i32) -> i8
  65 !CHECK:    acc.yield %[[UPDATED_VAL]] : i8
  66 !CHECK:  }
  67     !$acc atomic
  68       i1 = i1 + 1
  69     !$acc end atomic
  70 !CHECK:  return
  71 !CHECK: }
  72 end program acc_atomic_update_test
  73