flang/test/Fir/OpenACC/propagate-attr-folding.fir

   1 // RUN: fir-opt %s --opt-bufferization | FileCheck %s
   2
   3 // Check that OpenACC attributes are propagated to the defining operations when
   4 // fir.box_addr is folded in bufferization optimization.
   5
   6 func.func @_QPsub1(%arg0: !fir.ref<!fir.array<?x?xf32>> {fir.bindc_name = "a"}, %arg1: !fir.ref<i32> {fir.bindc_name = "n1"}, %arg2: !fir.ref<i32> {fir.bindc_name = "n2"}) {
   7   %c1 = arith.constant 1 : index
   8   %c0 = arith.constant 0 : index
   9   %0 = fir.declare %arg1 {uniq_name = "_QFsub1En1"} : (!fir.ref<i32>) -> !fir.ref<i32>
  10   %1 = fir.declare %arg2 {uniq_name = "_QFsub1En2"} : (!fir.ref<i32>) -> !fir.ref<i32>
  11   %2 = fir.load %0 : !fir.ref<i32>
  12   %3 = fir.convert %2 : (i32) -> index
  13   %4 = arith.cmpi sgt, %3, %c0 : index
  14   %5 = arith.select %4, %3, %c0 : index
  15   %6 = fir.load %1 : !fir.ref<i32>
  16   %7 = fir.convert %6 : (i32) -> index
  17   %8 = arith.cmpi sgt, %7, %c0 : index
  18   %9 = arith.select %8, %7, %c0 : index
  19   %10 = fir.shape %5, %9 : (index, index) -> !fir.shape<2>
  20   %11 = fir.declare %arg0(%10) {uniq_name = "_QFsub1Ea"} : (!fir.ref<!fir.array<?x?xf32>>, !fir.shape<2>) -> !fir.ref<!fir.array<?x?xf32>>
  21   %12 = fir.embox %11(%10) : (!fir.ref<!fir.array<?x?xf32>>, !fir.shape<2>) -> !fir.box<!fir.array<?x?xf32>>
  22   %13:3 = fir.box_dims %12, %c0 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
  23   %14 = arith.subi %13#1, %c1 : index
  24   %15 = acc.bounds lowerbound(%c0 : index) upperbound(%14 : index) extent(%13#1 : index) stride(%13#2 : index) startIdx(%c1 : index) {strideInBytes = true}
  25   %16 = arith.muli %13#2, %13#1 : index
  26   %17:3 = fir.box_dims %12, %c1 : (!fir.box<!fir.array<?x?xf32>>, index) -> (index, index, index)
  27   %18 = arith.subi %17#1, %c1 : index
  28   %19 = acc.bounds lowerbound(%c0 : index) upperbound(%18 : index) extent(%17#1 : index) stride(%16 : index) startIdx(%c1 : index) {strideInBytes = true}
  29   %20 = fir.box_addr %12 {acc.declare = #acc.declare<dataClause =  acc_present>} : (!fir.box<!fir.array<?x?xf32>>) -> !fir.ref<!fir.array<?x?xf32>>
  30   %21 = acc.present varPtr(%20 : !fir.ref<!fir.array<?x?xf32>>) bounds(%15, %19) -> !fir.ref<!fir.array<?x?xf32>> {name = "a"}
  31   %22 = acc.declare_enter dataOperands(%21 : !fir.ref<!fir.array<?x?xf32>>)
  32   acc.declare_exit token(%22)
  33   return
  34 }
  35
  36 // CHECK-LABEL: func.func @_QPsub1(
  37 // CHECK-SAME: %[[ARG0:.*]]: !fir.ref<!fir.array<?x?xf32>> {fir.bindc_name = "a"}
  38 // CHECK: %[[DECL:.*]] = fir.declare %[[ARG0]](%{{.*}}) {acc.declare = #acc.declare<dataClause =  acc_present>, uniq_name = "_QFsub1Ea"} : (!fir.ref<!fir.array<?x?xf32>>, !fir.shape<2>) -> !fir.ref<!fir.array<?x?xf32>>
  39 // CHECK: %[[PRES:.*]] = acc.present varPtr(%[[DECL]] : !fir.ref<!fir.array<?x?xf32>>) bounds(%{{.*}}, %{{.*}}) -> !fir.ref<!fir.array<?x?xf32>> {name = "a"}
  40 // CHECK: %{{.*}} = acc.declare_enter dataOperands(%[[PRES]] : !fir.ref<!fir.array<?x?xf32>>)