[llvm] 0f9d299 - [VP] getDeclarationForParams
Simon Moll via llvm-commits
llvm-commits at lists.llvm.org
Tue Jun 8 05:42:52 PDT 2021
Author: Simon Moll
Date: 2021-06-08T14:21:28+02:00
New Revision: 0f9d299122f1223e93c54f10401a608f5d481314
URL: https://github.com/llvm/llvm-project/commit/0f9d299122f1223e93c54f10401a608f5d481314
DIFF: https://github.com/llvm/llvm-project/commit/0f9d299122f1223e93c54f10401a608f5d481314.diff
LOG: [VP] getDeclarationForParams
`VPIntrinsic::getDeclarationForParams` creates a vp intrinsic
declaration for parameters you want to call it with. This is in
preparation of a new builder class that makes emitting vp intrinsic code
nearly as convenient as using a plain ir builder (aka `VectorBuilder`,
to be used by D99750).
Reviewed By: frasercrmck, craig.topper, vkmr
Differential Revision: https://reviews.llvm.org/D102686
Added:
Modified:
llvm/include/llvm/IR/IntrinsicInst.h
llvm/lib/IR/IntrinsicInst.cpp
llvm/unittests/IR/VPIntrinsicTest.cpp
Removed:
################################################################################
diff --git a/llvm/include/llvm/IR/IntrinsicInst.h b/llvm/include/llvm/IR/IntrinsicInst.h
index 338103a82ee87..606c2b570c1b7 100644
--- a/llvm/include/llvm/IR/IntrinsicInst.h
+++ b/llvm/include/llvm/IR/IntrinsicInst.h
@@ -389,6 +389,11 @@ class DbgLabelInst : public DbgInfoIntrinsic {
/// This is the common base class for vector predication intrinsics.
class VPIntrinsic : public IntrinsicInst {
public:
+ /// \brief Declares a llvm.vp.* intrinsic in \p M that matches the parameters
+ /// \p Params.
+ static Function *getDeclarationForParams(Module *M, Intrinsic::ID,
+ ArrayRef<Value *> Params);
+
static Optional<unsigned> getMaskParamPos(Intrinsic::ID IntrinsicID);
static Optional<unsigned> getVectorLengthParamPos(Intrinsic::ID IntrinsicID);
diff --git a/llvm/lib/IR/IntrinsicInst.cpp b/llvm/lib/IR/IntrinsicInst.cpp
index 37ee2f34a4042..30c7ceb3c3f0e 100644
--- a/llvm/lib/IR/IntrinsicInst.cpp
+++ b/llvm/lib/IR/IntrinsicInst.cpp
@@ -421,6 +421,17 @@ bool VPIntrinsic::canIgnoreVectorLengthParam() const {
return false;
}
+Function *VPIntrinsic::getDeclarationForParams(Module *M, Intrinsic::ID VPID,
+ ArrayRef<Value *> Params) {
+ assert(isVPIntrinsic(VPID) && "not a VP intrinsic");
+
+ // TODO: Extend this for other VP intrinsics as they are upstreamed. This
+ // works for binary arithmetic VP intrinsics.
+ auto *VPFunc = Intrinsic::getDeclaration(M, VPID, Params[0]->getType());
+ assert(VPFunc && "Could not declare VP intrinsic");
+ return VPFunc;
+}
+
Instruction::BinaryOps BinaryOpIntrinsic::getBinaryOp() const {
switch (getIntrinsicID()) {
case Intrinsic::uadd_with_overflow:
diff --git a/llvm/unittests/IR/VPIntrinsicTest.cpp b/llvm/unittests/IR/VPIntrinsicTest.cpp
index ce77fe437dcf8..f16e55e6f247c 100644
--- a/llvm/unittests/IR/VPIntrinsicTest.cpp
+++ b/llvm/unittests/IR/VPIntrinsicTest.cpp
@@ -222,4 +222,36 @@ TEST_F(VPIntrinsicTest, IntrinsicIDRoundTrip) {
ASSERT_NE(FullTripCounts, 0u);
}
+/// Check that VPIntrinsic::getDeclarationForParams works.
+TEST_F(VPIntrinsicTest, VPIntrinsicDeclarationForParams) {
+ std::unique_ptr<Module> M = CreateVPDeclarationModule();
+ assert(M);
+
+ auto OutM = std::make_unique<Module>("", M->getContext());
+
+ for (auto &F : *M) {
+ auto *FuncTy = F.getFunctionType();
+
+ // Declare intrinsic anew with explicit types.
+ std::vector<Value *> Values;
+ for (auto *ParamTy : FuncTy->params())
+ Values.push_back(UndefValue::get(ParamTy));
+
+ ASSERT_NE(F.getIntrinsicID(), Intrinsic::not_intrinsic);
+ auto *NewDecl = VPIntrinsic::getDeclarationForParams(
+ OutM.get(), F.getIntrinsicID(), Values);
+ ASSERT_TRUE(NewDecl);
+
+ // Check that 'old decl' == 'new decl'.
+ ASSERT_EQ(F.getIntrinsicID(), NewDecl->getIntrinsicID());
+ auto ItNewParams = NewDecl->getFunctionType()->param_begin();
+ auto EndItNewParams = NewDecl->getFunctionType()->param_end();
+ for (auto *ParamTy : FuncTy->params()) {
+ ASSERT_NE(ItNewParams, EndItNewParams);
+ ASSERT_EQ(*ItNewParams, ParamTy);
+ ++ItNewParams;
+ }
+ }
+}
+
} // end anonymous namespace
More information about the llvm-commits
mailing list