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 : !fir.ref<i32> alloc {
  21 ^bb0(%arg0: !fir.ref<i32>):
  22   %0 = fir.alloca i32 {bindc_name = "i", pinned, uniq_name = "_QFEi"}
  23   %1:2 = hlfir.declare %0 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  24   omp.yield(%1#0 : !fir.ref<i32>)
  25 }
  26 omp.private {type = firstprivate} @_QFEptra_firstprivate_ref_box_ptr_Uxi32 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>> alloc {
  27 ^bb0(%arg0: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>):
  28   %0 = fir.alloca !fir.box<!fir.ptr<!fir.array<?xi32>>> {bindc_name = "ptra", pinned, uniq_name = "_QFEptra"}
  29   %1:2 = hlfir.declare %0 {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>>>>)
  30   omp.yield(%1#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>)
  31 } copy {
  32 ^bb0(%arg0: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>, %arg1: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>):
  33   %0 = fir.load %arg0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  34   fir.store %0 to %arg1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  35   omp.yield(%arg1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>)
  36 }
  37 func.func @_QQmain() attributes {fir.bindc_name = "main"} {
  38   %0 = fir.address_of(@_QFEarraya) : !fir.ref<!fir.array<10xi32>>
  39   %c10 = arith.constant 10 : index
  40   %1 = fir.shape %c10 : (index) -> !fir.shape<1>
  41   %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>>)
  42   %3 = fir.address_of(@_QFEarrayb) : !fir.ref<!fir.array<10xi32>>
  43   %c10_0 = arith.constant 10 : index
  44   %4 = fir.shape %c10_0 : (index) -> !fir.shape<1>
  45   %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>>)
  46   %6 = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
  47   %7:2 = hlfir.declare %6 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  48   %8 = fir.address_of(@_QFEptra) : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  49   %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>>>>)
  50   %10 = fir.shape %c10 : (index) -> !fir.shape<1>
  51   %11 = fir.embox %2#1(%10) : (!fir.ref<!fir.array<10xi32>>, !fir.shape<1>) -> !fir.box<!fir.ptr<!fir.array<?xi32>>>
  52   fir.store %11 to %9#1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  53   omp.teams {
  54     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>) {
  55       %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>>>>)
  56       %13:2 = hlfir.declare %arg1 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  57       %c1_i32 = arith.constant 1 : i32
  58       %c10_i32 = arith.constant 10 : i32
  59       %c1_i32_1 = arith.constant 1 : i32
  60       omp.distribute {
  61         omp.wsloop {
  62           omp.loop_nest (%arg2) : i32 = (%c1_i32) to (%c10_i32) inclusive step (%c1_i32_1) {
  63             fir.store %arg2 to %13#1 : !fir.ref<i32>
  64             %14 = fir.load %13#0 : !fir.ref<i32>
  65             %15 = fir.convert %14 : (i32) -> i64
  66             %16 = hlfir.designate %2#0 (%15)  : (!fir.ref<!fir.array<10xi32>>, i64) -> !fir.ref<i32>
  67             %17 = fir.load %16 : !fir.ref<i32>
  68             %18 = fir.load %12#0 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xi32>>>>
  69             %19 = fir.load %13#0 : !fir.ref<i32>
  70             %20 = fir.convert %19 : (i32) -> i64
  71             %21 = hlfir.designate %18 (%20) {test.ptr = "ptrA" } : (!fir.box<!fir.ptr<!fir.array<?xi32>>>, i64) -> !fir.ref<i32>
  72             %22 = fir.load %21 : !fir.ref<i32>
  73             %23 = arith.addi %17, %22 : i32
  74             %24 = fir.load %13#0 : !fir.ref<i32>
  75             %25 = fir.convert %24 : (i32) -> i64
  76             %26 = hlfir.designate %2#0 (%25) {test.ptr = "ArrayA"}  : (!fir.ref<!fir.array<10xi32>>, i64) -> !fir.ref<i32>
  77             hlfir.assign %23 to %26 : i32, !fir.ref<i32>
  78             omp.yield
  79           }
  80         } {omp.composite}
  81       } {omp.composite}
  82       omp.terminator
  83     } {omp.composite}
  84     omp.terminator
  85   }
  86   return
  87 }
  88 fir.global internal @_QFEarraya target : !fir.array<10xi32> {
  89   %0 = fir.zero_bits !fir.array<10xi32>
  90   fir.has_value %0 : !fir.array<10xi32>
  91 }
  92 fir.global internal @_QFEarrayb : !fir.array<10xi32> {
  93   %0 = fir.zero_bits !fir.array<10xi32>
  94   fir.has_value %0 : !fir.array<10xi32>
  95 }
  96 fir.global internal @_QFEptra : !fir.box<!fir.ptr<!fir.array<?xi32>>> {
  97   %0 = fir.zero_bits !fir.ptr<!fir.array<?xi32>>
  98   %c0 = arith.constant 0 : index
  99   %1 = fir.shape %c0 : (index) -> !fir.shape<1>
 100   %2 = fir.embox %0(%1) : (!fir.ptr<!fir.array<?xi32>>, !fir.shape<1>) -> !fir.box<!fir.ptr<!fir.array<?xi32>>>
 101   fir.has_value %2 : !fir.box<!fir.ptr<!fir.array<?xi32>>>
 102 }