diff --git a/flang/lib/Optimizer/HLFIR/Transforms/BufferizeHLFIR.cpp b/flang/lib/Optimizer/HLFIR/Transforms/BufferizeHLFIR.cpp index bb6d495ae3d3..9b894027a9f2 100644 --- a/flang/lib/Optimizer/HLFIR/Transforms/BufferizeHLFIR.cpp +++ b/flang/lib/Optimizer/HLFIR/Transforms/BufferizeHLFIR.cpp @@ -320,6 +320,27 @@ struct SetLengthOpConversion } }; +struct GetLengthOpConversion + : public mlir::OpConversionPattern { + using mlir::OpConversionPattern::OpConversionPattern; + explicit GetLengthOpConversion(mlir::MLIRContext *ctx) + : mlir::OpConversionPattern{ctx} {} + mlir::LogicalResult + matchAndRewrite(hlfir::GetLengthOp getLength, OpAdaptor adaptor, + mlir::ConversionPatternRewriter &rewriter) const override { + mlir::Location loc = getLength->getLoc(); + auto module = getLength->getParentOfType(); + fir::FirOpBuilder builder(rewriter, fir::getKindMapping(module)); + hlfir::Entity bufferizedExpr = getBufferizedExprStorage(adaptor.getExpr()); + mlir::Value length = hlfir::genCharLength(loc, builder, bufferizedExpr); + if (!length) + return rewriter.notifyMatchFailure( + getLength, "could not deduce length from GetLengthOp operand"); + rewriter.replaceOp(getLength, length); + return mlir::success(); + } +}; + static bool allOtherUsesAreDestroys(mlir::Value value, mlir::Operation *currentUse) { for (mlir::Operation *useOp : value.getUsers()) @@ -662,12 +683,12 @@ public: auto module = this->getOperation(); auto *context = &getContext(); mlir::RewritePatternSet patterns(context); - patterns - .insert(context); + patterns.insert(context); mlir::ConversionTarget target(*context); // Note that YieldElementOp is not marked as an illegal operation. // It must be erased by its parent converter and there is no explicit diff --git a/flang/test/HLFIR/get_length_codegen.fir b/flang/test/HLFIR/get_length_codegen.fir new file mode 100644 index 000000000000..3b3785c82667 --- /dev/null +++ b/flang/test/HLFIR/get_length_codegen.fir @@ -0,0 +1,32 @@ +// Test hlfir.get_length code generation +// RUN: fir-opt %s --bufferize-hlfir | FileCheck %s + +func.func @_QPtest_char_get_length(%arg0: !fir.boxchar<1> {fir.bindc_name = "ch"}) -> index { + %c3 = arith.constant 3 : index + %0:2 = fir.unboxchar %arg0 : (!fir.boxchar<1>) -> (!fir.ref>, index) + %1:2 = hlfir.declare %0#0 typeparams %0#1 {uniq_name = "_QFtest_char_get_lengthEch"} : (!fir.ref>, index) -> (!fir.boxchar<1>, !fir.ref>) + %2 = fir.alloca i32 {bindc_name = "x", uniq_name = "_QFtest_char_get_lengthEx"} + %3:2 = hlfir.declare %2 {uniq_name = "_QFtest_char_get_lengthEx"} : (!fir.ref) -> (!fir.ref, !fir.ref) + %4 = fir.address_of(@_QQcl.616263) : !fir.ref> + %5:2 = hlfir.declare %4 typeparams %c3 {fortran_attrs = #fir.var_attrs, uniq_name = "_QQcl.616263"} : (!fir.ref>, index) -> (!fir.ref>, !fir.ref>) + %6 = arith.addi %0#1, %c3 : index + %7 = hlfir.concat %5#0, %1#0 len %6 : (!fir.ref>, !fir.boxchar<1>, index) -> !hlfir.expr> + %8:3 = hlfir.associate %7 typeparams %6 {uniq_name = ".tmp.assign"} : (!hlfir.expr>, index) -> (!fir.boxchar<1>, !fir.ref>, i1) + %9 = hlfir.as_expr %8#0 : (!fir.boxchar<1>) -> !hlfir.expr> + %10 = hlfir.get_length %9 : (!hlfir.expr>) -> index + hlfir.destroy %9 : !hlfir.expr> + hlfir.end_associate %8#1, %8#2 : !fir.ref>, i1 + return %10 : index +} +fir.global linkonce @_QQcl.616263 constant : !fir.char<1,3> { + %0 = fir.string_lit "abc"(3) : !fir.char<1,3> + fir.has_value %0 : !fir.char<1,3> +} +// CHECK-LABEL: func.func @_QPtest_char_get_length( +// CHECK-SAME: %[[VAL_0:.*]]: !fir.boxchar<1> {fir.bindc_name = "ch"}) -> index { +// CHECK: %[[VAL_1:.*]] = arith.constant 3 : index +// CHECK: %[[VAL_2:.*]]:2 = fir.unboxchar %[[VAL_0]] : (!fir.boxchar<1>) -> (!fir.ref>, index) +// CHECK: %[[VAL_9:.*]] = arith.addi %[[VAL_1]], %[[VAL_2]]#1 : index +// CHECK: %[[VAL_33:.*]]:2 = hlfir.declare %[[VAL_31:.*]] typeparams %[[VAL_9]] {uniq_name = ".tmp"} : (!fir.ref>, index) -> (!fir.boxchar<1>, !fir.ref>) +// CHECK: return %[[VAL_9]] : index +// CHECK: }