diff --git a/flang/lib/Lower/IntrinsicCall.cpp b/flang/lib/Lower/IntrinsicCall.cpp index 104fb7e6792f6..16a250e659200 100644 --- a/flang/lib/Lower/IntrinsicCall.cpp +++ b/flang/lib/Lower/IntrinsicCall.cpp @@ -493,6 +493,7 @@ struct IntrinsicLibrary { fir::ExtendedValue genMatmul(mlir::Type, llvm::ArrayRef); fir::ExtendedValue genMaxloc(mlir::Type, llvm::ArrayRef); fir::ExtendedValue genMaxval(mlir::Type, llvm::ArrayRef); + fir::ExtendedValue genMerge(mlir::Type, llvm::ArrayRef); fir::ExtendedValue genMinloc(mlir::Type, llvm::ArrayRef); fir::ExtendedValue genMinval(mlir::Type, llvm::ArrayRef); mlir::Value genMod(mlir::Type, llvm::ArrayRef); @@ -764,6 +765,7 @@ static constexpr IntrinsicHandler handlers[]{ {"dim", asValue}, {"mask", asBox, handleDynamicOptional}}}, /*isElemental=*/false}, + {"merge", &I::genMerge}, {"min", &I::genExtremum}, {"minloc", &I::genMinloc, @@ -2743,6 +2745,27 @@ IntrinsicLibrary::genMaxval(mlir::Type resultType, stmtCtx, "unexpected result for Maxval", args); } +// MERGE +fir::ExtendedValue +IntrinsicLibrary::genMerge(mlir::Type, + llvm::ArrayRef args) { + assert(args.size() == 3); + mlir::Value arg0 = fir::getBase(args[0]); + mlir::Value arg1 = fir::getBase(args[1]); + mlir::Value arg2 = fir::getBase(args[2]); + mlir::Type type0 = fir::unwrapRefType(arg0.getType()); + bool isCharRslt = fir::isa_char(type0); // result is same as first argument + mlir::Value mask = builder.createConvert(loc, builder.getI1Type(), arg2); + auto rslt = builder.create(loc, mask, arg0, arg1); + if (isCharRslt) { + // Need a CharBoxValue for character results + const fir::CharBoxValue *charBox = args[0].getCharBox(); + fir::CharBoxValue charRslt(rslt, charBox->getLen()); + return charRslt; + } + return rslt; +} + // MINLOC fir::ExtendedValue IntrinsicLibrary::genMinloc(mlir::Type resultType, diff --git a/flang/test/Lower/Intrinsics/merge.f90 b/flang/test/Lower/Intrinsics/merge.f90 new file mode 100644 index 0000000000000..a9668e4104268 --- /dev/null +++ b/flang/test/Lower/Intrinsics/merge.f90 @@ -0,0 +1,43 @@ +! RUN: bbc -emit-fir %s -o - | FileCheck %s +! RUN: %flang_fc1 -emit-fir %s -o - | FileCheck %s + +! CHECK-LABEL: func @_QPmerge_test( +! CHECK-SAME: %[[arg0:.*]]: !fir.ref>{{.*}}, %[[arg1:.*]]: index{{.*}}, %[[arg2:[^:]+]]: !fir.boxchar<1>{{.*}}, %[[arg3:[^:]+]]: !fir.boxchar<1>{{.*}}, %[[arg4:.*]]: !fir.ref>{{.*}}) -> !fir.boxchar<1> { +function merge_test(o1, o2, mask) +character :: o1, o2, merge_test +logical :: mask +merge_test = merge(o1, o2, mask) +! CHECK: %[[a0:.*]]:2 = fir.unboxchar %[[arg2]] : (!fir.boxchar<1>) -> (!fir.ref>, index) +! CHECK-DAG: %[[a1:.*]]:2 = fir.unboxchar %[[arg3]] : (!fir.boxchar<1>) -> (!fir.ref>, index) +! CHECK: %[[a2:.*]] = fir.load %[[arg4]] : !fir.ref> +! CHECK: %[[a3:.*]] = fir.convert %[[a2]] : (!fir.logical<4>) -> i1 +! CHECK: %[[a4:.*]] = arith.select %[[a3]], %[[a0]]#0, %[[a1]]#0 : !fir.ref> +! CHECK-DAG: %{{.*}} = fir.convert %[[a4]] : (!fir.ref>) -> !fir.ref +end + +! CHECK-LABEL: func @_QPmerge_test2( +! CHECK-SAME: %[[arg0:[^:]+]]: !fir.ref{{.*}}, %[[arg1:[^:]+]]: !fir.ref{{.*}}, %[[arg2:.*]]: !fir.ref>{{.*}}) -> i32 { +function merge_test2(o1, o2, mask) +integer :: o1, o2, merge_test2 +logical :: mask +merge_test2 = merge(o1, o2, mask) +! CHECK: %[[a1:.*]] = fir.load %[[arg0]] : !fir.ref +! CHECK: %[[a2:.*]] = fir.load %[[arg1]] : !fir.ref +! CHECK: %[[a3:.*]] = fir.load %[[arg2]] : !fir.ref> +! CHECK: %[[a4:.*]] = fir.convert %[[a3]] : (!fir.logical<4>) -> i1 +! CHECK: %{{.*}} = arith.select %[[a4]], %[[a1]], %[[a2]] : i32 +end + +! CHECK-LABEL: func @_QPmerge_test3( +! CHECK-SAME: %[[arg0:[^:]+]]: !fir.ref>>{{.*}}, %[[arg1:[^:]+]]: !fir.ref>{{.*}}, %[[arg2:[^:]+]]: !fir.ref>{{.*}}, %[[arg3:.*]]: !fir.ref>{{.*}}) { +subroutine merge_test3(result, o1, o2, mask) +type t + integer :: i +end type +type(t) :: result(10), o1, o2 +logical :: mask +result = merge(o1, o2, mask) +! CHECK: %[[mask:.*]] = fir.load %[[arg3]] : !fir.ref> +! CHECK: %[[mask_cast:.*]] = fir.convert %[[mask]] : (!fir.logical<4>) -> i1 +! CHECK: = arith.select %[[mask_cast]], %[[arg1]], %[[arg2]] : !fir.ref> +end