maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[PatternMatch] Add m_Store pattern match helper 

Differential Revision: https://reviews.llvm.org/D48279

llvm-svn: 335100
This commit is contained in:
Sjoerd Meijer 2018-06-20 07:27:45 +00:00
parent d23b6831de
commit c607901446
2 changed files with 63 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
llvm/include/llvm/IR/PatternMatch.h
View File

@ -1192,6 +1192,33 @@ template <typename OpTy> inline LoadClass_match<OpTy> m_Load(const OpTy &Op) {
return LoadClass_match<OpTy>(Op);
}
//===----------------------------------------------------------------------===//
// Matcher for StoreInst classes
//
template <typename ValueOp_t, typename PointerOp_t> struct StoreClass_match {
ValueOp_t ValueOp;
PointerOp_t PointerOp;
StoreClass_match(const ValueOp_t &ValueOpMatch,
const PointerOp_t &PointerOpMatch) :
ValueOp(ValueOpMatch), PointerOp(PointerOpMatch) {}
template <typename OpTy> bool match(OpTy *V) {
if (auto *LI = dyn_cast<StoreInst>(V))
return ValueOp.match(LI->getValueOperand()) &&
PointerOp.match(LI->getPointerOperand());
return false;
}
};
/// Matches StoreInst.
template <typename ValueOpTy, typename PointerOpTy>
inline StoreClass_match<ValueOpTy, PointerOpTy>
m_Store(const ValueOpTy &ValueOp, const PointerOpTy &PointerOp) {
return StoreClass_match<ValueOpTy, PointerOpTy>(ValueOp, PointerOp);
}
//===----------------------------------------------------------------------===//
// Matchers for unary operators
//

36
llvm/unittests/IR/PatternMatch.cpp
View File

@ -390,6 +390,42 @@ TEST_F(PatternMatchTest, OverflowingBinOps) {
EXPECT_FALSE(m_NUWShl(m_Value(), m_Value()).match(IRB.CreateNUWAdd(L, R)));
}
TEST_F(PatternMatchTest, LoadStoreOps) {
// Create this load/store sequence:
//
// %p = alloca i32*
// %0 = load i32*, i32** %p
// store i32 42, i32* %0
Value *Alloca = IRB.CreateAlloca(IRB.getInt32Ty());
Value *LoadInst = IRB.CreateLoad(Alloca);
Value *FourtyTwo = IRB.getInt32(42);
Value *StoreInst = IRB.CreateStore(FourtyTwo, Alloca);
Value *MatchLoad, *MatchStoreVal, *MatchStorePointer;
EXPECT_TRUE(m_Load(m_Value(MatchLoad)).match(LoadInst));
EXPECT_EQ(Alloca, MatchLoad);
EXPECT_TRUE(m_Load(m_Specific(Alloca)).match(LoadInst));
EXPECT_FALSE(m_Load(m_Value(MatchLoad)).match(Alloca));
EXPECT_TRUE(m_Store(m_Value(MatchStoreVal), m_Value(MatchStorePointer))
.match(StoreInst));
EXPECT_EQ(FourtyTwo, MatchStoreVal);
EXPECT_EQ(Alloca, MatchStorePointer);
EXPECT_FALSE(m_Store(m_Value(MatchStoreVal), m_Value(MatchStorePointer))
.match(Alloca));
EXPECT_TRUE(m_Store(m_SpecificInt(42), m_Specific(Alloca))
.match(StoreInst));
EXPECT_FALSE(m_Store(m_SpecificInt(42), m_Specific(FourtyTwo))
.match(StoreInst));
EXPECT_FALSE(m_Store(m_SpecificInt(43), m_Specific(Alloca))
.match(StoreInst));
}
TEST_F(PatternMatchTest, VectorOps) {
// Build up small tree of vector operations
//