xref: /llvm-project/flang/test/Lower/OpenACC/acc-atomic-update-hlfir.f90 (revision dfcf9fe1408168699c27bc8e8c877bf78b8fd390)
1! This test checks lowering of atomic and atomic update constructs with HLFIR
2! RUN: bbc -hlfir -fopenacc -emit-hlfir %s -o - | FileCheck %s
3! RUN: %flang_fc1 -emit-hlfir -fopenacc %s -o - | FileCheck %s
4
5!CHECK-LABEL: @_QPsb
6subroutine sb
7!CHECK:   %[[W_REF:.*]] = fir.alloca i32 {bindc_name = "w", uniq_name = "_QFsbEw"}
8!CHECK:   %[[W_DECL:.*]]:2 = hlfir.declare %[[W_REF]] {uniq_name = "_QFsbEw"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
9!CHECK:   %[[X_REF:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFsbEx"}
10!CHECK:   %[[X_DECL:.*]]:2 = hlfir.declare %[[X_REF]] {uniq_name = "_QFsbEx"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
11!CHECK:   %[[Y_REF:.*]] = fir.alloca i32 {bindc_name = "y", uniq_name = "_QFsbEy"}
12!CHECK:   %[[Y_DECL:.*]]:2 = hlfir.declare %[[Y_REF]] {uniq_name = "_QFsbEy"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
13!CHECK:   %[[Z_REF:.*]] = fir.alloca i32 {bindc_name = "z", uniq_name = "_QFsbEz"}
14!CHECK:   %[[Z_DECL:.*]]:2 = hlfir.declare %[[Z_REF]] {uniq_name = "_QFsbEz"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
15  integer :: w, x, y, z
16
17!CHECK:   %[[Y_VAL:.*]] = fir.load %[[Y_DECL]]#0 : !fir.ref<i32>
18!CHECK:   acc.atomic.update   %[[X_DECL]]#1 : !fir.ref<i32> {
19!CHECK:   ^bb0(%[[ARG_X:.*]]: i32):
20!CHECK:     %[[X_UPDATE_VAL:.*]] = arith.addi %[[ARG_X]], %[[Y_VAL]] : i32
21!CHECK:     acc.yield %[[X_UPDATE_VAL]] : i32
22!CHECK:   }
23  !$acc atomic update
24    x = x + y
25
26!CHECK:   %[[Y_VAL:.*]] = fir.load %[[Y_DECL]]#0 : !fir.ref<i32>
27!CHECK:   acc.atomic.update %[[X_DECL]]#1 : !fir.ref<i32> {
28!CHECK:   ^bb0(%[[ARG_X:.*]]: i32):
29!CHECK:     %[[X_UPDATE_VAL:.*]] = arith.ori %[[ARG_X]], %[[Y_VAL]] : i32
30!CHECK:     acc.yield %[[X_UPDATE_VAL]] : i32
31!CHECK:   }
32  !$acc atomic update
33    x = ior(x,y)
34
35!CHECK:   %[[W_VAL:.*]] = fir.load %[[W_DECL]]#0 : !fir.ref<i32>
36!CHECK:   %[[X_VAL:.*]] = fir.load %[[X_DECL]]#0 : !fir.ref<i32>
37!CHECK:   %[[Y_VAL:.*]] = fir.load %[[Y_DECL]]#0 : !fir.ref<i32>
38!CHECK:   acc.atomic.update %[[Z_DECL]]#1 : !fir.ref<i32> {
39!CHECK:   ^bb0(%[[ARG_Z:.*]]: i32):
40!CHECK:     %[[WX_CMP:.*]] = arith.cmpi slt, %[[W_VAL]], %[[X_VAL]] : i32
41!CHECK:     %[[WX_MIN:.*]] = arith.select %[[WX_CMP]], %[[W_VAL]], %[[X_VAL]] : i32
42!CHECK:     %[[WXY_CMP:.*]] = arith.cmpi slt, %[[WX_MIN]], %[[Y_VAL]] : i32
43!CHECK:     %[[WXY_MIN:.*]] = arith.select %[[WXY_CMP]], %[[WX_MIN]], %[[Y_VAL]] : i32
44!CHECK:     %[[WXYZ_CMP:.*]] = arith.cmpi slt, %[[WXY_MIN]], %[[ARG_Z]] : i32
45!CHECK:     %[[WXYZ_MIN:.*]] = arith.select %[[WXYZ_CMP]], %[[WXY_MIN]], %[[ARG_Z]] : i32
46!CHECK:     acc.yield %[[WXYZ_MIN]] : i32
47!CHECK:   }
48  !$acc atomic update
49    z = min(w,x,y,z)
50
51!CHECK:   return
52end subroutine
53