90 lines · plain
1! This test checks lowering of atomic and atomic update constructs2! RUN: bbc -fopenacc -emit-hlfir %s -o - | FileCheck %s3! RUN: %flang_fc1 -fopenacc -emit-hlfir %s -o - | FileCheck %s4 5program acc_atomic_update_test6 interface7 integer function func(x)8 integer :: x9 end function func10 end interface11 integer :: x, y, z12 integer, pointer :: a, b13 integer, target :: c, d14 integer(1) :: i115 16 a=>c17 b=>d18 19!CHECK: %[[A:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "a", uniq_name = "_QFEa"}20!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>>>)21!CHECK: %[[B:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "b", uniq_name = "_QFEb"}22!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>>>)23!CHECK: %[[C_ADDR:.*]] = fir.address_of(@_QFEc) : !fir.ref<i32>24!CHECK: %[[D_ADDR:.*]] = fir.address_of(@_QFEd) : !fir.ref<i32>25!CHECK: %[[I1:.*]] = fir.alloca i8 {bindc_name = "i1", uniq_name = "_QFEi1"}26!CHECK: %[[I1_DECL:.*]]:2 = hlfir.declare %[[I1]] {uniq_name = "_QFEi1"} : (!fir.ref<i8>) -> (!fir.ref<i8>, !fir.ref<i8>)27!CHECK: %[[X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFEx"}28!CHECK: %[[X_DECL:.*]]:2 = hlfir.declare %[[X]] {uniq_name = "_QFEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)29!CHECK: %[[Y:.*]] = fir.alloca i32 {bindc_name = "y", uniq_name = "_QFEy"}30!CHECK: %[[Y_DECL:.*]]:2 = hlfir.declare %[[Y]] {uniq_name = "_QFEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)31!CHECK: %[[Z:.*]] = fir.alloca i32 {bindc_name = "z", uniq_name = "_QFEz"}32!CHECK: %[[Z_DECL:.*]]:2 = hlfir.declare %[[Z]] {uniq_name = "_QFEz"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)33!CHECK: %[[LOAD_A:.*]] = fir.load %[[A_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>34!CHECK: %[[BOX_ADDR_A:.*]] = fir.box_addr %[[LOAD_A]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>35!CHECK: %[[LOAD_B:.*]] = fir.load %[[B_DECL]]#0 : !fir.ref<!fir.box<!fir.ptr<i32>>>36!CHECK: %[[BOX_ADDR_B:.*]] = fir.box_addr %[[LOAD_B]] : (!fir.box<!fir.ptr<i32>>) -> !fir.ptr<i32>37!CHECK: %[[LOAD_BOX_ADDR_B:.*]] = fir.load %[[BOX_ADDR_B]] : !fir.ptr<i32>38!CHECK: acc.atomic.update %[[BOX_ADDR_A]] : !fir.ptr<i32> {39!CHECK: ^bb0(%[[ARG0:.*]]: i32):40!CHECK: %[[ADD:.*]] = arith.addi %[[ARG0]], %[[LOAD_BOX_ADDR_B]] : i3241!CHECK: acc.yield %[[ADD]] : i3242!CHECK: }43 44 !$acc atomic update45 a = a + b46 47!CHECK: {{.*}} = arith.constant 1 : i3248!CHECK: acc.atomic.update %[[Y_DECL]]#0 : !fir.ref<i32> {49!CHECK: ^bb0(%[[ARG:.*]]: i32):50!CHECK: %[[RESULT:.*]] = arith.addi %[[ARG]], {{.*}} : i3251!CHECK: acc.yield %[[RESULT]] : i3252!CHECK: }53!CHECK: %[[LOADED_X:.*]] = fir.load %[[X_DECL]]#0 : !fir.ref<i32>54!CHECK: acc.atomic.update %[[Z_DECL]]#0 : !fir.ref<i32> {55!CHECK: ^bb0(%[[ARG:.*]]: i32):56!CHECK: %[[RESULT:.*]] = arith.muli %[[LOADED_X]], %[[ARG]] : i3257!CHECK: acc.yield %[[RESULT]] : i3258!CHECK: }59 !$acc atomic60 y = y + 161 !$acc atomic update62 z = x * z63 64!CHECK: %[[C1_VAL:.*]] = arith.constant 1 : i3265!CHECK: acc.atomic.update %[[I1_DECL]]#0 : !fir.ref<i8> {66!CHECK: ^bb0(%[[VAL:.*]]: i8):67!CHECK: %[[CVT_VAL:.*]] = fir.convert %[[VAL]] : (i8) -> i3268!CHECK: %[[ADD_VAL:.*]] = arith.addi %[[CVT_VAL]], %[[C1_VAL]] : i3269!CHECK: %[[UPDATED_VAL:.*]] = fir.convert %[[ADD_VAL]] : (i32) -> i870!CHECK: acc.yield %[[UPDATED_VAL]] : i871!CHECK: }72 !$acc atomic73 i1 = i1 + 174 !$acc end atomic75 76!CHECK: %[[VAL_44:.*]]:3 = hlfir.associate %{{.*}} {adapt.valuebyref} : (i32) -> (!fir.ref<i32>, !fir.ref<i32>, i1)77!CHECK: %[[VAL_45:.*]] = fir.call @_QPfunc(%[[VAL_44]]#0) fastmath<contract> : (!fir.ref<i32>) -> i3278!CHECK: acc.atomic.update %[[X_DECL]]#0 : !fir.ref<i32> {79!CHECK: ^bb0(%[[VAL_46:.*]]: i32):80!CHECK: %[[VAL_47:.*]] = arith.addi %[[VAL_46]], %[[VAL_45]] : i3281!CHECK: acc.yield %[[VAL_47]] : i3282!CHECK: }83!CHECK: hlfir.end_associate %[[VAL_44]]#1, %[[VAL_44]]#2 : !fir.ref<i32>, i184 !$acc atomic update85 x = x + func(z + 1)86 !$acc end atomic87!CHECK: return88!CHECK: }89end program acc_atomic_update_test90