flang/test/Analysis/AliasAnalysis/alias-analysis-omp-teams-distribute-private-ptr.mlir

   1 // Use --mlir-disable-threading so that the AA queries are serialized
   2 // as well as its diagnostic output.
   3 // RUN: fir-opt %s -pass-pipeline='builtin.module(func.func(test-fir-alias-analysis))' -split-input-file --mlir-disable-threading 2>&1 | FileCheck %s
   4
   5 // Fortran code:
   6 // program main
   7 // integer, target :: arrayA(10)
   8 // integer, pointer, dimension(:) :: ptrA
   9 // integer :: i
  10 // ptrA => arrayA
  11 // !$omp teams distribute parallel do firstprivate(ptrA)
  12 // do i = 1, 10
  13 //   arrayA(i) = arrayA(i) + ptrA(i);
  14 // end do
  15 // end program main
  16
  17 // CHECK-LABEL: Testing : "_QQmain"
  18 // CHECK-DAG:   ptrA#0 <-> ArrayA#0: MayAlias
  19
  20 omp.private {type = private} @_QFEi_private_ref_i32 : i32
  21 omp.private {type = firstprivate} @_QFEptra_firstprivate_ref_box_ptr_Uxi32 : !fir.box<!fir.ptr<!fir.array<?xi32>>> copy {
  22 ^bb0(%arg0: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>, %arg1: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>):
  23   %0 = fir.load %arg0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  24   fir.store %0 to %arg1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  25   omp.yield(%arg1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>)
  26 }
  27 func.func @_QQmain() attributes {fir.bindc_name = "main"} {
  28   %0 = fir.address_of(@_QFEarraya) : !fir.ref<!fir.array<10xi32>>
  29   %c10 = arith.constant 10 : index
  30   %1 = fir.shape %c10 : (index) -> !fir.shape<1>
  31   %2:2 = hlfir.declare %0(%1) {fortran_attrs = #fir.var_attrs<target>, uniq_name = "_QFEarraya"} : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<10xi32>>, !fir.ref<!fir.array<10xi32>>)
  32   %3 = fir.address_of(@_QFEarrayb) : !fir.ref<!fir.array<10xi32>>
  33   %c10_0 = arith.constant 10 : index
  34   %4 = fir.shape %c10_0 : (index) -> !fir.shape<1>
  35   %5:2 = hlfir.declare %3(%4) {uniq_name = "_QFEarrayb"} : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<10xi32>>, !fir.ref<!fir.array<10xi32>>)
  36   %6 = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
  37   %7:2 = hlfir.declare %6 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  38   %8 = fir.address_of(@_QFEptra) : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  39   %9:2 = hlfir.declare %8 {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFEptra"} : (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>) -> (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>, !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>)
  40   %10 = fir.shape %c10 : (index) -> !fir.shape<1>
  41   %11 = fir.embox %2#1(%10) : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> !fir.box<!fir.ptr<!fir.array<?xi32>>>
  42   fir.store %11 to %9#1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  43   omp.teams {
  44     omp.parallel private(@_QFEptra_firstprivate_ref_box_ptr_Uxi32 %9#0 -> %arg0, @_QFEi_private_ref_i32 %7#0 -> %arg1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>, !fir.ref<i32>) {
  45       %12:2 = hlfir.declare %arg0 {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFEptra"} : (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>) -> (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>, !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>)
  46       %13:2 = hlfir.declare %arg1 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  47       %c1_i32 = arith.constant 1 : i32
  48       %c10_i32 = arith.constant 10 : i32
  49       %c1_i32_1 = arith.constant 1 : i32
  50       omp.distribute {
  51         omp.wsloop {
  52           omp.loop_nest (%arg2) : i32 = (%c1_i32) to (%c10_i32) inclusive step (%c1_i32_1) {
  53             fir.store %arg2 to %13#1 : !fir.ref<i32>
  54             %14 = fir.load %13#0 : !fir.ref<i32>
  55             %15 = fir.convert %14 : (i32) -> i64
  56             %16 = hlfir.designate %2#0 (%15)  : (!fir.ref<!fir.array<10xi32>>, i64) -> !fir.ref<i32>
  57             %17 = fir.load %16 : !fir.ref<i32>
  58             %18 = fir.load %12#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  59             %19 = fir.load %13#0 : !fir.ref<i32>
  60             %20 = fir.convert %19 : (i32) -> i64
  61             %21 = hlfir.designate %18 (%20) {test.ptr = "ptrA" } : (!fir.box<!fir.ptr<!fir.array<?xi32>>>, i64) -> !fir.ref<i32>
  62             %22 = fir.load %21 : !fir.ref<i32>
  63             %23 = arith.addi %17, %22 : i32
  64             %24 = fir.load %13#0 : !fir.ref<i32>
  65             %25 = fir.convert %24 : (i32) -> i64
  66             %26 = hlfir.designate %2#0 (%25) {test.ptr = "ArrayA"}  : (!fir.ref<!fir.array<10xi32>>, i64) -> !fir.ref<i32>
  67             hlfir.assign %23 to %26 : i32, !fir.ref<i32>
  68             omp.yield
  69           }
  70         } {omp.composite}
  71       } {omp.composite}
  72       omp.terminator
  73     } {omp.composite}
  74     omp.terminator
  75   }
  76   return
  77 }
  78 fir.global internal @_QFEarraya target : !fir.array<10xi32> {
  79   %0 = fir.zero_bits !fir.array<10xi32>
  80   fir.has_value %0 : !fir.array<10xi32>
  81 }
  82 fir.global internal @_QFEarrayb : !fir.array<10xi32> {
  83   %0 = fir.zero_bits !fir.array<10xi32>
  84   fir.has_value %0 : !fir.array<10xi32>
  85 }
  86 fir.global internal @_QFEptra : !fir.box<!fir.ptr<!fir.array<?xi32>>> {
  87   %0 = fir.zero_bits !fir.ptr<!fir.array<?xi32>>
  88   %c0 = arith.constant 0 : index
  89   %1 = fir.shape %c0 : (index) -> !fir.shape<1>
  90   %2 = fir.embox %0(%1) : (!fir.ptr<!fir.array<?xi32>>, !fir.shape<1>) -> !fir.box<!fir.ptr<!fir.array<?xi32>>>
  91   fir.has_value %2 : !fir.box<!fir.ptr<!fir.array<?xi32>>>
  92 }