mirror of
https://github.com/intel/llvm.git
synced 2026-01-27 14:50:42 +08:00
252efeba90
A long chain of fir.pack_arrays might require multiple iterations of the greedy rewriter, if we use down-top traversal. The rewriter may not converge in 10 (default) iterations. It is not an error, but it was reported as such. This patch changes the traversal to top-down and also disabled the hard error, if the rewriter does not converge soon enough.
794 lines
45 KiB
Plaintext
794 lines
45 KiB
Plaintext
// Test that the redundant fir.[un]pack_array operations
|
|
// are optimized away, when the source is statically known
|
|
// to be contiguous.
|
|
// RUN: fir-opt --optimize-array-repacking %s | FileCheck %s
|
|
|
|
// FIR is produced by compiling the sources with -mllvm -inline-all.
|
|
// module inner
|
|
// contains
|
|
// subroutine inner_repack1(x)
|
|
// real :: x(:)
|
|
// end subroutine inner_repack1
|
|
// subroutine inner_repack2(x)
|
|
// real :: x(:,:)
|
|
// end subroutine inner_repack2
|
|
// end module inner
|
|
|
|
// subroutine test_explicit_shape_cst(x)
|
|
// real :: x(100)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_explicit_shape_cst
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_explicit_shape_cst(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_explicit_shape_cst(%arg0: !fir.ref<!fir.array<100xf32>> {fir.bindc_name = "x"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c100 = arith.constant 100 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.shape %c100 : (index) -> !fir.shape<1>
|
|
%2 = fir.declare %arg0(%1) dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_cstEx"} : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<100xf32>>
|
|
%3 = fir.shape %c50 : (index) -> !fir.shape<1>
|
|
%4 = fir.array_coor %2(%1) %c1 : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, index) -> !fir.ref<f32>
|
|
%5 = fir.convert %4 : (!fir.ref<f32>) -> !fir.ref<!fir.array<50xf32>>
|
|
%6 = fir.embox %5(%3) : (!fir.ref<!fir.array<50xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%7 = fir.convert %6 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFtest_explicit_shape_cstFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_explicit_shape_var(x, n, l, u)
|
|
// integer :: n, l, u
|
|
// real :: x(n)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_explicit_shape_var
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_explicit_shape_var(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_explicit_shape_var(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "n"}, %arg2: !fir.ref<i32> {fir.bindc_name = "l"}, %arg3: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c0 = arith.constant 0 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_varEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_varEn"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.declare %arg3 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_varEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%4 = fir.load %2 : !fir.ref<i32>
|
|
%5 = fir.convert %4 : (i32) -> index
|
|
%6 = arith.cmpi sgt, %5, %c0 : index
|
|
%7 = arith.select %6, %5, %c0 : index
|
|
%8 = fir.shape %7 : (index) -> !fir.shape<1>
|
|
%9 = fir.declare %arg0(%8) dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_varEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%10 = fir.load %1 : !fir.ref<i32>
|
|
%11 = fir.load %3 : !fir.ref<i32>
|
|
%12 = fir.convert %10 : (i32) -> index
|
|
%13 = fir.convert %11 : (i32) -> index
|
|
%14 = fir.slice %12, %13, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%15 = fir.embox %9(%8) [%14] : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%16 = fir.dummy_scope : !fir.dscope
|
|
%17 = fir.pack_array %15 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%18 = fir.declare %17 dummy_scope %16 {uniq_name = "_QFtest_explicit_shape_varFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %17 to %15 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_assumed_size_cst(x)
|
|
// real :: x(*)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_assumed_size_cst
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_assumed_size_cst(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_assumed_size_cst(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c-1 = arith.constant -1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.shape %c-1 : (index) -> !fir.shape<1>
|
|
%2 = fir.declare %arg0(%1) dummy_scope %0 {uniq_name = "_QFtest_assumed_size_cstEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%3 = fir.shape %c50 : (index) -> !fir.shape<1>
|
|
%4 = fir.array_coor %2(%1) %c1 : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, index) -> !fir.ref<f32>
|
|
%5 = fir.convert %4 : (!fir.ref<f32>) -> !fir.ref<!fir.array<50xf32>>
|
|
%6 = fir.embox %5(%3) : (!fir.ref<!fir.array<50xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%7 = fir.convert %6 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFtest_assumed_size_cstFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_assumed_size_var(x, l, u)
|
|
// integer :: l, u
|
|
// real :: x(*)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_assumed_size_var
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_assumed_size_var(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_assumed_size_var(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "l"}, %arg2: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c-1 = arith.constant -1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_assumed_size_varEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_assumed_size_varEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.shape %c-1 : (index) -> !fir.shape<1>
|
|
%4 = fir.declare %arg0(%3) dummy_scope %0 {uniq_name = "_QFtest_assumed_size_varEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%5 = fir.load %1 : !fir.ref<i32>
|
|
%6 = fir.load %2 : !fir.ref<i32>
|
|
%7 = fir.convert %5 : (i32) -> index
|
|
%8 = fir.convert %6 : (i32) -> index
|
|
%9 = fir.slice %7, %8, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%10 = fir.embox %4(%3) [%9] : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.dummy_scope : !fir.dscope
|
|
%12 = fir.pack_array %10 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%13 = fir.declare %12 dummy_scope %11 {uniq_name = "_QFtest_assumed_size_varFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %12 to %10 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_cst(x)
|
|
// real, allocatable :: x(:)
|
|
// call repack(x(10:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_cst
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_cst(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_cst(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>> {fir.bindc_name = "x"}) {
|
|
%c0 = arith.constant 0 : index
|
|
%c41 = arith.constant 41 : index
|
|
%c10 = arith.constant 10 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_cstEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%2 = fir.load %1 : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%3 = fir.shape %c41 : (index) -> !fir.shape<1>
|
|
%4 = fir.box_addr %2 : (!fir.box<!fir.heap<!fir.array<?xf32>>>) -> !fir.heap<!fir.array<?xf32>>
|
|
%5:3 = fir.box_dims %2, %c0 : (!fir.box<!fir.heap<!fir.array<?xf32>>>, index) -> (index, index, index)
|
|
%6 = fir.shape_shift %5#0, %5#1 : (index, index) -> !fir.shapeshift<1>
|
|
%7 = fir.array_coor %4(%6) %c10 : (!fir.heap<!fir.array<?xf32>>, !fir.shapeshift<1>, index) -> !fir.ref<f32>
|
|
%8 = fir.convert %7 : (!fir.ref<f32>) -> !fir.ref<!fir.array<41xf32>>
|
|
%9 = fir.embox %8(%3) : (!fir.ref<!fir.array<41xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<41xf32>>
|
|
%10 = fir.convert %9 : (!fir.box<!fir.array<41xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.dummy_scope : !fir.dscope
|
|
%12 = fir.pack_array %10 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%13 = fir.declare %12 dummy_scope %11 {uniq_name = "_QFtest_allocatable_cstFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %12 to %10 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_var(x, l, u)
|
|
// integer :: l, u
|
|
// real, allocatable :: x(:)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_var
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_var(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_var(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "l"}, %arg2: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c0 = arith.constant 0 : index
|
|
%c1 = arith.constant 1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_allocatable_varEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_allocatable_varEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_varEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%4 = fir.load %3 : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%5 = fir.load %1 : !fir.ref<i32>
|
|
%6 = fir.load %2 : !fir.ref<i32>
|
|
%7 = fir.convert %5 : (i32) -> index
|
|
%8 = fir.convert %6 : (i32) -> index
|
|
%9 = fir.box_addr %4 : (!fir.box<!fir.heap<!fir.array<?xf32>>>) -> !fir.heap<!fir.array<?xf32>>
|
|
%10:3 = fir.box_dims %4, %c0 : (!fir.box<!fir.heap<!fir.array<?xf32>>>, index) -> (index, index, index)
|
|
%11 = fir.shape_shift %10#0, %10#1 : (index, index) -> !fir.shapeshift<1>
|
|
%12 = fir.slice %7, %8, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%13 = fir.embox %9(%11) [%12] : (!fir.heap<!fir.array<?xf32>>, !fir.shapeshift<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.dummy_scope : !fir.dscope
|
|
%15 = fir.pack_array %13 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%16 = fir.declare %15 dummy_scope %14 {uniq_name = "_QFtest_allocatable_varFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %15 to %13 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_full(x)
|
|
// real, allocatable :: x(:,:)
|
|
// call repack(x(:,:))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:,:)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_full
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_full(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_full(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>> {fir.bindc_name = "x"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c0 = arith.constant 0 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_fullEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>
|
|
%2 = fir.load %1 : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>
|
|
%3:3 = fir.box_dims %2, %c0 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>, index) -> (index, index, index)
|
|
%4:3 = fir.box_dims %2, %c1 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>, index) -> (index, index, index)
|
|
%5 = arith.addi %3#0, %3#1 : index
|
|
%6 = arith.subi %5, %c1 : index
|
|
%7 = arith.addi %4#0, %4#1 : index
|
|
%8 = arith.subi %7, %c1 : index
|
|
%9 = fir.box_addr %2 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>) -> !fir.heap<!fir.array<?x?xf32>>
|
|
%10 = fir.shape_shift %3#0, %3#1, %4#0, %4#1 : (index, index, index, index) -> !fir.shapeshift<2>
|
|
%11 = fir.slice %3#0, %6, %c1, %4#0, %8, %c1 : (index, index, index, index, index, index) -> !fir.slice<2>
|
|
%12 = fir.embox %9(%10) [%11] : (!fir.heap<!fir.array<?x?xf32>>, !fir.shapeshift<2>, !fir.slice<2>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%13 = fir.dummy_scope : !fir.dscope
|
|
%14 = fir.pack_array %12 heap innermost : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%15 = fir.declare %14 dummy_scope %13 {uniq_name = "_QFtest_allocatable_fullFrepackEx"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xf32>>
|
|
fir.unpack_array %14 to %12 heap : !fir.box<!fir.array<?x?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_explicit_shape_cst_chain(x)
|
|
// real :: x(100)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_explicit_shape_cst_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_explicit_shape_cst_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_explicit_shape_cst_chain(%arg0: !fir.ref<!fir.array<100xf32>> {fir.bindc_name = "x"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c100 = arith.constant 100 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.shape %c100 : (index) -> !fir.shape<1>
|
|
%2 = fir.declare %arg0(%1) dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_cst_chainEx"} : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<100xf32>>
|
|
%3 = fir.shape %c50 : (index) -> !fir.shape<1>
|
|
%4 = fir.array_coor %2(%1) %c1 : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, index) -> !fir.ref<f32>
|
|
%5 = fir.convert %4 : (!fir.ref<f32>) -> !fir.ref<!fir.array<50xf32>>
|
|
%6 = fir.embox %5(%3) : (!fir.ref<!fir.array<50xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%7 = fir.convert %6 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFtest_explicit_shape_cst_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.rebox %10 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%12 = fir.dummy_scope : !fir.dscope
|
|
%13 = fir.pack_array %11 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.declare %13 dummy_scope %12 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %13 to %11 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_explicit_shape_var_chain(x, n, l, u)
|
|
// integer :: n, l, u
|
|
// real :: x(n)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_explicit_shape_var_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_explicit_shape_var_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_explicit_shape_var_chain(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "n"}, %arg2: !fir.ref<i32> {fir.bindc_name = "l"}, %arg3: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c0 = arith.constant 0 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_var_chainEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_var_chainEn"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.declare %arg3 dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_var_chainEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%4 = fir.load %2 : !fir.ref<i32>
|
|
%5 = fir.convert %4 : (i32) -> index
|
|
%6 = arith.cmpi sgt, %5, %c0 : index
|
|
%7 = arith.select %6, %5, %c0 : index
|
|
%8 = fir.shape %7 : (index) -> !fir.shape<1>
|
|
%9 = fir.declare %arg0(%8) dummy_scope %0 {uniq_name = "_QFtest_explicit_shape_var_chainEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%10 = fir.load %1 : !fir.ref<i32>
|
|
%11 = fir.load %3 : !fir.ref<i32>
|
|
%12 = fir.convert %10 : (i32) -> index
|
|
%13 = fir.convert %11 : (i32) -> index
|
|
%14 = fir.slice %12, %13, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%15 = fir.embox %9(%8) [%14] : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%16 = fir.dummy_scope : !fir.dscope
|
|
%17 = fir.pack_array %15 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%18 = fir.declare %17 dummy_scope %16 {uniq_name = "_QFtest_explicit_shape_var_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%19 = fir.rebox %18 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%20 = fir.dummy_scope : !fir.dscope
|
|
%21 = fir.pack_array %19 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%22 = fir.declare %21 dummy_scope %20 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %21 to %19 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %17 to %15 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_assumed_size_cst_chain(x)
|
|
// real :: x(*)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_assumed_size_cst_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_assumed_size_cst_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_assumed_size_cst_chain(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c-1 = arith.constant -1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.shape %c-1 : (index) -> !fir.shape<1>
|
|
%2 = fir.declare %arg0(%1) dummy_scope %0 {uniq_name = "_QFtest_assumed_size_cst_chainEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%3 = fir.shape %c50 : (index) -> !fir.shape<1>
|
|
%4 = fir.array_coor %2(%1) %c1 : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, index) -> !fir.ref<f32>
|
|
%5 = fir.convert %4 : (!fir.ref<f32>) -> !fir.ref<!fir.array<50xf32>>
|
|
%6 = fir.embox %5(%3) : (!fir.ref<!fir.array<50xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%7 = fir.convert %6 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFtest_assumed_size_cst_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.rebox %10 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%12 = fir.dummy_scope : !fir.dscope
|
|
%13 = fir.pack_array %11 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.declare %13 dummy_scope %12 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %13 to %11 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_assumed_size_var_chain(x, l, u)
|
|
// integer :: l, u
|
|
// real :: x(*)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_assumed_size_var_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_assumed_size_var_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_assumed_size_var_chain(%arg0: !fir.ref<!fir.array<?xf32>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "l"}, %arg2: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c-1 = arith.constant -1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_assumed_size_var_chainEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_assumed_size_var_chainEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.shape %c-1 : (index) -> !fir.shape<1>
|
|
%4 = fir.declare %arg0(%3) dummy_scope %0 {uniq_name = "_QFtest_assumed_size_var_chainEx"} : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<?xf32>>
|
|
%5 = fir.load %1 : !fir.ref<i32>
|
|
%6 = fir.load %2 : !fir.ref<i32>
|
|
%7 = fir.convert %5 : (i32) -> index
|
|
%8 = fir.convert %6 : (i32) -> index
|
|
%9 = fir.slice %7, %8, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%10 = fir.embox %4(%3) [%9] : (!fir.ref<!fir.array<?xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.dummy_scope : !fir.dscope
|
|
%12 = fir.pack_array %10 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%13 = fir.declare %12 dummy_scope %11 {uniq_name = "_QFtest_assumed_size_var_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.rebox %13 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%15 = fir.dummy_scope : !fir.dscope
|
|
%16 = fir.pack_array %14 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%17 = fir.declare %16 dummy_scope %15 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %16 to %14 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %12 to %10 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_cst_chain(x)
|
|
// real, allocatable :: x(:)
|
|
// call repack(x(10:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_cst_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_cst_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_cst_chain(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>> {fir.bindc_name = "x"}) {
|
|
%c0 = arith.constant 0 : index
|
|
%c41 = arith.constant 41 : index
|
|
%c10 = arith.constant 10 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_cst_chainEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%2 = fir.load %1 : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%3 = fir.shape %c41 : (index) -> !fir.shape<1>
|
|
%4 = fir.box_addr %2 : (!fir.box<!fir.heap<!fir.array<?xf32>>>) -> !fir.heap<!fir.array<?xf32>>
|
|
%5:3 = fir.box_dims %2, %c0 : (!fir.box<!fir.heap<!fir.array<?xf32>>>, index) -> (index, index, index)
|
|
%6 = fir.shape_shift %5#0, %5#1 : (index, index) -> !fir.shapeshift<1>
|
|
%7 = fir.array_coor %4(%6) %c10 : (!fir.heap<!fir.array<?xf32>>, !fir.shapeshift<1>, index) -> !fir.ref<f32>
|
|
%8 = fir.convert %7 : (!fir.ref<f32>) -> !fir.ref<!fir.array<41xf32>>
|
|
%9 = fir.embox %8(%3) : (!fir.ref<!fir.array<41xf32>>, !fir.shape<1>) -> !fir.box<!fir.array<41xf32>>
|
|
%10 = fir.convert %9 : (!fir.box<!fir.array<41xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%11 = fir.dummy_scope : !fir.dscope
|
|
%12 = fir.pack_array %10 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%13 = fir.declare %12 dummy_scope %11 {uniq_name = "_QFtest_allocatable_cst_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.rebox %13 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%15 = fir.dummy_scope : !fir.dscope
|
|
%16 = fir.pack_array %14 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%17 = fir.declare %16 dummy_scope %15 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %16 to %14 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %12 to %10 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_var_chain(x, l, u)
|
|
// integer :: l, u
|
|
// real, allocatable :: x(:)
|
|
// call repack(x(l:u))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:)
|
|
// call inner_repack1(x)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_var_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_var_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_var_chain(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "l"}, %arg2: !fir.ref<i32> {fir.bindc_name = "u"}) {
|
|
%c0 = arith.constant 0 : index
|
|
%c1 = arith.constant 1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFtest_allocatable_var_chainEl"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.declare %arg2 dummy_scope %0 {uniq_name = "_QFtest_allocatable_var_chainEu"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%3 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_var_chainEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%4 = fir.load %3 : !fir.ref<!fir.box<!fir.heap<!fir.array<?xf32>>>>
|
|
%5 = fir.load %1 : !fir.ref<i32>
|
|
%6 = fir.load %2 : !fir.ref<i32>
|
|
%7 = fir.convert %5 : (i32) -> index
|
|
%8 = fir.convert %6 : (i32) -> index
|
|
%9 = fir.box_addr %4 : (!fir.box<!fir.heap<!fir.array<?xf32>>>) -> !fir.heap<!fir.array<?xf32>>
|
|
%10:3 = fir.box_dims %4, %c0 : (!fir.box<!fir.heap<!fir.array<?xf32>>>, index) -> (index, index, index)
|
|
%11 = fir.shape_shift %10#0, %10#1 : (index, index) -> !fir.shapeshift<1>
|
|
%12 = fir.slice %7, %8, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%13 = fir.embox %9(%11) [%12] : (!fir.heap<!fir.array<?xf32>>, !fir.shapeshift<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%14 = fir.dummy_scope : !fir.dscope
|
|
%15 = fir.pack_array %13 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%16 = fir.declare %15 dummy_scope %14 {uniq_name = "_QFtest_allocatable_var_chainFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%17 = fir.rebox %16 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%18 = fir.dummy_scope : !fir.dscope
|
|
%19 = fir.pack_array %17 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%20 = fir.declare %19 dummy_scope %18 {uniq_name = "_QMinnerFinner_repack1Ex"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %19 to %17 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %15 to %13 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine test_allocatable_full_chain(x)
|
|
// real, allocatable :: x(:,:)
|
|
// call repack(x(:,:))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// use inner
|
|
// real :: x(:,:)
|
|
// call inner_repack2(x)
|
|
// end subroutine repack
|
|
// end subroutine test_allocatable_full_chain
|
|
//
|
|
// CHECK-LABEL: func.func @_QPtest_allocatable_full_chain(
|
|
// CHECK-NOT: fir.{{.*}}pack_array
|
|
func.func @_QPtest_allocatable_full_chain(%arg0: !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>> {fir.bindc_name = "x"}) {
|
|
%c1 = arith.constant 1 : index
|
|
%c0 = arith.constant 0 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<allocatable>, uniq_name = "_QFtest_allocatable_full_chainEx"} : (!fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>
|
|
%2 = fir.load %1 : !fir.ref<!fir.box<!fir.heap<!fir.array<?x?xf32>>>>
|
|
%3:3 = fir.box_dims %2, %c0 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>, index) -> (index, index, index)
|
|
%4:3 = fir.box_dims %2, %c1 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>, index) -> (index, index, index)
|
|
%5 = arith.addi %3#0, %3#1 : index
|
|
%6 = arith.subi %5, %c1 : index
|
|
%7 = arith.addi %4#0, %4#1 : index
|
|
%8 = arith.subi %7, %c1 : index
|
|
%9 = fir.box_addr %2 : (!fir.box<!fir.heap<!fir.array<?x?xf32>>>) -> !fir.heap<!fir.array<?x?xf32>>
|
|
%10 = fir.shape_shift %3#0, %3#1, %4#0, %4#1 : (index, index, index, index) -> !fir.shapeshift<2>
|
|
%11 = fir.slice %3#0, %6, %c1, %4#0, %8, %c1 : (index, index, index, index, index, index) -> !fir.slice<2>
|
|
%12 = fir.embox %9(%10) [%11] : (!fir.heap<!fir.array<?x?xf32>>, !fir.shapeshift<2>, !fir.slice<2>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%13 = fir.dummy_scope : !fir.dscope
|
|
%14 = fir.pack_array %12 heap innermost : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%15 = fir.declare %14 dummy_scope %13 {uniq_name = "_QFtest_allocatable_full_chainFrepackEx"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xf32>>
|
|
%16 = fir.rebox %15 : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%17 = fir.dummy_scope : !fir.dscope
|
|
%18 = fir.pack_array %16 heap innermost : (!fir.box<!fir.array<?x?xf32>>) -> !fir.box<!fir.array<?x?xf32>>
|
|
%19 = fir.declare %18 dummy_scope %17 {uniq_name = "_QMinnerFinner_repack2Ex"} : (!fir.box<!fir.array<?x?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xf32>>
|
|
fir.unpack_array %18 to %16 heap : !fir.box<!fir.array<?x?xf32>>
|
|
fir.unpack_array %14 to %12 heap : !fir.box<!fir.array<?x?xf32>>
|
|
return
|
|
}
|
|
|
|
// TODO: if both fir.pack_array have the same property,
|
|
// then the second one is redundant, because the first
|
|
// repack makes 'x' contiguous.
|
|
// subroutine neg_test_assumed_shape(x)
|
|
// real :: x(:)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine neg_test_assumed_shape
|
|
//
|
|
// CHECK-LABEL: func.func @_QPneg_test_assumed_shape(
|
|
// CHECK: fir.pack_array
|
|
// CHECK: fir.pack_array
|
|
// CHECK: fir.unpack_array
|
|
// CHECK: fir.unpack_array
|
|
func.func @_QPneg_test_assumed_shape(%arg0: !fir.box<!fir.array<?xf32>> {fir.bindc_name = "x"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.pack_array %arg0 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%2 = fir.declare %1 dummy_scope %0 {uniq_name = "_QFneg_test_assumed_shapeEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
%3 = fir.rebox %2 : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%4 = fir.slice %c1, %c50, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%5 = fir.rebox %3 [%4] : (!fir.box<!fir.array<?xf32>>, !fir.slice<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%6 = fir.convert %5 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%7 = fir.dummy_scope : !fir.dscope
|
|
%8 = fir.pack_array %6 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%9 = fir.declare %8 dummy_scope %7 {uniq_name = "_QFneg_test_assumed_shapeFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %8 to %6 heap : !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %1 to %arg0 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine neg_test_non_contig_slice_cst(x)
|
|
// real :: x(100)
|
|
// call repack(x(1:50:2))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine neg_test_non_contig_slice_cst
|
|
//
|
|
// CHECK-LABEL: func.func @_QPneg_test_non_contig_slice_cst(
|
|
// CHECK: fir.pack_array
|
|
// CHECK: fir.unpack_array
|
|
func.func @_QPneg_test_non_contig_slice_cst(%arg0: !fir.ref<!fir.array<100xf32>> {fir.bindc_name = "x"}) {
|
|
%c2 = arith.constant 2 : index
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c100 = arith.constant 100 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.shape %c100 : (index) -> !fir.shape<1>
|
|
%2 = fir.declare %arg0(%1) dummy_scope %0 {uniq_name = "_QFneg_test_non_contig_slice_cstEx"} : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<100xf32>>
|
|
%3 = fir.slice %c1, %c50, %c2 : (index, index, index) -> !fir.slice<1>
|
|
%4 = fir.embox %2(%1) [%3] : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<25xf32>>
|
|
%5 = fir.convert %4 : (!fir.box<!fir.array<25xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%6 = fir.dummy_scope : !fir.dscope
|
|
%7 = fir.pack_array %5 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.declare %7 dummy_scope %6 {uniq_name = "_QFneg_test_non_contig_slice_cstFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %7 to %5 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine neg_test_non_contig_slice_var(x, s)
|
|
// integer :: s
|
|
// real :: x(100)
|
|
// call repack(x(1:50:s))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine neg_test_non_contig_slice_var
|
|
//
|
|
// CHECK-LABEL: func.func @_QPneg_test_non_contig_slice_var(
|
|
// CHECK: fir.pack_array
|
|
// CHECK: fir.unpack_array
|
|
func.func @_QPneg_test_non_contig_slice_var(%arg0: !fir.ref<!fir.array<100xf32>> {fir.bindc_name = "x"}, %arg1: !fir.ref<i32> {fir.bindc_name = "s"}) {
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%c100 = arith.constant 100 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg1 dummy_scope %0 {uniq_name = "_QFneg_test_non_contig_slice_varEs"} : (!fir.ref<i32>, !fir.dscope) -> !fir.ref<i32>
|
|
%2 = fir.shape %c100 : (index) -> !fir.shape<1>
|
|
%3 = fir.declare %arg0(%2) dummy_scope %0 {uniq_name = "_QFneg_test_non_contig_slice_varEx"} : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.dscope) -> !fir.ref<!fir.array<100xf32>>
|
|
%4 = fir.load %1 : !fir.ref<i32>
|
|
%5 = fir.convert %4 : (i32) -> index
|
|
%6 = fir.slice %c1, %c50, %5 : (index, index, index) -> !fir.slice<1>
|
|
%7 = fir.embox %3(%2) [%6] : (!fir.ref<!fir.array<100xf32>>, !fir.shape<1>, !fir.slice<1>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFneg_test_non_contig_slice_varFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// subroutine neg_test_pointer(x)
|
|
// real, pointer :: x(:)
|
|
// call repack(x(1:50))
|
|
// contains
|
|
// subroutine repack(x)
|
|
// real :: x(:)
|
|
// end subroutine repack
|
|
// end subroutine neg_test_pointer
|
|
//
|
|
// CHECK-LABEL: func.func @_QPneg_test_pointer(
|
|
// CHECK: fir.pack_array
|
|
// CHECK: fir.unpack_array
|
|
func.func @_QPneg_test_pointer(%arg0: !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>> {fir.bindc_name = "x"}) {
|
|
%c0 = arith.constant 0 : index
|
|
%c50 = arith.constant 50 : index
|
|
%c1 = arith.constant 1 : index
|
|
%0 = fir.dummy_scope : !fir.dscope
|
|
%1 = fir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs<pointer>, uniq_name = "_QFneg_test_pointerEx"} : (!fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>, !fir.dscope) -> !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
|
|
%2 = fir.load %1 : !fir.ref<!fir.box<!fir.ptr<!fir.array<?xf32>>>>
|
|
%3:3 = fir.box_dims %2, %c0 : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, index) -> (index, index, index)
|
|
%4 = fir.shift %3#0 : (index) -> !fir.shift<1>
|
|
%5 = fir.slice %c1, %c50, %c1 : (index, index, index) -> !fir.slice<1>
|
|
%6 = fir.rebox %2(%4) [%5] : (!fir.box<!fir.ptr<!fir.array<?xf32>>>, !fir.shift<1>, !fir.slice<1>) -> !fir.box<!fir.array<50xf32>>
|
|
%7 = fir.convert %6 : (!fir.box<!fir.array<50xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%8 = fir.dummy_scope : !fir.dscope
|
|
%9 = fir.pack_array %7 heap whole : (!fir.box<!fir.array<?xf32>>) -> !fir.box<!fir.array<?xf32>>
|
|
%10 = fir.declare %9 dummy_scope %8 {uniq_name = "_QFneg_test_pointerFrepackEx"} : (!fir.box<!fir.array<?xf32>>, !fir.dscope) -> !fir.box<!fir.array<?xf32>>
|
|
fir.unpack_array %9 to %7 heap : !fir.box<!fir.array<?xf32>>
|
|
return
|
|
}
|
|
|
|
// Test a long chain of fir.pack_array operations.
|
|
// The rewriter used to use a down-top traversal that optimized
|
|
// fir.pack_array operations starting from the innermost one.
|
|
// The rewriter did not converge in 10 (default) iterations
|
|
// causing the pass to report a failure.
|
|
// A top-down traversal should fix this an allow optimizing
|
|
// all the repackings.
|
|
// CHECK-LABEL: func.func @test_long_chain(
|
|
// CHECK-NOT: fir.pack_array
|
|
// CHECK-NOT: fir.unpack_array
|
|
func.func @test_long_chain(%pred: i1) {
|
|
%c10 = arith.constant 10 : index
|
|
%3 = fir.dummy_scope : !fir.dscope
|
|
%4 = fir.address_of(@aaa) : !fir.ref<!fir.array<10x10xi32>>
|
|
%5 = fir.shape %c10, %c10 : (index, index) -> !fir.shape<2>
|
|
%6 = fir.declare %4(%5) {uniq_name = "aaa"} : (!fir.ref<!fir.array<10x10xi32>>, !fir.shape<2>) -> !fir.ref<!fir.array<10x10xi32>>
|
|
%9 = fir.embox %6(%5) : (!fir.ref<!fir.array<10x10xi32>>, !fir.shape<2>) -> !fir.box<!fir.array<10x10xi32>>
|
|
%10 = fir.convert %9 : (!fir.box<!fir.array<10x10xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%11 = fir.dummy_scope : !fir.dscope
|
|
%12 = fir.pack_array %10 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%13 = fir.declare %12 dummy_scope %11 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%14 = fir.rebox %13 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb17, ^bb1
|
|
^bb1: // pred: ^bb0
|
|
%20 = fir.dummy_scope : !fir.dscope
|
|
%21 = fir.pack_array %14 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%22 = fir.declare %21 dummy_scope %20 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%23 = fir.rebox %22 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%28 = fir.dummy_scope : !fir.dscope
|
|
%29 = fir.pack_array %23 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%30 = fir.declare %29 dummy_scope %28 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%31 = fir.rebox %30 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb16, ^bb2
|
|
^bb2: // pred: ^bb1
|
|
%37 = fir.dummy_scope : !fir.dscope
|
|
%38 = fir.pack_array %31 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%39 = fir.declare %38 dummy_scope %37 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%40 = fir.rebox %39 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%45 = fir.dummy_scope : !fir.dscope
|
|
%46 = fir.pack_array %40 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%47 = fir.declare %46 dummy_scope %45 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%48 = fir.rebox %47 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb15, ^bb3
|
|
^bb3: // pred: ^bb2
|
|
%54 = fir.dummy_scope : !fir.dscope
|
|
%55 = fir.pack_array %48 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%56 = fir.declare %55 dummy_scope %54 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%57 = fir.rebox %56 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%62 = fir.dummy_scope : !fir.dscope
|
|
%63 = fir.pack_array %57 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%64 = fir.declare %63 dummy_scope %62 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%65 = fir.rebox %64 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb14, ^bb4
|
|
^bb4: // pred: ^bb3
|
|
%71 = fir.dummy_scope : !fir.dscope
|
|
%72 = fir.pack_array %65 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%73 = fir.declare %72 dummy_scope %71 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%74 = fir.rebox %73 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%79 = fir.dummy_scope : !fir.dscope
|
|
%80 = fir.pack_array %74 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%81 = fir.declare %80 dummy_scope %79 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%82 = fir.rebox %81 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb13, ^bb5
|
|
^bb5: // pred: ^bb4
|
|
%88 = fir.dummy_scope : !fir.dscope
|
|
%89 = fir.pack_array %82 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%90 = fir.declare %89 dummy_scope %88 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%91 = fir.rebox %90 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%96 = fir.dummy_scope : !fir.dscope
|
|
%97 = fir.pack_array %91 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%98 = fir.declare %97 dummy_scope %96 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%99 = fir.rebox %98 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb12, ^bb6
|
|
^bb6: // pred: ^bb5
|
|
%105 = fir.dummy_scope : !fir.dscope
|
|
%106 = fir.pack_array %99 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%107 = fir.declare %106 dummy_scope %105 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%108 = fir.rebox %107 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%113 = fir.dummy_scope : !fir.dscope
|
|
%114 = fir.pack_array %108 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%115 = fir.declare %114 dummy_scope %113 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%116 = fir.rebox %115 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb11, ^bb7
|
|
^bb7: // pred: ^bb6
|
|
%122 = fir.dummy_scope : !fir.dscope
|
|
%123 = fir.pack_array %116 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%124 = fir.declare %123 dummy_scope %122 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%125 = fir.rebox %124 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%130 = fir.dummy_scope : !fir.dscope
|
|
%131 = fir.pack_array %125 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
%132 = fir.declare %131 dummy_scope %130 {uniq_name = "aaa"} : (!fir.box<!fir.array<?x?xi32>>, !fir.dscope) -> !fir.box<!fir.array<?x?xi32>>
|
|
%133 = fir.rebox %132 : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
cf.cond_br %pred, ^bb9, ^bb8
|
|
^bb8: // pred: ^bb7
|
|
%139 = fir.dummy_scope : !fir.dscope
|
|
%140 = fir.pack_array %133 heap innermost : (!fir.box<!fir.array<?x?xi32>>) -> !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %140 to %133 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb9
|
|
^bb9: // 2 preds: ^bb7, ^bb8
|
|
fir.unpack_array %131 to %125 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb10
|
|
^bb10: // pred: ^bb9
|
|
fir.unpack_array %123 to %116 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb11
|
|
^bb11: // 2 preds: ^bb6, ^bb10
|
|
fir.unpack_array %114 to %108 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %106 to %99 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb12
|
|
^bb12: // 2 preds: ^bb5, ^bb11
|
|
fir.unpack_array %97 to %91 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %89 to %82 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb13
|
|
^bb13: // 2 preds: ^bb4, ^bb12
|
|
fir.unpack_array %80 to %74 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %72 to %65 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb14
|
|
^bb14: // 2 preds: ^bb3, ^bb13
|
|
fir.unpack_array %63 to %57 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %55 to %48 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb15
|
|
^bb15: // 2 preds: ^bb2, ^bb14
|
|
fir.unpack_array %46 to %40 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %38 to %31 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb16
|
|
^bb16: // 2 preds: ^bb1, ^bb15
|
|
fir.unpack_array %29 to %23 heap : !fir.box<!fir.array<?x?xi32>>
|
|
fir.unpack_array %21 to %14 heap : !fir.box<!fir.array<?x?xi32>>
|
|
cf.br ^bb17
|
|
^bb17: // 2 preds: ^bb0, ^bb16
|
|
fir.unpack_array %12 to %10 heap : !fir.box<!fir.array<?x?xi32>>
|
|
return
|
|
}
|