diff --git a/flang/lib/Optimizer/HLFIR/Transforms/LowerHLFIRIntrinsics.cpp b/flang/lib/Optimizer/HLFIR/Transforms/LowerHLFIRIntrinsics.cpp index 3c29d6877e8de..e0167cc12b8a3 100644 --- a/flang/lib/Optimizer/HLFIR/Transforms/LowerHLFIRIntrinsics.cpp +++ b/flang/lib/Optimizer/HLFIR/Transforms/LowerHLFIRIntrinsics.cpp @@ -469,33 +469,49 @@ struct MatmulTransposeOpConversion } }; -class CShiftOpConversion : public HlfirIntrinsicConversion { - using HlfirIntrinsicConversion::HlfirIntrinsicConversion; +// A converter for hlfir.cshift and hlfir.eoshift. +template +class ArrayShiftOpConversion : public HlfirIntrinsicConversion { + using HlfirIntrinsicConversion::HlfirIntrinsicConversion; + using HlfirIntrinsicConversion::lowerArguments; + using HlfirIntrinsicConversion::processReturnValue; + using typename HlfirIntrinsicConversion::IntrinsicArgument; llvm::LogicalResult - matchAndRewrite(hlfir::CShiftOp cshift, - mlir::PatternRewriter &rewriter) const override { - fir::FirOpBuilder builder{rewriter, cshift.getOperation()}; - const mlir::Location &loc = cshift->getLoc(); + matchAndRewrite(T op, mlir::PatternRewriter &rewriter) const override { + fir::FirOpBuilder builder{rewriter, op.getOperation()}; + const mlir::Location &loc = op->getLoc(); - llvm::SmallVector inArgs; - mlir::Value array = cshift.getArray(); + llvm::SmallVector inArgs; + llvm::StringRef intrinsicName{[]() { + if constexpr (std::is_same_v) + return "eoshift"; + else if constexpr (std::is_same_v) + return "cshift"; + else + llvm_unreachable("unsupported array shift"); + }()}; + + mlir::Value array = op.getArray(); inArgs.push_back({array, array.getType()}); - mlir::Value shift = cshift.getShift(); + mlir::Value shift = op.getShift(); inArgs.push_back({shift, shift.getType()}); - inArgs.push_back({cshift.getDim(), builder.getI32Type()}); + if constexpr (std::is_same_v) { + mlir::Value boundary = op.getBoundary(); + inArgs.push_back({boundary, boundary ? boundary.getType() : nullptr}); + } + inArgs.push_back({op.getDim(), builder.getI32Type()}); - auto *argLowering = fir::getIntrinsicArgumentLowering("cshift"); + auto *argLowering = fir::getIntrinsicArgumentLowering(intrinsicName); llvm::SmallVector args = - lowerArguments(cshift, inArgs, rewriter, argLowering); + lowerArguments(op, inArgs, rewriter, argLowering); - mlir::Type scalarResultType = - hlfir::getFortranElementType(cshift.getType()); + mlir::Type scalarResultType = hlfir::getFortranElementType(op.getType()); - auto [resultExv, mustBeFreed] = - fir::genIntrinsicCall(builder, loc, "cshift", scalarResultType, args); + auto [resultExv, mustBeFreed] = fir::genIntrinsicCall( + builder, loc, intrinsicName, scalarResultType, args); - processReturnValue(cshift, resultExv, mustBeFreed, builder, rewriter); + processReturnValue(op, resultExv, mustBeFreed, builder, rewriter); return mlir::success(); } }; @@ -547,7 +563,8 @@ class LowerHLFIRIntrinsics AnyOpConversion, SumOpConversion, ProductOpConversion, TransposeOpConversion, CountOpConversion, DotProductOpConversion, MaxvalOpConversion, MinvalOpConversion, MinlocOpConversion, - MaxlocOpConversion, CShiftOpConversion, ReshapeOpConversion>(context); + MaxlocOpConversion, ArrayShiftOpConversion, + ArrayShiftOpConversion, ReshapeOpConversion>(context); // While conceptually this pass is performing dialect conversion, we use // pattern rewrites here instead of dialect conversion because this pass diff --git a/flang/test/HLFIR/eoshift-lowering.fir b/flang/test/HLFIR/eoshift-lowering.fir new file mode 100644 index 0000000000000..7bfc3e21f0527 --- /dev/null +++ b/flang/test/HLFIR/eoshift-lowering.fir @@ -0,0 +1,294 @@ +// Test hlfir.eoshift operation lowering to fir runtime call +// RUN: fir-opt %s -lower-hlfir-intrinsics | FileCheck %s + +// 1d boxed vector shift by scalar +func.func @eoshift1(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !fir.ref {fir.bindc_name = "sh"}) { + %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) + %1:2 = hlfir.declare %arg1 {uniq_name = "sh"} : (!fir.ref) -> (!fir.ref, !fir.ref) + %2 = hlfir.eoshift %0#0 %1#0 : (!fir.box>, !fir.ref) -> !hlfir.expr + hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> + return +} +// CHECK-LABEL: func.func @eoshift1( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: !fir.ref {fir.bindc_name = "sh"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_5:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_6:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) +// CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "sh"} : (!fir.ref) -> (!fir.ref, !fir.ref) +// CHECK: %[[VAL_8:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_9:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1> +// CHECK: %[[VAL_10:.*]] = fir.embox %[[VAL_8]](%[[VAL_9]]) : (!fir.heap>, !fir.shape<1>) -> !fir.box>> +// CHECK: fir.store %[[VAL_10]] to %[[VAL_5]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_7]]#0 : !fir.ref +// CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_5]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_6]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_11]] : (i32) -> i64 +// CHECK: fir.call @_FortranAEoshiftVector(%[[VAL_13]], %[[VAL_14]], %[[VAL_15]], %[[BOUNDARY]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, i64, !fir.box, !fir.ref, i32) -> () + +// 2d boxed array shift by scalar +func.func @eoshift2(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: i32 {fir.bindc_name = "sh"}) { + %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) + %2 = hlfir.eoshift %0#0 %arg1 : (!fir.box>, i32) -> !hlfir.expr + hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> + return +} +// CHECK-LABEL: func.func @eoshift2( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: i32 {fir.bindc_name = "sh"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_5:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_7:.*]] = fir.alloca i32 +// CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) +// CHECK: fir.store %[[VAL_1]] to %[[VAL_7]] : !fir.ref +// CHECK: %[[VAL_9:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_9]](%[[VAL_10]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_11]] to %[[VAL_6]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_12:.*]] = fir.embox %[[VAL_7]] : (!fir.ref) -> !fir.box +// CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_8]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_12]] : (!fir.box) -> !fir.box +// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 +// CHECK: fir.call @_FortranAEoshift(%[[VAL_14]], %[[VAL_15]], %[[VAL_16]], %[[BOUNDARY]], %[[VAL_17]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// 2d boxed array shift by boxed array +func.func @eoshift3(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !fir.box> {fir.bindc_name = "sh"}) { + %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) + %1:2 = hlfir.declare %arg1 {uniq_name = "sh"} : (!fir.box>) -> (!fir.box>, !fir.box>) + %2 = hlfir.eoshift %0#0 %1#0 : (!fir.box>, !fir.box>) -> !hlfir.expr + hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> + return +} +// CHECK-LABEL: func.func @eoshift3( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: !fir.box> {fir.bindc_name = "sh"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_5:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) +// CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_1]] {uniq_name = "sh"} : (!fir.box>) -> (!fir.box>, !fir.box>) +// CHECK: %[[VAL_9:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_10:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_9]](%[[VAL_10]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_11]] to %[[VAL_6]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_13:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_14:.*]] = fir.convert %[[VAL_7]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_8]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 +// CHECK: fir.call @_FortranAEoshift(%[[VAL_13]], %[[VAL_14]], %[[VAL_15]], %[[BOUNDARY]], %[[VAL_16]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// 2d boxed array shift by array expr +func.func @eoshift4(%arg0: !fir.box> {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}) { + %0:2 = hlfir.declare %arg0 {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) + %2 = hlfir.eoshift %0#0 %arg1 : (!fir.box>, !hlfir.expr) -> !hlfir.expr + hlfir.assign %2 to %0#0 : !hlfir.expr, !fir.box> + return +} +// CHECK-LABEL: func.func @eoshift4( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.box> {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_5:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_7:.*]]:2 = hlfir.declare %[[VAL_0]] {uniq_name = "a"} : (!fir.box>) -> (!fir.box>, !fir.box>) +// CHECK: %[[VAL_8:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> +// CHECK: %[[VAL_9:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_8]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) +// CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 0 : index} : (!fir.shape<1>) -> index +// CHECK: %[[VAL_11:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_12:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_13:.*]] = fir.embox %[[VAL_11]](%[[VAL_12]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_13]] to %[[VAL_6]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_10]] : (index) -> !fir.shape<1> +// CHECK: %[[VAL_15:.*]] = fir.embox %[[VAL_9]]#1(%[[VAL_14]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> +// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_7]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]] : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_20:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 +// CHECK: fir.call @_FortranAEoshift(%[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %[[BOUNDARY]], %[[VAL_20]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// 2d array expr shift by array expr +func.func @eoshift5(%arg0: !hlfir.expr {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}) { + %2 = hlfir.eoshift %arg0 %arg1 : (!hlfir.expr, !hlfir.expr) -> !hlfir.expr + hlfir.destroy %2 : !hlfir.expr + return +} +// CHECK-LABEL: func.func @eoshift5( +// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_5:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_7:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr) -> !fir.shape<2> +// CHECK: %[[VAL_8:.*]]:3 = hlfir.associate %[[VAL_0]](%[[VAL_7]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<2>) -> (!fir.box>, !fir.ref>, i1) +// CHECK: %[[VAL_9:.*]] = hlfir.get_extent %[[VAL_7]] {dim = 0 : index} : (!fir.shape<2>) -> index +// CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_7]] {dim = 1 : index} : (!fir.shape<2>) -> index +// CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_9]], %[[VAL_10]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_15:.*]] = fir.embox %[[VAL_8]]#1(%[[VAL_14]]) : (!fir.ref>, !fir.shape<2>) -> !fir.box> +// CHECK: %[[VAL_11:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> +// CHECK: %[[VAL_12:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_11]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) +// CHECK: %[[VAL_13:.*]] = hlfir.get_extent %[[VAL_11]] {dim = 0 : index} : (!fir.shape<1>) -> index +// CHECK: %[[VAL_16:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_17:.*]] = fir.shape %[[VAL_5]], %[[VAL_5]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_18:.*]] = fir.embox %[[VAL_16]](%[[VAL_17]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_18]] to %[[VAL_6]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_19:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1> +// CHECK: %[[VAL_20:.*]] = fir.embox %[[VAL_12]]#1(%[[VAL_19]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> +// CHECK: %[[VAL_22:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_23:.*]] = fir.convert %[[VAL_15]] : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_20]] : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_4]] : (index) -> i32 +// CHECK: fir.call @_FortranAEoshift(%[[VAL_22]], %[[VAL_23]], %[[VAL_24]], %[[BOUNDARY]], %[[VAL_25]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// 2d array expr shift by array expr with explicit dim +func.func @eoshift6(%arg0: !hlfir.expr {fir.bindc_name = "a"}, %arg1: !hlfir.expr {fir.bindc_name = "sh"}, %dim : i16) { + %2 = hlfir.eoshift %arg0 %arg1 dim %dim : (!hlfir.expr, !hlfir.expr, i16) -> !hlfir.expr + hlfir.destroy %2 : !hlfir.expr + return +} +// CHECK-LABEL: func.func @eoshift6( +// CHECK-SAME: %[[VAL_0:.*]]: !hlfir.expr {fir.bindc_name = "a"}, +// CHECK-SAME: %[[VAL_1:.*]]: !hlfir.expr {fir.bindc_name = "sh"}, +// CHECK-SAME: %[[VAL_2:.*]]: i16) { +// CHECK: %[[VAL_3:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_6:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_7:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_8:.*]] = hlfir.shape_of %[[VAL_0]] : (!hlfir.expr) -> !fir.shape<2> +// CHECK: %[[VAL_9:.*]]:3 = hlfir.associate %[[VAL_0]](%[[VAL_8]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<2>) -> (!fir.box>, !fir.ref>, i1) +// CHECK: %[[VAL_10:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 0 : index} : (!fir.shape<2>) -> index +// CHECK: %[[VAL_11:.*]] = hlfir.get_extent %[[VAL_8]] {dim = 1 : index} : (!fir.shape<2>) -> index +// CHECK: %[[VAL_16:.*]] = fir.shape %[[VAL_10]], %[[VAL_11]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_17:.*]] = fir.embox %[[VAL_9]]#1(%[[VAL_16]]) : (!fir.ref>, !fir.shape<2>) -> !fir.box> +// CHECK: %[[VAL_12:.*]] = hlfir.shape_of %[[VAL_1]] : (!hlfir.expr) -> !fir.shape<1> +// CHECK: %[[VAL_13:.*]]:3 = hlfir.associate %[[VAL_1]](%[[VAL_12]]) {adapt.valuebyref} : (!hlfir.expr, !fir.shape<1>) -> (!fir.box>, !fir.ref>, i1) +// CHECK: %[[VAL_14:.*]] = hlfir.get_extent %[[VAL_12]] {dim = 0 : index} : (!fir.shape<1>) -> index +// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_2]] : (i16) -> i32 +// CHECK: %[[VAL_18:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_19:.*]] = fir.shape %[[VAL_6]], %[[VAL_6]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_20:.*]] = fir.embox %[[VAL_18]](%[[VAL_19]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_20]] to %[[VAL_7]] : !fir.ref>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_21:.*]] = fir.shape %[[VAL_14]] : (index) -> !fir.shape<1> +// CHECK: %[[VAL_22:.*]] = fir.embox %[[VAL_13]]#1(%[[VAL_21]]) : (!fir.ref>, !fir.shape<1>) -> !fir.box> +// CHECK: %[[VAL_24:.*]] = fir.convert %[[VAL_7]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_25:.*]] = fir.convert %[[VAL_17]] : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_26:.*]] = fir.convert %[[VAL_22]] : (!fir.box>) -> !fir.box +// CHECK: fir.call @_FortranAEoshift(%[[VAL_24]], %[[VAL_25]], %[[VAL_26]], %[[BOUNDARY]], %[[VAL_15]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// shift of polymorphic array +func.func @eoshift7(%arg0: !fir.ref>>>>, %arg1: !fir.ref) { + %0 = fir.dummy_scope : !fir.dscope + %1:2 = hlfir.declare %arg0 dummy_scope %0 {fortran_attrs = #fir.var_attrs, uniq_name = "a"} : (!fir.ref>>>>, !fir.dscope) -> (!fir.ref>>>>, !fir.ref>>>>) + %2:2 = hlfir.declare %arg1 dummy_scope %0 {uniq_name = "sh"} : (!fir.ref, !fir.dscope) -> (!fir.ref, !fir.ref) + %c2_i32 = arith.constant 2 : i32 + %3 = fir.load %1#0 : !fir.ref>>>> + %4 = hlfir.eoshift %3 %c2_i32 : (!fir.class>>>, i32) -> !hlfir.expr?> + hlfir.assign %4 to %1#0 realloc : !hlfir.expr?>, !fir.ref>>>> + hlfir.destroy %4 : !hlfir.expr?> + return +} +// CHECK-LABEL: func.func @eoshift7( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.ref>>>>, +// CHECK-SAME: %[[VAL_1:.*]]: !fir.ref) { +// CHECK: %[[VAL_2:.*]] = arith.constant true +// CHECK: %[[VAL_4:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_5:.*]] = arith.constant 2 : i32 +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.class>>> +// CHECK: %[[VAL_7:.*]] = fir.alloca i32 +// CHECK: %[[VAL_8:.*]] = fir.dummy_scope : !fir.dscope +// CHECK: %[[VAL_9:.*]]:2 = hlfir.declare %[[VAL_0]] dummy_scope %[[VAL_8]] {fortran_attrs = #fir.var_attrs, uniq_name = "a"} : (!fir.ref>>>>, !fir.dscope) -> (!fir.ref>>>>, !fir.ref>>>>) +// CHECK: %[[VAL_10:.*]]:2 = hlfir.declare %[[VAL_1]] dummy_scope %[[VAL_8]] {uniq_name = "sh"} : (!fir.ref, !fir.dscope) -> (!fir.ref, !fir.ref) +// CHECK: %[[VAL_11:.*]] = fir.load %[[VAL_9]]#0 : !fir.ref>>>> +// CHECK: fir.store %[[VAL_5]] to %[[VAL_7]] : !fir.ref +// CHECK: %[[VAL_12:.*]] = fir.zero_bits !fir.heap>> +// CHECK: %[[VAL_13:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1> +// CHECK: %[[VAL_14:.*]] = fir.embox %[[VAL_12]](%[[VAL_13]]) source_box %[[VAL_11]] : (!fir.heap>>, !fir.shape<1>, !fir.class>>>) -> !fir.class>>> +// CHECK: fir.store %[[VAL_14]] to %[[VAL_6]] : !fir.ref>>>> +// CHECK: %[[BOUNDARY:.*]] = fir.absent !fir.box +// CHECK: %[[VAL_15:.*]] = fir.load %[[VAL_7]] : !fir.ref +// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>>) -> !fir.ref> +// CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_11]] : (!fir.class>>>) -> !fir.box +// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]] : (i32) -> i64 +// CHECK: fir.call @_FortranAEoshiftVector(%[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %[[BOUNDARY]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, i64, !fir.box, !fir.ref, i32) -> () + +// shift with the present scalar boundary and dim +func.func @_QPeoshift8(%arg0: !fir.box> {fir.bindc_name = "array"}) { + %cst = arith.constant 3.000000e+00 : f32 + %c2_i32 = arith.constant 2 : i32 + %0 = fir.dummy_scope : !fir.dscope + %1:2 = hlfir.declare %arg0 dummy_scope %0 {uniq_name = "_QFeoshift8Earray"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) + %2 = hlfir.eoshift %1#0 %c2_i32 boundary %cst dim %c2_i32 : (!fir.box>, i32, f32, i32) -> !hlfir.expr + hlfir.assign %2 to %1#0 : !hlfir.expr, !fir.box> + hlfir.destroy %2 : !hlfir.expr + return +} +// CHECK-LABEL: func.func @_QPeoshift8( +// CHECK-SAME: %[[ARG0:.*]]: !fir.box> {fir.bindc_name = "array"}) { +// CHECK: %[[VAL_3:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_4:.*]] = arith.constant 2 : i32 +// CHECK: %[[VAL_5:.*]] = arith.constant 3.000000e+00 : f32 +// CHECK: %[[VAL_6:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_7:.*]] = fir.alloca f32 +// CHECK: %[[VAL_8:.*]] = fir.alloca i32 +// CHECK: %[[VAL_9:.*]] = fir.dummy_scope : !fir.dscope +// CHECK: %[[VAL_10:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[VAL_9]] {uniq_name = "_QFeoshift8Earray"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) +// CHECK: fir.store %[[VAL_4]] to %[[VAL_8]] : !fir.ref +// CHECK: fir.store %[[VAL_5]] to %[[VAL_7]] : !fir.ref +// CHECK: %[[VAL_11:.*]] = fir.embox %[[VAL_7]] : (!fir.ref) -> !fir.box +// CHECK: %[[VAL_12:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_13:.*]] = fir.shape %[[VAL_3]], %[[VAL_3]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_14:.*]] = fir.embox %[[VAL_12]](%[[VAL_13]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_14]] to %[[VAL_6]] : !fir.ref>>> +// CHECK: %[[VAL_15:.*]] = fir.embox %[[VAL_8]] : (!fir.ref) -> !fir.box +// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_6]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_10]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_15]] : (!fir.box) -> !fir.box +// CHECK: %[[VAL_20:.*]] = fir.convert %[[VAL_11]] : (!fir.box) -> !fir.box +// CHECK: fir.call @_FortranAEoshift(%[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %[[VAL_20]], %[[VAL_4]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> () + +// shift with the present array boundary +func.func @_QPeoshift9(%arg0: !fir.box> {fir.bindc_name = "array"}, %arg1: !fir.box> {fir.bindc_name = "boundary"}) { + %c2_i32 = arith.constant 2 : i32 + %0 = fir.dummy_scope : !fir.dscope + %1:2 = hlfir.declare %arg0 dummy_scope %0 {uniq_name = "_QFeoshift9Earray"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) + %2:2 = hlfir.declare %arg1 dummy_scope %0 {uniq_name = "_QFeoshift9Eboundary"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) + %3 = hlfir.eoshift %1#0 %c2_i32 boundary %2#0 : (!fir.box>, i32, !fir.box>) -> !hlfir.expr + hlfir.assign %3 to %1#0 : !hlfir.expr, !fir.box> + hlfir.destroy %3 : !hlfir.expr + return +} +// CHECK-LABEL: func.func @_QPeoshift9( +// CHECK-SAME: %[[ARG0:.*]]: !fir.box> {fir.bindc_name = "array"}, +// CHECK-SAME: %[[ARG1:.*]]: !fir.box> {fir.bindc_name = "boundary"}) { +// CHECK: %[[VAL_2:.*]] = arith.constant 1 : index +// CHECK: %[[VAL_3:.*]] = arith.constant 0 : index +// CHECK: %[[VAL_4:.*]] = arith.constant 2 : i32 +// CHECK: %[[VAL_5:.*]] = fir.alloca !fir.box>> +// CHECK: %[[VAL_6:.*]] = fir.alloca i32 +// CHECK: %[[VAL_7:.*]] = fir.dummy_scope : !fir.dscope +// CHECK: %[[VAL_8:.*]]:2 = hlfir.declare %[[ARG0]] dummy_scope %[[VAL_7]] {uniq_name = "_QFeoshift9Earray"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) +// CHECK: %[[VAL_9:.*]]:2 = hlfir.declare %[[ARG1]] dummy_scope %[[VAL_7]] {uniq_name = "_QFeoshift9Eboundary"} : (!fir.box>, !fir.dscope) -> (!fir.box>, !fir.box>) +// CHECK: fir.store %[[VAL_4]] to %[[VAL_6]] : !fir.ref +// CHECK: %[[VAL_10:.*]] = fir.zero_bits !fir.heap> +// CHECK: %[[VAL_11:.*]] = fir.shape %[[VAL_3]], %[[VAL_3]] : (index, index) -> !fir.shape<2> +// CHECK: %[[VAL_12:.*]] = fir.embox %[[VAL_10]](%[[VAL_11]]) : (!fir.heap>, !fir.shape<2>) -> !fir.box>> +// CHECK: fir.store %[[VAL_12]] to %[[VAL_5]] : !fir.ref>>> +// CHECK: %[[VAL_13:.*]] = fir.embox %[[VAL_6]] : (!fir.ref) -> !fir.box +// CHECK: %[[VAL_15:.*]] = fir.convert %[[VAL_5]] : (!fir.ref>>>) -> !fir.ref> +// CHECK: %[[VAL_16:.*]] = fir.convert %[[VAL_8]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_17:.*]] = fir.convert %[[VAL_13]] : (!fir.box) -> !fir.box +// CHECK: %[[VAL_18:.*]] = fir.convert %[[VAL_9]]#1 : (!fir.box>) -> !fir.box +// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_2]] : (index) -> i32 +// CHECK: fir.call @_FortranAEoshift(%[[VAL_15]], %[[VAL_16]], %[[VAL_17]], %[[VAL_18]], %[[VAL_19]], %{{.*}}, %{{.*}}) : (!fir.ref>, !fir.box, !fir.box, !fir.box, i32, !fir.ref, i32) -> ()