42 lines · plain
1! RUN: %flang_fc1 -emit-hlfir -mmlir --enable-delayed-privatization-staging=true -o - %s | FileCheck %s2 3subroutine do_concurrent_reduce4 implicit none5 integer :: s, i6 7 do concurrent (i=1:10) reduce(+:s)8 s = s + 19 end do10end11 12! CHECK-LABEL: fir.declare_reduction @add_reduction_i32 : i32 init {13! CHECK: ^bb0(%[[ARG0:.*]]: i32):14! CHECK: %[[VAL_0:.*]] = arith.constant 0 : i3215! CHECK: fir.yield(%[[VAL_0]] : i32)16! CHECK: } combiner {17! CHECK: ^bb0(%[[VAL_1:.*]]: i32, %[[VAL_2:.*]]: i32):18! CHECK: %[[VAL_3:.*]] = arith.addi %[[VAL_1]], %[[VAL_2]] : i3219! CHECK: fir.yield(%[[VAL_3]] : i32)20! CHECK: }21 22! CHECK-LABEL: func.func @_QPdo_concurrent_reduce() {23! CHECK: %[[S_ALLOC:.*]] = fir.alloca i32 {bindc_name = "s", uniq_name = "_QFdo_concurrent_reduceEs"}24! CHECK: %[[S_DECL:.*]]:2 = hlfir.declare %[[S_ALLOC]] {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)25 26! CHECK: fir.do_concurrent {27! CHECK: %[[VAL_0:.*]] = fir.alloca i32 {bindc_name = "i"}28! CHECK: %[[VAL_1:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "_QFdo_concurrent_reduceEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)29! CHECK: fir.do_concurrent.loop (%{{.*}}) = (%{{.*}}) to (%{{.*}}) step (%{{[^[:space:]]+}})30! CHECK-SAME: reduce(@add_reduction_i32 #fir.reduce_attr<add> %[[S_DECL]]#0 -> %[[S_ARG:.*]] : !fir.ref<i32>) {31 32! CHECK: %[[S_ARG_DECL:.*]]:2 = hlfir.declare %[[S_ARG]] {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)33! CHECK: %[[S_ARG_VAL:.*]] = fir.load %[[S_ARG_DECL]]#0 : !fir.ref<i32>34! CHECK: %[[C1:.*]] = arith.constant 1 : i3235! CHECK: %[[RED_UPDATE:.*]] = arith.addi %[[S_ARG_VAL]], %[[C1]] : i3236! CHECK: hlfir.assign %[[RED_UPDATE]] to %[[S_ARG_DECL]]#0 : i32, !fir.ref<i32>37 38! CHECK: }39! CHECK: }40! CHECK: return41! CHECK: }42