12611a7fc7
Whereas backward-slice matching provides support to limit traversal by specifying the desired depth level, this pull request introduces support for limiting traversal with a nested matcher (adding forward-slice also). It also adds support for variadic operators, including `anyOf` and `allOf`. Rather than simply stopping traversal when an operation named foo is encountered, one can now define a matcher that specifies different exit conditions. Variadic support implementation within mlir-query is very similar to clang-query.
34 lines
1.2 KiB
C++
34 lines
1.2 KiB
C++
//===--- MatchersInternal.cpp----------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Query/Matcher/MatchersInternal.h"
|
|
#include "llvm/ADT/SetVector.h"
|
|
|
|
namespace mlir::query::matcher {
|
|
|
|
namespace internal {
|
|
|
|
bool allOfVariadicOperator(Operation *op, SetVector<Operation *> *matchedOps,
|
|
ArrayRef<DynMatcher> innerMatchers) {
|
|
return llvm::all_of(innerMatchers, [&](const DynMatcher &matcher) {
|
|
if (matchedOps)
|
|
return matcher.match(op, *matchedOps);
|
|
return matcher.match(op);
|
|
});
|
|
}
|
|
bool anyOfVariadicOperator(Operation *op, SetVector<Operation *> *matchedOps,
|
|
ArrayRef<DynMatcher> innerMatchers) {
|
|
return llvm::any_of(innerMatchers, [&](const DynMatcher &matcher) {
|
|
if (matchedOps)
|
|
return matcher.match(op, *matchedOps);
|
|
return matcher.match(op);
|
|
});
|
|
}
|
|
} // namespace internal
|
|
} // namespace mlir::query::matcher
|