84 lines · plain
1// Tests mapping reductions and local from fir to OpenMP.2 3// RUN: fir-opt --omp-do-concurrent-conversion="map-to=host" %s | FileCheck %s4 5fir.declare_reduction @add_reduction_i32 : i32 init {6^bb0(%arg0: i32):7 %c0_i32 = arith.constant 0 : i328 fir.yield(%c0_i32 : i32)9} combiner {10^bb0(%arg0: i32, %arg1: i32):11 %0 = arith.addi %arg0, %arg1 : i3212 fir.yield(%0 : i32)13}14 fir.local {type = local} @_QFdo_concurrent_reduceEl_private_i32 : i3215 func.func @_QPdo_concurrent_reduce() {16 %3 = fir.alloca i32 {bindc_name = "l", uniq_name = "_QFdo_concurrent_reduceEl"}17 %4:2 = hlfir.declare %3 {uniq_name = "_QFdo_concurrent_reduceEl"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)18 %5 = fir.alloca i32 {bindc_name = "s", uniq_name = "_QFdo_concurrent_reduceEs"}19 %6:2 = hlfir.declare %5 {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)20 %c1 = arith.constant 1 : index21 fir.do_concurrent {22 %9 = fir.alloca i32 {bindc_name = "i"}23 %10:2 = hlfir.declare %9 {uniq_name = "_QFdo_concurrent_reduceEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)24 fir.do_concurrent.loop (%arg0) = (%c1) to (%c1) step (%c1) local(@_QFdo_concurrent_reduceEl_private_i32 %4#0 -> %arg1 : !fir.ref<i32>) reduce(@add_reduction_i32 #fir.reduce_attr<add> %6#0 -> %arg2 : !fir.ref<i32>) {25 %11 = fir.convert %arg0 : (index) -> i3226 fir.store %11 to %10#0 : !fir.ref<i32>27 %12:2 = hlfir.declare %arg1 {uniq_name = "_QFdo_concurrent_reduceEl"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)28 %13:2 = hlfir.declare %arg2 {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)29 %c1_i32_0 = arith.constant 1 : i3230 hlfir.assign %c1_i32_0 to %12#0 : i32, !fir.ref<i32>31 %14 = fir.load %13#0 : !fir.ref<i32>32 %15 = fir.load %12#0 : !fir.ref<i32>33 %16 = arith.addi %14, %15 : i3234 hlfir.assign %16 to %13#0 : i32, !fir.ref<i32>35 }36 }37 return38}39 40// CHECK-LABEL: omp.declare_reduction @add_reduction_i32.omp : i32 init {41// CHECK: ^bb0(%[[VAL_0:.*]]: i32):42// CHECK: %[[VAL_1:.*]] = arith.constant 0 : i3243// CHECK: omp.yield(%[[VAL_1]] : i32)44 45// CHECK-LABEL: } combiner {46// CHECK: ^bb0(%[[VAL_0:.*]]: i32, %[[VAL_1:.*]]: i32):47// CHECK: %[[VAL_2:.*]] = arith.addi %[[VAL_0]], %[[VAL_1]] : i3248// CHECK: omp.yield(%[[VAL_2]] : i32)49// CHECK: }50 51// CHECK: omp.private {type = private} @_QFdo_concurrent_reduceEl_private_i32.omp : i3252 53// CHECK-LABEL: func.func @_QPdo_concurrent_reduce() {54// CHECK: %[[VAL_0:.*]] = fir.alloca i32 {bindc_name = "i"}55// CHECK: %[[VAL_1:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "_QFdo_concurrent_reduceEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)56// CHECK: %[[VAL_2:.*]] = fir.alloca i32 {bindc_name = "l", uniq_name = "_QFdo_concurrent_reduceEl"}57// CHECK: %[[VAL_3:.*]]:2 = hlfir.declare %[[VAL_2]] {uniq_name = "_QFdo_concurrent_reduceEl"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)58// CHECK: %[[VAL_4:.*]] = fir.alloca i32 {bindc_name = "s", uniq_name = "_QFdo_concurrent_reduceEs"}59// CHECK: %[[VAL_5:.*]]:2 = hlfir.declare %[[VAL_4]] {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)60// CHECK: %[[VAL_6:.*]] = arith.constant 1 : index61// CHECK: omp.parallel {62// CHECK: %[[VAL_7:.*]] = fir.alloca i32 {bindc_name = "i"}63// CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_7]] {uniq_name = "_QFdo_concurrent_reduceEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)64// CHECK: omp.wsloop private(@_QFdo_concurrent_reduceEl_private_i32.omp %[[VAL_3]]#0 -> %[[VAL_9:.*]] : !fir.ref<i32>) reduction(@add_reduction_i32.omp %[[VAL_5]]#0 -> %[[VAL_10:.*]] : !fir.ref<i32>) {65// CHECK: omp.loop_nest (%[[VAL_11:.*]]) : index = (%[[VAL_6]]) to (%[[VAL_6]]) inclusive step (%[[VAL_6]]) {66// CHECK: %[[VAL_12:.*]] = fir.convert %[[VAL_11]] : (index) -> i3267// CHECK: fir.store %[[VAL_12]] to %[[VAL_8]]#0 : !fir.ref<i32>68// CHECK: %[[VAL_13:.*]]:2 = hlfir.declare %[[VAL_9]] {uniq_name = "_QFdo_concurrent_reduceEl"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)69// CHECK: %[[VAL_14:.*]]:2 = hlfir.declare %[[VAL_10]] {uniq_name = "_QFdo_concurrent_reduceEs"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)70// CHECK: %[[VAL_15:.*]] = arith.constant 1 : i3271// CHECK: hlfir.assign %[[VAL_15]] to %[[VAL_13]]#0 : i32, !fir.ref<i32>72// CHECK: %[[VAL_16:.*]] = fir.load %[[VAL_14]]#0 : !fir.ref<i32>73// CHECK: %[[VAL_17:.*]] = fir.load %[[VAL_13]]#0 : !fir.ref<i32>74// CHECK: %[[VAL_18:.*]] = arith.addi %[[VAL_16]], %[[VAL_17]] : i3275// CHECK: hlfir.assign %[[VAL_18]] to %[[VAL_14]]#0 : i32, !fir.ref<i32>76// CHECK: omp.yield77// CHECK: }78// CHECK: }79// CHECK: omp.terminator80// CHECK: }81// CHECK: return82// CHECK: }83 84