1! RUN: bbc -fopenmp -emit-fir %s -o - | FileCheck %s
2
3! This test checks the lowering of atomic write
4
5!CHECK: func @_QQmain() {
6!CHECK: %[[VAR_X:.*]] = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFEx"}
7!CHECK: %[[VAR_Y:.*]] = fir.alloca i32 {bindc_name = "y", uniq_name = "_QFEy"}
8!CHECK: %[[VAR_Z:.*]] = fir.alloca i32 {bindc_name = "z", uniq_name = "_QFEz"}
9!CHECK: %[[CONST_44:.*]] = arith.constant 44 : i32
10!CHECK: omp.atomic.write %[[VAR_X]] = %[[CONST_44]] hint(uncontended) memory_order(seq_cst) : !fir.ref<i32>, i32
11!CHECK: %[[CONST_7:.*]] = arith.constant 7 : i32
12!CHECK: {{.*}} = fir.load %[[VAR_Y]] : !fir.ref<i32>
13!CHECK: %[[VAR_7y:.*]] = arith.muli %[[CONST_7]], {{.*}} : i32
14!CHECK: omp.atomic.write %[[VAR_X]] = %[[VAR_7y]] memory_order(relaxed) : !fir.ref<i32>, i32
15!CHECK: %[[CONST_10:.*]] = arith.constant 10 : i32
16!CHECK: {{.*}} = fir.load %[[VAR_X]] : !fir.ref<i32>
17!CHECK: {{.*}} = arith.muli %[[CONST_10]], {{.*}} : i32
18!CHECK: {{.*}} = fir.load %[[VAR_Z]] : !fir.ref<i32>
19!CHECK: %[[CONST_2:.*]] = arith.constant 2 : i32
20!CHECK: {{.*}} = arith.divsi {{.*}}, %[[CONST_2]] : i32
21!CHECK: {{.*}} = arith.addi {{.*}}, {{.*}} : i32
22!CHECK: omp.atomic.write %[[VAR_Y]] = {{.*}} hint(speculative) memory_order(release) : !fir.ref<i32>, i32
23!CHECK: return
24!CHECK: }
25
26program OmpAtomicWrite
27    use omp_lib
28    integer :: x, y, z
29    !$omp atomic seq_cst write hint(omp_sync_hint_uncontended)
30        x = 8*4 + 12
31
32    !$omp atomic write relaxed
33        x = 7 * y
34
35    !$omp atomic write release hint(omp_sync_hint_speculative)
36        y = 10*x + z/2
37end program OmpAtomicWrite
38
39! Test lowering atomic read for pointer variables.
40! Please notice to use %[[VAL_1]] for operands of atomic operation, instead
41! of %[[VAL_0]].
42
43!CHECK-LABEL: func.func @_QPatomic_write_pointer() {
44!CHECK:         %[[VAL_0:.*]] = fir.alloca !fir.box<!fir.ptr<i32>> {bindc_name = "x", uniq_name = "_QFatomic_write_pointerEx"}
45!CHECK:         %[[VAL_1:.*]] = fir.alloca !fir.ptr<i32> {uniq_name = "_QFatomic_write_pointerEx.addr"}
46!CHECK:         %[[VAL_2:.*]] = fir.zero_bits !fir.ptr<i32>
47!CHECK:         fir.store %[[VAL_2]] to %[[VAL_1]] : !fir.ref<!fir.ptr<i32>>
48!CHECK:         %[[VAL_3:.*]] = arith.constant 1 : i32
49!CHECK:         %[[VAL_4:.*]] = fir.load %[[VAL_1]] : !fir.ref<!fir.ptr<i32>>
50!CHECK:         omp.atomic.write %[[VAL_4]] = %[[VAL_3]]   : !fir.ptr<i32>, i32
51!CHECK:         %[[VAL_5:.*]] = arith.constant 2 : i32
52!CHECK:         %[[VAL_6:.*]] = fir.load %[[VAL_1]] : !fir.ref<!fir.ptr<i32>>
53!CHECK:         fir.store %[[VAL_5]] to %[[VAL_6]] : !fir.ptr<i32>
54!CHECK:         return
55!CHECK:       }
56
57subroutine atomic_write_pointer()
58  integer, pointer :: x
59
60  !$omp atomic write
61    x = 1
62
63  x = 2
64end
65
66