Files
llvm/flang/test/HLFIR/apply-codegen.fir
Jean Perier 61a124db56 [flang] hlfir.apply codegen
Lower hlfir.apply to hlfir.designate in bufferization.

Differential Revision: https://reviews.llvm.org/D140101
2022-12-16 09:27:58 +01:00

28 lines
1.7 KiB
Plaintext

// Test hlfir.apply code generation
// RUN: fir-opt %s -bufferize-hlfir | FileCheck %s
func.func @numeric_apply(%arg0 : !fir.ref<!fir.array<100xi32>>) {
%expr = hlfir.as_expr %arg0 : (!fir.ref<!fir.array<100xi32>>) -> !hlfir.expr<100xi32>
%c42 = arith.constant 42 : index
%elem = hlfir.apply %expr, %c42 : (!hlfir.expr<100xi32>, index) -> i32
return
}
// CHECK-LABEL: func.func @numeric_apply(
// CHECK: %[[VAL_5:.*]]:2 = hlfir.declare %{{.*}}(%{{.*}}) {uniq_name = ".tmp"}
// CHECK: %[[VAL_9:.*]] = arith.constant 42 : index
// CHECK: %[[VAL_10:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_9]]) : (!fir.heap<!fir.array<100xi32>>, index) -> !fir.ref<i32>
// CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_10]] : !fir.ref<i32>
func.func @character_apply(%arg0 : !fir.ref<!fir.array<?x!fir.char<2,?>>>, %l: index, %n: index) {
%shape = fir.shape %n : (index) -> !fir.shape<1>
%c:2 = hlfir.declare %arg0(%shape) typeparams %l {uniq_name = "c"} : (!fir.ref<!fir.array<?x!fir.char<2,?>>>, !fir.shape<1>, index) -> (!fir.box<!fir.array<?x!fir.char<2,?>>>, !fir.ref<!fir.array<?x!fir.char<2,?>>>)
%expr = hlfir.as_expr %c#0 : (!fir.box<!fir.array<?x!fir.char<2,?>>>) -> !hlfir.expr<?x!fir.char<2,?>>
%c42 = arith.constant 42 : index
%elem = hlfir.apply %expr, %c42 typeparams %l: (!hlfir.expr<?x!fir.char<2,?>>, index, index) -> !hlfir.expr<!fir.char<2,?>>
return
}
// CHECK-LABEL: func.func @character_apply(
// CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %{{.*}}(%{{.*}}) typeparams %[[VAL_1:.*]] {uniq_name = ".tmp"}
// CHECK: %[[VAL_11:.*]] = arith.constant 42 : index
// CHECK: %[[VAL_12:.*]] = hlfir.designate %[[VAL_7]]#0 (%[[VAL_11]]) typeparams %[[VAL_1]] : (!fir.box<!fir.array<?x!fir.char<2,?>>>, index, index) -> !fir.boxchar<2>