Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-02-06 01:22:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[flang][lowering] Add support for lowering of the ibset intrinsic

Browse Source 
This patch adds support for lowering of the `ibset` intrinsic from Fortran
to the FIR dialect of MLIR.

This is part of the upstreaming effort from the `fir-dev` branch in [1].

[1] https://github.com/flang-compiler/f18-llvm-project

Co-authored-by: Jean Perier <jperier@nvidia.com>
Co-authored-by: Valentin Clement <clementval@gmail.com>
Co-authored-by: V Donaldson <vdonaldson@nvidia.com>

Differential Revision: https://reviews.llvm.org/D121717
This commit is contained in:
Andrzej Warzynski 2022-03-15 17:07:07 +00:00
parent 91fad1167a
commit 63bbac1065
2 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
flang/lib/Lower/IntrinsicCall.cpp
View File

@ -449,6 +449,7 @@ struct IntrinsicLibrary {
/// in the llvm::ArrayRef.
mlir::Value genIand(mlir::Type, llvm::ArrayRef<mlir::Value>);
mlir::Value genIbits(mlir::Type, llvm::ArrayRef<mlir::Value>);
mlir::Value genIbset(mlir::Type, llvm::ArrayRef<mlir::Value>);
fir::ExtendedValue genLbound(mlir::Type, llvm::ArrayRef<fir::ExtendedValue>);
fir::ExtendedValue genNull(mlir::Type, llvm::ArrayRef<fir::ExtendedValue>);
fir::ExtendedValue genLen(mlir::Type, llvm::ArrayRef<fir::ExtendedValue>);
@ -595,6 +596,7 @@ static constexpr IntrinsicHandler handlers[]{
/*isElemental=*/false},
{"iand", &I::genIand},
{"ibits", &I::genIbits},
{"ibset", &I::genIbset},
{"len",
&I::genLen,
{{{"string", asInquired}, {"kind", asValue}}},
@ -1682,6 +1684,20 @@ mlir::Value IntrinsicLibrary::genIbits(mlir::Type resultType,
return builder.create<mlir::arith::SelectOp>(loc, lenIsZero, zero, res2);
}
// IBSET
mlir::Value IntrinsicLibrary::genIbset(mlir::Type resultType,
llvm::ArrayRef<mlir::Value> args) {
// A conformant IBSET(I,POS) call satisfies:
// POS >= 0
// POS < BIT_SIZE(I)
// Return: I | (1 << POS)
assert(args.size() == 2);
mlir::Value pos = builder.createConvert(loc, resultType, args[1]);
mlir::Value one = builder.createIntegerConstant(loc, resultType, 1);
auto mask = builder.create<mlir::arith::ShLIOp>(loc, one, pos);
return builder.create<mlir::arith::OrIOp>(loc, args[0], mask);
}
// LEN
// Note that this is only used for an unrestricted intrinsic LEN call.
// Other uses of LEN are rewritten as descriptor inquiries by the front-end.

17
flang/test/Lower/Intrinsics/ibset.f90 Normal file
View File

@ -0,0 +1,17 @@
! RUN: bbc -emit-fir %s -o - | FileCheck %s
! RUN: %flang_fc1 -emit-fir %s -o - | FileCheck %s
! CHECK-LABEL: ibset_test
function ibset_test(i, j)
! CHECK-DAG: %[[result:.*]] = fir.alloca i32 {bindc_name = "ibset_test"
! CHECK-DAG: %[[i:.*]] = fir.load %arg0 : !fir.ref<i32>
! CHECK-DAG: %[[j:.*]] = fir.load %arg1 : !fir.ref<i32>
! CHECK-DAG: %[[VAL_5:.*]] = arith.constant 1 : i32
! CHECK: %[[VAL_6:.*]] = arith.shli %[[VAL_5]], %[[j]] : i32
! CHECK: %[[VAL_7:.*]] = arith.ori %[[i]], %[[VAL_6]] : i32
! CHECK: fir.store %[[VAL_7]] to %[[result]] : !fir.ref<i32>
! CHECK: %[[VAL_8:.*]] = fir.load %[[result]] : !fir.ref<i32>
! CHECK: return %[[VAL_8]] : i32
ibset_test = ibset(i, j)
end