101 lines · plain
1!RUN: %flang_fc1 -emit-hlfir -ffast-math -fopenmp -fopenmp-version=60 %s -o - | FileCheck %s2 3subroutine f00(x, y)4 implicit none5 real :: x, y6 7 !$omp atomic update8 x = ((x + 1) + y) + 29end10 11!CHECK-LABEL: func.func @_QPf0012!CHECK: %[[X:[0-9]+]]:2 = hlfir.declare %arg013!CHECK: %[[Y:[0-9]+]]:2 = hlfir.declare %arg114!CHECK: %cst = arith.constant 1.000000e+00 : f3215!CHECK: %[[LOAD_Y:[0-9]+]] = fir.load %[[Y]]#0 : !fir.ref<f32>16!CHECK: %[[Y_1:[0-9]+]] = arith.addf %cst, %[[LOAD_Y]] fastmath<fast> : f3217!CHECK: %cst_0 = arith.constant 2.000000e+00 : f3218!CHECK: %[[Y_1_2:[0-9]+]] = arith.addf %[[Y_1]], %cst_0 fastmath<fast> : f3219!CHECK: omp.atomic.update memory_order(relaxed) %[[X]]#0 : !fir.ref<f32> {20!CHECK: ^bb0(%[[ARG:arg[0-9]+]]: f32):21!CHECK: %[[ARG_P:[0-9]+]] = arith.addf %[[ARG]], %[[Y_1_2]] fastmath<fast> : f3222!CHECK: omp.yield(%[[ARG_P]] : f32)23!CHECK: }24 25 26subroutine f01(x, y, z)27 implicit none28 complex :: x, y, z29 30 !$omp atomic update31 x = (x + y) + z32end33 34!CHECK-LABEL: func.func @_QPf0135!CHECK: %[[X:[0-9]+]]:2 = hlfir.declare %arg036!CHECK: %[[Y:[0-9]+]]:2 = hlfir.declare %arg137!CHECK: %[[Z:[0-9]+]]:2 = hlfir.declare %arg238!CHECK: %[[LOAD_Y:[0-9]+]] = fir.load %[[Y]]#0 : !fir.ref<complex<f32>>39!CHECK: %[[LOAD_Z:[0-9]+]] = fir.load %[[Z]]#0 : !fir.ref<complex<f32>>40!CHECK: %[[Y_Z:[0-9]+]] = fir.addc %[[LOAD_Y]], %[[LOAD_Z]] {fastmath = #arith.fastmath<fast>} : complex<f32>41!CHECK: omp.atomic.update memory_order(relaxed) %[[X]]#0 : !fir.ref<complex<f32>> {42!CHECK: ^bb0(%[[ARG:arg[0-9]+]]: complex<f32>):43!CHECK: %[[ARG_P:[0-9]+]] = fir.addc %[[ARG]], %[[Y_Z]] {fastmath = #arith.fastmath<fast>} : complex<f32>44!CHECK: omp.yield(%[[ARG_P]] : complex<f32>)45!CHECK: }46 47 48subroutine f02(x, y)49 implicit none50 complex :: x51 real :: y52 53 !$omp atomic update54 x = (real(x) + y) + 155end56 57!CHECK-LABEL: func.func @_QPf0258!CHECK: %[[X:[0-9]+]]:2 = hlfir.declare %arg059!CHECK: %[[Y:[0-9]+]]:2 = hlfir.declare %arg160!CHECK: %[[LOAD_Y:[0-9]+]] = fir.load %[[Y]]#0 : !fir.ref<f32>61!CHECK: %cst = arith.constant 1.000000e+00 : f3262!CHECK: %[[Y_1:[0-9]+]] = arith.addf %[[LOAD_Y]], %cst fastmath<fast> : f3263!CHECK: omp.atomic.update memory_order(relaxed) %[[X]]#0 : !fir.ref<complex<f32>> {64!CHECK: ^bb0(%[[ARG:arg[0-9]+]]: complex<f32>):65!CHECK: %[[ARG_X:[0-9]+]] = fir.extract_value %[[ARG]], [0 : index] : (complex<f32>) -> f3266!CHECK: %[[ARG_P:[0-9]+]] = arith.addf %[[ARG_X]], %[[Y_1]] fastmath<fast> : f3267!CHECK: %cst_0 = arith.constant 0.000000e+00 : f3268!CHECK: %[[CPLX:[0-9]+]] = fir.undefined complex<f32>69!CHECK: %[[CPLX_I:[0-9]+]] = fir.insert_value %[[CPLX]], %[[ARG_P]], [0 : index] : (complex<f32>, f32) -> complex<f32>70!CHECK: %[[CPLX_R:[0-9]+]] = fir.insert_value %[[CPLX_I]], %cst_0, [1 : index] : (complex<f32>, f32) -> complex<f32>71!CHECK: omp.yield(%[[CPLX_R]] : complex<f32>)72!CHECK: }73 74 75subroutine f03(x, a, b, c)76 implicit none77 real(kind=4) :: x78 real(kind=8) :: a, b, c79 80 !$omp atomic update81 x = ((b + a) + x) + c82end83 84!CHECK-LABEL: func.func @_QPf0385!CHECK: %[[A:[0-9]+]]:2 = hlfir.declare %arg186!CHECK: %[[B:[0-9]+]]:2 = hlfir.declare %arg287!CHECK: %[[C:[0-9]+]]:2 = hlfir.declare %arg388!CHECK: %[[X:[0-9]+]]:2 = hlfir.declare %arg089!CHECK: %[[LOAD_B:[0-9]+]] = fir.load %[[B]]#0 : !fir.ref<f64>90!CHECK: %[[LOAD_A:[0-9]+]] = fir.load %[[A]]#0 : !fir.ref<f64>91!CHECK: %[[A_B:[0-9]+]] = arith.addf %[[LOAD_B]], %[[LOAD_A]] fastmath<fast> : f6492!CHECK: %[[LOAD_C:[0-9]+]] = fir.load %[[C]]#0 : !fir.ref<f64>93!CHECK: %[[A_B_C:[0-9]+]] = arith.addf %[[A_B]], %[[LOAD_C]] fastmath<fast> : f6494!CHECK: omp.atomic.update memory_order(relaxed) %[[X]]#0 : !fir.ref<f32> {95!CHECK: ^bb0(%[[ARG:arg[0-9]+]]: f32):96!CHECK: %[[ARG_8:[0-9]+]] = fir.convert %[[ARG]] : (f32) -> f6497!CHECK: %[[ARG_P:[0-9]+]] = arith.addf %[[ARG_8]], %[[A_B_C]] fastmath<fast> : f6498!CHECK: %[[ARG_4:[0-9]+]] = fir.convert %[[ARG_P]] : (f64) -> f3299!CHECK: omp.yield(%[[ARG_4]] : f32)100!CHECK: }101