60 lines · plain
1// RUN: fir-opt --opt-bufferization %s | FileCheck %s2 3// tonto case where optimized bufferization conservatively4// did not pull elemental into the assignment due to5// the designate op in between:6// module test7// type my_type8// real, dimension(:,:), pointer :: p => null()9// end type my_type10// contains11// subroutine assign1(self, i)12// type(my_type) :: self13// real, dimension(3) :: ct, q14// integer :: i15// self%p(:,i) = q - ct16// end subroutine assign117// end module test18 19// CHECK-LABEL: func.func @_QMtestPassign1(20// CHECK-NOT: hlfir.elemental21// CHECK-NOT: hlfir.assign{{.*}}array22func.func @_QMtestPassign1(%arg0: !fir.ref<!fir.type<_QMtestTmy_type{p:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>}>> {fir.bindc_name = "self"}, %arg1: !fir.ref<i32> {fir.bindc_name = "i"}) {23 %c1 = arith.constant 1 : index24 %c0 = arith.constant 0 : index25 %c3 = arith.constant 3 : index26 %0 = fir.dummy_scope : !fir.dscope27 %1 = fir.alloca !fir.array<3xf32> {bindc_name = "ct", uniq_name = "_QMtestFassign1Ect"}28 %2 = fir.shape %c3 : (index) -> !fir.shape<1>29 %3:2 = hlfir.declare %1(%2) {uniq_name = "_QMtestFassign1Ect"} : (!fir.ref<!fir.array<3xf32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<3xf32>>, !fir.ref<!fir.array<3xf32>>)30 %4:2 = hlfir.declare %arg1 dummy_scope %0 {uniq_name = "_QMtestFassign1Ei"} : (!fir.ref<i32>, !fir.dscope) -> (!fir.ref<i32>, !fir.ref<i32>)31 %5 = fir.alloca !fir.array<3xf32> {bindc_name = "q", uniq_name = "_QMtestFassign1Eq"}32 %6:2 = hlfir.declare %5(%2) {uniq_name = "_QMtestFassign1Eq"} : (!fir.ref<!fir.array<3xf32>>, !fir.shape<1>) -> (!fir.ref<!fir.array<3xf32>>, !fir.ref<!fir.array<3xf32>>)33 %7:2 = hlfir.declare %arg0 dummy_scope %0 {uniq_name = "_QMtestFassign1Eself"} : (!fir.ref<!fir.type<_QMtestTmy_type{p:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>}>>, !fir.dscope) -> (!fir.ref<!fir.type<_QMtestTmy_type{p:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>}>>, !fir.ref<!fir.type<_QMtestTmy_type{p:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>}>>)34 %8 = hlfir.elemental %2 unordered : (!fir.shape<1>) -> !hlfir.expr<3xf32> {35 ^bb0(%arg2: index):36 %22 = hlfir.designate %6#0 (%arg2) : (!fir.ref<!fir.array<3xf32>>, index) -> !fir.ref<f32>37 %23 = hlfir.designate %3#0 (%arg2) : (!fir.ref<!fir.array<3xf32>>, index) -> !fir.ref<f32>38 %24 = fir.load %22 : !fir.ref<f32>39 %25 = fir.load %23 : !fir.ref<f32>40 %26 = arith.subf %24, %25 fastmath<contract> : f3241 hlfir.yield_element %26 : f3242 }43 %9 = hlfir.designate %7#0{"p"} {fortran_attrs = #fir.var_attrs<pointer>} : (!fir.ref<!fir.type<_QMtestTmy_type{p:!fir.box<!fir.ptr<!fir.array<?x?xf32>>>}>>) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>44 %10 = fir.load %9 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?x?xf32>>>>45 %11:3 = fir.box_dims %10, %c0 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index) -> (index, index, index)46 %12 = arith.addi %11#0, %11#1 : index47 %13 = arith.subi %12, %c1 : index48 %14 = arith.subi %13, %11#0 : index49 %15 = arith.addi %14, %c1 : index50 %16 = arith.cmpi sgt, %15, %c0 : index51 %17 = arith.select %16, %15, %c0 : index52 %18 = fir.load %4#0 : !fir.ref<i32>53 %19 = fir.convert %18 : (i32) -> i6454 %20 = fir.shape %17 : (index) -> !fir.shape<1>55 %21 = hlfir.designate %10 (%11#0:%13:%c1, %19) shape %20 : (!fir.box<!fir.ptr<!fir.array<?x?xf32>>>, index, index, index, i64, !fir.shape<1>) -> !fir.box<!fir.array<?xf32>>56 hlfir.assign %8 to %21 : !hlfir.expr<3xf32>, !fir.box<!fir.array<?xf32>>57 hlfir.destroy %8 : !hlfir.expr<3xf32>58 return59}60