[llvm] c1f6f30 - [PatternMatch] Add single index InsertValue matcher.
Florian Hahn via llvm-commits
llvm-commits at lists.llvm.org
Thu Nov 12 13:27:39 PST 2020
Author: Florian Hahn
Date: 2020-11-12T21:27:18Z
New Revision: c1f6f300404a7c948f849d59f555ce138456a0e5
URL: https://github.com/llvm/llvm-project/commit/c1f6f300404a7c948f849d59f555ce138456a0e5
DIFF: https://github.com/llvm/llvm-project/commit/c1f6f300404a7c948f849d59f555ce138456a0e5.diff
LOG: [PatternMatch] Add single index InsertValue matcher.
This patch adds a new matcher for single index InsertValue instructions,
similar to the existing matcher for ExtractValue.
Reviewed By: lebedev.ri
Differential Revision: https://reviews.llvm.org/D91352
Added:
Modified:
llvm/include/llvm/IR/PatternMatch.h
llvm/unittests/IR/PatternMatch.cpp
Removed:
################################################################################
diff --git a/llvm/include/llvm/IR/PatternMatch.h b/llvm/include/llvm/IR/PatternMatch.h
index 562557bdd5c5..c78ea48b4378 100644
--- a/llvm/include/llvm/IR/PatternMatch.h
+++ b/llvm/include/llvm/IR/PatternMatch.h
@@ -2298,6 +2298,29 @@ inline ExtractValue_match<Ind, Val_t> m_ExtractValue(const Val_t &V) {
return ExtractValue_match<Ind, Val_t>(V);
}
+/// Matcher for a single index InsertValue instruction.
+template <int Ind, typename T0, typename T1> struct InsertValue_match {
+ T0 Op0;
+ T1 Op1;
+
+ InsertValue_match(const T0 &Op0, const T1 &Op1) : Op0(Op0), Op1(Op1) {}
+
+ template <typename OpTy> bool match(OpTy *V) {
+ if (auto *I = dyn_cast<InsertValueInst>(V)) {
+ return Op0.match(I->getOperand(0)) && Op1.match(I->getOperand(1)) &&
+ I->getNumIndices() == 1 && Ind == I->getIndices()[0];
+ }
+ return false;
+ }
+};
+
+/// Matches a single index InsertValue instruction.
+template <int Ind, typename Val_t, typename Elt_t>
+inline InsertValue_match<Ind, Val_t, Elt_t> m_InsertValue(const Val_t &Val,
+ const Elt_t &Elt) {
+ return InsertValue_match<Ind, Val_t, Elt_t>(Val, Elt);
+}
+
/// Matches patterns for `vscale`. This can either be a call to `llvm.vscale` or
/// the constant expression
/// `ptrtoint(gep <vscale x 1 x i8>, <vscale x 1 x i8>* null, i32 1>`
diff --git a/llvm/unittests/IR/PatternMatch.cpp b/llvm/unittests/IR/PatternMatch.cpp
index 86d9ed5e4055..9a1a73a5c06f 100644
--- a/llvm/unittests/IR/PatternMatch.cpp
+++ b/llvm/unittests/IR/PatternMatch.cpp
@@ -1581,6 +1581,27 @@ TEST_F(PatternMatchTest, ConstantPredicateType) {
match(CF32NaNWithUndef, cstfp_pred_ty<always_false_pred<APFloat>>()));
}
+TEST_F(PatternMatchTest, InsertValue) {
+ Type *StructTy = StructType::create(IRB.getContext(),
+ {IRB.getInt32Ty(), IRB.getInt64Ty()});
+ Value *Ins0 =
+ IRB.CreateInsertValue(UndefValue::get(StructTy), IRB.getInt32(20), 0);
+ Value *Ins1 = IRB.CreateInsertValue(Ins0, IRB.getInt64(90), 1);
+
+ EXPECT_TRUE(match(Ins0, m_InsertValue<0>(m_Value(), m_Value())));
+ EXPECT_FALSE(match(Ins0, m_InsertValue<1>(m_Value(), m_Value())));
+ EXPECT_FALSE(match(Ins1, m_InsertValue<0>(m_Value(), m_Value())));
+ EXPECT_TRUE(match(Ins1, m_InsertValue<1>(m_Value(), m_Value())));
+
+ EXPECT_TRUE(match(Ins0, m_InsertValue<0>(m_Undef(), m_SpecificInt(20))));
+ EXPECT_FALSE(match(Ins0, m_InsertValue<0>(m_Undef(), m_SpecificInt(0))));
+
+ EXPECT_TRUE(
+ match(Ins1, m_InsertValue<1>(m_InsertValue<0>(m_Value(), m_Value()),
+ m_SpecificInt(90))));
+ EXPECT_FALSE(match(IRB.getInt64(99), m_InsertValue<0>(m_Value(), m_Value())));
+}
+
template <typename T> struct MutableConstTest : PatternMatchTest { };
typedef ::testing::Types<std::tuple<Value*, Instruction*>,
More information about the llvm-commits
mailing list