|
| 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 | +} |
0 commit comments