Summary: Introduce m_Constant() which allows matching a constant operation without forcing the user also to capture the attribute value.
Differential Revision: https://reviews.llvm.org/D72397
/// Creates a matcher instance that binds the constant attribute value to
/// bind_value if match succeeds.
constant_op_binder(AttrT *bind_value) : bind_value(bind_value) {}
+ /// Creates a matcher instance that doesn't bind if match succeeds.
+ constant_op_binder() : bind_value(nullptr) {}
bool match(Operation *op) {
if (op->getNumOperands() > 0 || op->getNumResults() != 1)
SmallVector<OpFoldResult, 1> foldedOp;
if (succeeded(op->fold(/*operands=*/llvm::None, foldedOp))) {
if (auto attr = foldedOp.front().dyn_cast<Attribute>()) {
- if ((*bind_value = attr.dyn_cast<AttrT>()))
+ if (auto attrT = attr.dyn_cast<AttrT>()) {
+ if (bind_value)
+ *bind_value = attrT;
return true;
+ }
}
}
return false;
} // end namespace detail
+/// Matches a constant foldable operation.
+inline detail::constant_op_binder<Attribute> m_Constant() {
+ return detail::constant_op_binder<Attribute>();
+}
+
/// Matches a value from a constant foldable operation and writes the value to
/// bind_value.
template <typename AttrT>
};
// If this operation is already a constant, there is nothing to do.
- Attribute unused;
- if (matchPattern(op, m_Constant(&unused)))
+ if (matchPattern(op, m_Constant()))
return cleanupFailure();
// Check to see if any operands to the operation is constant and whether
// CHECK-LABEL: test2
// CHECK: Pattern add(add(a, constant), a) matched and bound constant to: 1.000000e+00
+// CHECK: Pattern add(add(a, constant), a) matched
auto a = m_Val(f.getArgument(0));
FloatAttr floatAttr;
auto p = m_Op<MulFOp>(a, m_Op<AddFOp>(a, m_Constant(&floatAttr)));
+ auto p1 = m_Op<MulFOp>(a, m_Op<AddFOp>(a, m_Constant()));
// Last operation that is not the terminator.
Operation *lastOp = f.getBody().front().back().getPrevNode();
if (p.match(lastOp))
llvm::outs()
<< "Pattern add(add(a, constant), a) matched and bound constant to: "
<< floatAttr.getValueAsDouble() << "\n";
+ if (p1.match(lastOp))
+ llvm::outs() << "Pattern add(add(a, constant), a) matched\n";
}
void TestMatchers::runOnFunction() {