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 5program 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: } 72end program acc_atomic_update_test 73 74