50 lines · plain
1! RUN: %flang_fc1 -emit-hlfir -mmlir --enable-delayed-privatization-staging=true -o - %s | FileCheck %s2 3subroutine do_concurrent_with_locality_specs4 implicit none5 integer :: i, local_var, local_init_var6 7 do concurrent (i=1:10) local(local_var) local_init(local_init_var)8 if (i < 5) then9 local_var = 4210 else11 local_init_var = 8412 end if13 end do14end subroutine15 16! CHECK: fir.local {type = local_init} @[[LOCAL_INIT_SYM:.*]] : i32 copy {17! CHECK: ^bb0(%[[ORIG_VAL:.*]]: !fir.ref<i32>, %[[LOCAL_VAL:.*]]: !fir.ref<i32>):18! CHECK: %[[ORIG_VAL_LD:.*]] = fir.load %[[ORIG_VAL]] : !fir.ref<i32>19! CHECK: hlfir.assign %[[ORIG_VAL_LD]] to %[[LOCAL_VAL]] : i32, !fir.ref<i32>20! CHECK: fir.yield(%[[LOCAL_VAL]] : !fir.ref<i32>)21! CHECK: }22 23! CHECK: fir.local {type = local} @[[LOCAL_SYM:.*]] : i3224 25! CHECK-LABEL: func.func @_QPdo_concurrent_with_locality_specs() {26! CHECK: %[[ORIG_LOCAL_INIT_ALLOC:.*]] = fir.alloca i32 {bindc_name = "local_init_var", {{.*}}}27! CHECK: %[[ORIG_LOCAL_INIT_DECL:.*]]:2 = hlfir.declare %[[ORIG_LOCAL_INIT_ALLOC]]28 29! CHECK: %[[ORIG_LOCAL_ALLOC:.*]] = fir.alloca i32 {bindc_name = "local_var", {{.*}}}30! CHECK: %[[ORIG_LOCAL_DECL:.*]]:2 = hlfir.declare %[[ORIG_LOCAL_ALLOC]]31 32! CHECK: fir.do_concurrent {33! CHECK: %[[IV_DECL:.*]]:2 = hlfir.declare %{{.*}}34 35! CHECK: fir.do_concurrent.loop (%{{.*}}) = (%{{.*}}) to (%{{.*}}) step (%{{.*}}) local(@[[LOCAL_SYM]] %[[ORIG_LOCAL_DECL]]#0 -> %[[LOCAL_ARG:.*]], @[[LOCAL_INIT_SYM]] %[[ORIG_LOCAL_INIT_DECL]]#0 -> %[[LOCAL_INIT_ARG:.*]] : !fir.ref<i32>, !fir.ref<i32>) {36! CHECK: %[[LOCAL_DECL:.*]]:2 = hlfir.declare %[[LOCAL_ARG]]37! CHECK: %[[LOCAL_INIT_DECL:.*]]:2 = hlfir.declare %[[LOCAL_INIT_ARG]]38 39! CHECK: fir.if %{{.*}} {40! CHECK: %[[C42:.*]] = arith.constant 42 : i3241! CHECK: hlfir.assign %[[C42]] to %[[LOCAL_DECL]]#0 : i32, !fir.ref<i32>42! CHECK: } else {43! CHECK: %[[C84:.*]] = arith.constant 84 : i3244! CHECK: hlfir.assign %[[C84]] to %[[LOCAL_INIT_DECL]]#0 : i32, !fir.ref<i32>45! CHECK: }46! CHECK: }47! CHECK: }48! CHECK: return49! CHECK: }50