103 lines · plain
1// RUN: fir-opt --opt-bufferization %s | FileCheck %s2 3// leslie3d case with two allocatable module variables4// that cannot alias:5// module les3d_data6// implicit real*8 (a-h,o-z)7// integer imax, jmax, kmax8// double precision,allocatable,dimension(:,:,:,:,:) :: q9// double precision,allocatable,dimension(:,:,:,:) :: du10// end module les3d_data11// subroutine update()12// use les3d_data13// implicit real*8(a-h,o-z)14// i2 = imax - 115// do k = 1, kmax - 116// do j = 1, jmax - 117// q(1:i2,j,k,1,m) = (q(1:i2,j,k,1,m) + du(1:i2,j,k,1))18// end do19// end do20// end subroutine update21 22func.func @_QPupdate() {23 %c1 = arith.constant 1 : index24 %c0 = arith.constant 0 : index25 %c1_i32 = arith.constant 1 : i3226 %0 = fir.address_of(@_QMles3d_dataEdu) : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>>27 %1:2 = hlfir.declare %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QMles3d_dataEdu"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>>) -> (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>>, !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>>)28 %2 = fir.alloca i32 {bindc_name = "i2", uniq_name = "_QFupdateEi2"}29 %3:2 = hlfir.declare %2 {uniq_name = "_QFupdateEi2"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)30 %4 = fir.address_of(@_QMles3d_dataEimax) : !fir.ref<i32>31 %5:2 = hlfir.declare %4 {uniq_name = "_QMles3d_dataEimax"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)32 %6 = fir.alloca i32 {bindc_name = "j", uniq_name = "_QFupdateEj"}33 %7:2 = hlfir.declare %6 {uniq_name = "_QFupdateEj"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)34 %8 = fir.address_of(@_QMles3d_dataEjmax) : !fir.ref<i32>35 %9:2 = hlfir.declare %8 {uniq_name = "_QMles3d_dataEjmax"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)36 %10 = fir.alloca i32 {bindc_name = "k", uniq_name = "_QFupdateEk"}37 %11:2 = hlfir.declare %10 {uniq_name = "_QFupdateEk"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)38 %12 = fir.address_of(@_QMles3d_dataEkmax) : !fir.ref<i32>39 %13:2 = hlfir.declare %12 {uniq_name = "_QMles3d_dataEkmax"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)40 %14 = fir.alloca i32 {bindc_name = "m", uniq_name = "_QFupdateEm"}41 %15:2 = hlfir.declare %14 {uniq_name = "_QFupdateEm"} : (!fir.ref<i32>) -> (!fir.ref<i32>, !fir.ref<i32>)42 %16 = fir.address_of(@_QMles3d_dataEq) : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>>43 %17:2 = hlfir.declare %16 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QMles3d_dataEq"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>>) -> (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>>, !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>>)44 %18 = fir.load %5#0 : !fir.ref<i32>45 %19 = arith.subi %18, %c1_i32 : i3246 hlfir.assign %19 to %3#0 : i32, !fir.ref<i32>47 %20 = fir.load %13#0 : !fir.ref<i32>48 %21 = arith.subi %20, %c1_i32 : i3249 %22 = fir.convert %21 : (i32) -> index50 %23 = fir.convert %c1 : (index) -> i3251 %24:2 = fir.do_loop %arg0 = %c1 to %22 step %c1 iter_args(%arg1 = %23) -> (index, i32) {52 fir.store %arg1 to %11#1 : !fir.ref<i32>53 %25 = fir.load %9#0 : !fir.ref<i32>54 %26 = arith.subi %25, %c1_i32 : i3255 %27 = fir.convert %26 : (i32) -> index56 %28:2 = fir.do_loop %arg2 = %c1 to %27 step %c1 iter_args(%arg3 = %23) -> (index, i32) {57 fir.store %arg3 to %7#1 : !fir.ref<i32>58 %32 = fir.load %17#0 : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>>59 %33 = fir.load %3#0 : !fir.ref<i32>60 %34 = fir.convert %33 : (i32) -> index61 %35 = arith.cmpi sgt, %34, %c0 : index62 %36 = arith.select %35, %34, %c0 : index63 %37 = fir.load %7#0 : !fir.ref<i32>64 %38 = fir.convert %37 : (i32) -> i6465 %39 = fir.load %11#0 : !fir.ref<i32>66 %40 = fir.convert %39 : (i32) -> i6467 %41 = fir.load %15#0 : !fir.ref<i32>68 %42 = fir.convert %41 : (i32) -> i6469 %43 = fir.shape %36 : (index) -> !fir.shape<1>70 %44 = hlfir.designate %32 (%c1:%34:%c1, %38, %40, %c1, %42) shape %43 : (!fir.box<!fir.heap<!fir.array<?x?x?x?x?xf64>>>, index, index, index, i64, i64, index, i64, !fir.shape<1>) -> !fir.box<!fir.array<?xf64>>71 %45 = fir.load %1#0 : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>>72 %46 = hlfir.designate %45 (%c1:%34:%c1, %38, %40, %c1) shape %43 : (!fir.box<!fir.heap<!fir.array<?x?x?x?xf64>>>, index, index, index, i64, i64, index, !fir.shape<1>) -> !fir.box<!fir.array<?xf64>>73 %47 = hlfir.elemental %43 unordered : (!fir.shape<1>) -> !hlfir.expr<?xf64> {74 ^bb0(%arg4: index):75 %51 = hlfir.designate %44 (%arg4) : (!fir.box<!fir.array<?xf64>>, index) -> !fir.ref<f64>76 %52 = hlfir.designate %46 (%arg4) : (!fir.box<!fir.array<?xf64>>, index) -> !fir.ref<f64>77 %53 = fir.load %51 : !fir.ref<f64>78 %54 = fir.load %52 : !fir.ref<f64>79 %55 = arith.addf %53, %54 fastmath<fast> : f6480 %56 = hlfir.no_reassoc %55 : f6481 hlfir.yield_element %56 : f6482 }83 hlfir.assign %47 to %44 : !hlfir.expr<?xf64>, !fir.box<!fir.array<?xf64>>84 hlfir.destroy %47 : !hlfir.expr<?xf64>85 %48 = arith.addi %arg2, %c1 : index86 %49 = fir.load %7#1 : !fir.ref<i32>87 %50 = arith.addi %49, %23 : i3288 fir.result %48, %50 : index, i3289 }90 fir.store %28#1 to %7#1 : !fir.ref<i32>91 %29 = arith.addi %arg0, %c1 : index92 %30 = fir.load %11#1 : !fir.ref<i32>93 %31 = arith.addi %30, %23 : i3294 fir.result %29, %31 : index, i3295 }96 fir.store %24#1 to %11#1 : !fir.ref<i32>97 return98}99// CHECK-LABEL: func.func @_QPupdate() {100// CHECK-NOT: hlfir.assign {{.*}}!fir.box<!fir.array<?xf64>>101// CHECK: hlfir.assign %{{.*}} to %{{.*}} : f64, !fir.ref<f64>102// CHECK-NOT: hlfir.assign {{.*}}!fir.box<!fir.array<?xf64>>103