flang/test/Analysis/AliasAnalysis/alias-analysis-omp-teams-distribute-private.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 //
   7 // program main
   8 // integer :: arrayA(10,10)
   9 // integer :: tmp(2)
  10 // integer :: i,j
  11 // !$omp teams distribute parallel do private(tmp)
  12 // do j = 1, 10
  13 //   do i = 1,10
  14 //     tmp = [i,j]
  15 //     arrayA = tmp(1)
  16 //   end do
  17 // end do
  18 // end program main
  19
  20 // CHECK-LABEL: Testing : "_QQmain"
  21 // CHECK-DAG: tmp_private_array#0 <-> unnamed_array#0: NoAlias
  22 // CHECK-DAG: tmp_private_array#1 <-> unnamed_array#0: NoAlias
  23
  24 omp.private {type = private} @_QFEi_private_ref_i32 : !fir.ref<i32> alloc {
  25 ^bb0(%arg0: !fir.ref<i32>):
  26   %0 = fir.alloca i32 {bindc_name = "i", pinned, uniq_name = "_QFEi"}
  27   %1:2 = hlfir.declare %0 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  28   omp.yield(%1#0 : !fir.ref<i32>)
  29 }
  30 omp.private {type = private} @_QFEj_private_ref_i32 : !fir.ref<i32> alloc {
  31 ^bb0(%arg0: !fir.ref<i32>):
  32   %0 = fir.alloca i32 {bindc_name = "j", pinned, uniq_name = "_QFEj"}
  33   %1:2 = hlfir.declare %0 {uniq_name = "_QFEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  34   omp.yield(%1#0 : !fir.ref<i32>)
  35 }
  36 omp.private {type = private} @_QFEtmp_private_ref_2xi32 : !fir.ref<!fir.array<2xi32>> alloc {
  37 ^bb0(%arg0: !fir.ref<!fir.array<2xi32>>):
  38   %c2 = arith.constant 2 : index
  39   %0 = fir.alloca !fir.array<2xi32> {bindc_name = "tmp", pinned, uniq_name = "_QFEtmp"}
  40   %1 = fir.shape %c2 : (index) -> !fir.shape<1>
  41   %2:2 = hlfir.declare %0(%1) {uniq_name = "_QFEtmp"} : (!fir.ref<!fir.array<2xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<2xi32>>, !fir.ref<!fir.array<2xi32>>)
  42   omp.yield(%2#0 : !fir.ref<!fir.array<2xi32>>)
  43 }
  44 func.func @_QQmain() attributes {fir.bindc_name = "main"} {
  45   %0 = fir.address_of(@_QFEarraya) : !fir.ref<!fir.array<10x10xi32>>
  46   %c10 = arith.constant 10 : index
  47   %c10_0 = arith.constant 10 : index
  48   %1 = fir.shape %c10, %c10_0 : (index, index) -> !fir.shape<2>
  49   %2:2 = hlfir.declare %0(%1) {uniq_name = "_QFEarraya"} : (!fir.ref<!fir.array<10x10xi32>>, !fir.shape<2>) -> (!fir.ref<!fir.array<10x10xi32>>, !fir.ref<!fir.array<10x10xi32>>)
  50   %3 = fir.alloca i32 {bindc_name = "i", uniq_name = "_QFEi"}
  51   %4:2 = hlfir.declare %3 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  52   %5 = fir.alloca i32 {bindc_name = "j", uniq_name = "_QFEj"}
  53   %6:2 = hlfir.declare %5 {uniq_name = "_QFEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  54   %c2 = arith.constant 2 : index
  55   %7 = fir.alloca !fir.array<2xi32> {bindc_name = "tmp", uniq_name = "_QFEtmp"}
  56   %8 = fir.shape %c2 : (index) -> !fir.shape<1>
  57   %9:2 = hlfir.declare %7(%8) {uniq_name = "_QFEtmp"} : (!fir.ref<!fir.array<2xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<2xi32>>, !fir.ref<!fir.array<2xi32>>)
  58   omp.teams {
  59     omp.parallel private(@_QFEtmp_private_ref_2xi32 %9#0 -> %arg0, @_QFEj_private_ref_i32 %6#0 -> %arg1, @_QFEi_private_ref_i32 %4#0 -> %arg2 : !fir.ref<!fir.array<2xi32>>, !fir.ref<i32>, !fir.ref<i32>) {
  60       %c2_1 = arith.constant 2 : index
  61       %10 = fir.shape %c2_1 : (index) -> !fir.shape<1>
  62       %11:2 = hlfir.declare %arg0(%10) {uniq_name = "_QFEtmp", test.ptr = "tmp_private_array"} : (!fir.ref<!fir.array<2xi32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<2xi32>>, !fir.ref<!fir.array<2xi32>>)
  63       %12:2 = hlfir.declare %arg1 {uniq_name = "_QFEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  64       %13:2 = hlfir.declare %arg2 {uniq_name = "_QFEi"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)
  65       %c1_i32 = arith.constant 1 : i32
  66       %c10_i32 = arith.constant 10 : i32
  67       %c1_i32_2 = arith.constant 1 : i32
  68       omp.distribute {
  69         omp.wsloop {
  70           omp.loop_nest (%arg3) : i32 = (%c1_i32) to (%c10_i32) inclusive step (%c1_i32_2) {
  71             fir.store %arg3 to %12#1 : !fir.ref<i32>
  72             %c1_i32_3 = arith.constant 1 : i32
  73             %14 = fir.convert %c1_i32_3 : (i32) -> index
  74             %c10_i32_4 = arith.constant 10 : i32
  75             %15 = fir.convert %c10_i32_4 : (i32) -> index
  76             %c1 = arith.constant 1 : index
  77             %16 = fir.convert %14 : (index) -> i32
  78             %17:2 = fir.do_loop %arg4 = %14 to %15 step %c1 iter_args(%arg5 = %16) -> (index, i32) {
  79               fir.store %arg5 to %13#1 : !fir.ref<i32>
  80               %c2_5 = arith.constant 2 : index
  81               %c1_6 = arith.constant 1 : index
  82               %c1_7 = arith.constant 1 : index
  83               %18 = fir.allocmem !fir.array<2xi32> {bindc_name = ".tmp.arrayctor", uniq_name = ""}
  84               %19 = fir.shape %c2_5 : (index) -> !fir.shape<1>
  85               %20:2 = hlfir.declare %18(%19) {uniq_name = ".tmp.arrayctor"} : (!fir.heap<!fir.array<2xi32>>, !fir.shape<1>) -> (!fir.heap<!fir.array<2xi32>>, !fir.heap<!fir.array<2xi32>>)
  86               %21 = fir.load %13#0 : !fir.ref<i32>
  87               %22 = arith.addi %c1_6, %c1_7 : index
  88               %23 = hlfir.designate %20#0 (%c1_6)  : (!fir.heap<!fir.array<2xi32>>, index) -> !fir.ref<i32>
  89               hlfir.assign %21 to %23 : i32, !fir.ref<i32>
  90               %24 = fir.load %12#0 : !fir.ref<i32>
  91               %25 = hlfir.designate %20#0 (%22)  : (!fir.heap<!fir.array<2xi32>>, index) -> !fir.ref<i32>
  92               hlfir.assign %24 to %25 : i32, !fir.ref<i32>
  93               %true = arith.constant true
  94               %26 = hlfir.as_expr %20#0 move %true {test.ptr = "unnamed_array"} : (!fir.heap<!fir.array<2xi32>>, i1) -> !hlfir.expr<2xi32>
  95               hlfir.assign %26 to %11#0 : !hlfir.expr<2xi32>, !fir.ref<!fir.array<2xi32>>
  96               hlfir.destroy %26 : !hlfir.expr<2xi32>
  97               %c1_8 = arith.constant 1 : index
  98               %27 = hlfir.designate %11#0 (%c1_8)  : (!fir.ref<!fir.array<2xi32>>, index) -> !fir.ref<i32>
  99               %28 = fir.load %27 : !fir.ref<i32>
 100               hlfir.assign %28 to %2#0 : i32, !fir.ref<!fir.array<10x10xi32>>
 101               %29 = arith.addi %arg4, %c1 : index
 102               %30 = fir.convert %c1 : (index) -> i32
 103               %31 = fir.load %13#1 : !fir.ref<i32>
 104               %32 = arith.addi %31, %30 : i32
 105               fir.result %29, %32 : index, i32
 106             }
 107             fir.store %17#1 to %13#1 : !fir.ref<i32>
 108             omp.yield
 109           }
 110         } {omp.composite}
 111       } {omp.composite}
 112       omp.terminator
 113     } {omp.composite}
 114     omp.terminator
 115   }
 116   return
 117 }
 118 fir.global internal @_QFEarraya : !fir.array<10x10xi32> {
 119   %0 = fir.zero_bits !fir.array<10x10xi32>
 120   fir.has_value %0 : !fir.array<10x10xi32>
 121 }