43 lines · plain
1// Test fir.do_concurrent.loop operation with array of symbol reference attributes2// This test demonstrates operations that have ArrayAttr containing SymbolRefAttr elements3 4// RUN: fir-opt %s --external-name-interop | fir-opt | FileCheck %s5 6// Define reduction operations that will be referenced in the symbol array7func.func @_QPadd_reduction_i32_init(%arg0: i32, %arg1: !fir.ref<i32>) {8 %0 = arith.constant 0 : i329 fir.store %0 to %arg1 : !fir.ref<i32>10 return11}12 13func.func @_QPadd_reduction_i32_combiner(%arg0: i32, %arg1: i32) -> i32 {14 %0 = arith.addi %arg0, %arg1 : i3215 return %0 : i3216}17 18// Define a local privatizer that will be referenced in local_syms19func.func @_QPlocal_var_privatizer(%arg0: !fir.ref<i32>) -> !fir.ref<i32> {20 return %arg0 : !fir.ref<i32>21}22 23// Test function demonstrating both local_syms and reduce_syms arrays24func.func @_QPtest_symbol_arrays(%i_lb: index, %i_ub: index, %i_st: index) {25 %local_var = fir.alloca i3226 %sum = fir.alloca i3227 28 fir.do_concurrent {29 %i = fir.alloca i3230 fir.do_concurrent.loop (%i_iv) = (%i_lb) to (%i_ub) step (%i_st)31 local(@_QPlocal_var_privatizer %local_var -> %local_arg : !fir.ref<i32>)32 reduce(@_QPadd_reduction_i32_init #fir.reduce_attr<add> %sum -> %sum_arg : !fir.ref<i32>) {33 %0 = fir.convert %i_iv : (index) -> i3234 fir.store %0 to %i : !fir.ref<i32>35 }36 }37 return38}39 40// CHECK: local(@local_var_privatizer_41// CHECK: reduce(@add_reduction_i32_init_42 43