xref: /llvm-project/flang/test/Transforms/generic-loop-rewriting.mlir (revision 1e2d5f7943d09d658a5fbacf661d2c6c361f857c)
181f544d4SKareem Ergawy// RUN: fir-opt --omp-generic-loop-conversion %s | FileCheck %s
281f544d4SKareem Ergawy
3*1e2d5f79SKareem Ergawyomp.private {type = private} @_QFteams_loopEi_private_ref_i32 : !fir.ref<i32> alloc {
481f544d4SKareem Ergawy^bb0(%arg0: !fir.ref<i32>):
581f544d4SKareem Ergawy  omp.yield(%arg0 : !fir.ref<i32>)
681f544d4SKareem Ergawy}
781f544d4SKareem Ergawy
8*1e2d5f79SKareem Ergawyfunc.func @_QPteams_loop() {
981f544d4SKareem Ergawy  %i = fir.alloca i32
1081f544d4SKareem Ergawy  omp.teams {
1181f544d4SKareem Ergawy    %c0 = arith.constant 0 : i32
1281f544d4SKareem Ergawy    %c10 = arith.constant 10 : i32
1381f544d4SKareem Ergawy    %c1 = arith.constant 1 : i32
14*1e2d5f79SKareem Ergawy    omp.loop private(@_QFteams_loopEi_private_ref_i32 %i -> %arg2 : !fir.ref<i32>) {
1581f544d4SKareem Ergawy      omp.loop_nest (%arg3) : i32 = (%c0) to (%c10) inclusive step (%c1) {
1681f544d4SKareem Ergawy        fir.store %arg3 to %arg2 : !fir.ref<i32>
1781f544d4SKareem Ergawy        omp.yield
1881f544d4SKareem Ergawy      }
1981f544d4SKareem Ergawy    }
2081f544d4SKareem Ergawy    omp.terminator
2181f544d4SKareem Ergawy  }
2281f544d4SKareem Ergawy  return
2381f544d4SKareem Ergawy}
2481f544d4SKareem Ergawy
25*1e2d5f79SKareem Ergawy// CHECK-LABEL: func.func @_QPteams_loop
26*1e2d5f79SKareem Ergawy// CHECK:         %[[I:.*]] = fir.alloca i32
2781f544d4SKareem Ergawy// CHECK:         omp.teams {
2881f544d4SKareem Ergawy//
2981f544d4SKareem Ergawy// TODO we probably need to move the `loop_nest` bounds ops from the `teams`
3081f544d4SKareem Ergawy// region to the `parallel` region to avoid making these values `shared`. We can
3181f544d4SKareem Ergawy// find the backward slices of these bounds that are within the `teams` region
3281f544d4SKareem Ergawy// and move these slices to the `parallel` op.
3381f544d4SKareem Ergawy
3481f544d4SKareem Ergawy// CHECK:           %[[LB:.*]] = arith.constant 0 : i32
3581f544d4SKareem Ergawy// CHECK:           %[[UB:.*]] = arith.constant 10 : i32
3681f544d4SKareem Ergawy// CHECK:           %[[STEP:.*]] = arith.constant 1 : i32
3781f544d4SKareem Ergawy//
38*1e2d5f79SKareem Ergawy// CHECK:           omp.parallel private(@{{.*}} %[[I]]
3981f544d4SKareem Ergawy// CHECK-SAME:        -> %[[I_PRIV_ARG:[^[:space:]]+]] : !fir.ref<i32>) {
4081f544d4SKareem Ergawy// CHECK:             omp.distribute {
4181f544d4SKareem Ergawy// CHECK:               omp.wsloop {
4281f544d4SKareem Ergawy//
4381f544d4SKareem Ergawy// CHECK:                 omp.loop_nest (%{{.*}}) : i32 =
4481f544d4SKareem Ergawy// CHECK-SAME:              (%[[LB]]) to (%[[UB]]) inclusive step (%[[STEP]]) {
4581f544d4SKareem Ergawy// CHECK:                   fir.store %{{.*}} to %[[I_PRIV_ARG]] : !fir.ref<i32>
4681f544d4SKareem Ergawy// CHECK:                 }
4781f544d4SKareem Ergawy// CHECK:               }
4881f544d4SKareem Ergawy// CHECK:             }
4981f544d4SKareem Ergawy// CHECK:           }
5081f544d4SKareem Ergawy// CHECK:         }
51