02d9f4d1f1
This commit adds the initial version of the mlir-query tool, which leverages the pre-existing matchers defined in mlir/include/mlir/IR/Matchers.h The tool provides the following set of basic queries: hasOpAttrName(string) hasOpName(string) isConstantOp() isNegInfFloat() isNegZeroFloat() isNonZero() isOne() isOneFloat() isPosInfFloat() isPosZeroFloat() isZero() isZeroFloat() Reviewed By: jpienaar Differential Revision: https://reviews.llvm.org/D155127
83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
//===---- Query.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/Query.h"
|
|
#include "QueryParser.h"
|
|
#include "mlir/Query/Matcher/MatchFinder.h"
|
|
#include "mlir/Query/QuerySession.h"
|
|
#include "mlir/Support/LogicalResult.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
namespace mlir::query {
|
|
|
|
QueryRef parse(llvm::StringRef line, const QuerySession &qs) {
|
|
return QueryParser::parse(line, qs);
|
|
}
|
|
|
|
std::vector<llvm::LineEditor::Completion>
|
|
complete(llvm::StringRef line, size_t pos, const QuerySession &qs) {
|
|
return QueryParser::complete(line, pos, qs);
|
|
}
|
|
|
|
static void printMatch(llvm::raw_ostream &os, QuerySession &qs, Operation *op,
|
|
const std::string &binding) {
|
|
auto fileLoc = op->getLoc()->findInstanceOf<FileLineColLoc>();
|
|
auto smloc = qs.getSourceManager().FindLocForLineAndColumn(
|
|
qs.getBufferId(), fileLoc.getLine(), fileLoc.getColumn());
|
|
qs.getSourceManager().PrintMessage(os, smloc, llvm::SourceMgr::DK_Note,
|
|
"\"" + binding + "\" binds here");
|
|
}
|
|
|
|
Query::~Query() = default;
|
|
|
|
mlir::LogicalResult InvalidQuery::run(llvm::raw_ostream &os,
|
|
QuerySession &qs) const {
|
|
os << errStr << "\n";
|
|
return mlir::failure();
|
|
}
|
|
|
|
mlir::LogicalResult NoOpQuery::run(llvm::raw_ostream &os,
|
|
QuerySession &qs) const {
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult HelpQuery::run(llvm::raw_ostream &os,
|
|
QuerySession &qs) const {
|
|
os << "Available commands:\n\n"
|
|
" match MATCHER, m MATCHER "
|
|
"Match the mlir against the given matcher.\n"
|
|
" quit "
|
|
"Terminates the query session.\n\n";
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult QuitQuery::run(llvm::raw_ostream &os,
|
|
QuerySession &qs) const {
|
|
qs.terminate = true;
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult MatchQuery::run(llvm::raw_ostream &os,
|
|
QuerySession &qs) const {
|
|
int matchCount = 0;
|
|
std::vector<Operation *> matches =
|
|
matcher::MatchFinder().getMatches(qs.getRootOp(), matcher);
|
|
os << "\n";
|
|
for (Operation *op : matches) {
|
|
os << "Match #" << ++matchCount << ":\n\n";
|
|
// Placeholder "root" binding for the initial draft.
|
|
printMatch(os, qs, op, "root");
|
|
}
|
|
os << matchCount << (matchCount == 1 ? " match.\n\n" : " matches.\n\n");
|
|
|
|
return mlir::success();
|
|
}
|
|
|
|
} // namespace mlir::query
|