9a277af2d4
This revision adds support for generating utilities for passes such as options/statistics/etc. that can be inferred from the tablegen definition. This removes additional boilerplate from the pass, and also makes it easier to remove the reliance on the pass registry to provide certain things(e.g. the pass argument). Differential Revision: https://reviews.llvm.org/D76659
87 lines
2.6 KiB
C++
87 lines
2.6 KiB
C++
//===- ViewRegionGraph.cpp - View/write graphviz graphs -------------------===//
|
|
//
|
|
// 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/Transforms/ViewRegionGraph.h"
|
|
#include "mlir/IR/RegionGraphTraits.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace llvm {
|
|
|
|
// Specialize DOTGraphTraits to produce more readable output.
|
|
template <> struct DOTGraphTraits<Region *> : public DefaultDOTGraphTraits {
|
|
using DefaultDOTGraphTraits::DefaultDOTGraphTraits;
|
|
|
|
static std::string getNodeLabel(Block *Block, Region *);
|
|
};
|
|
|
|
std::string DOTGraphTraits<Region *>::getNodeLabel(Block *Block, Region *) {
|
|
// Reuse the print output for the node labels.
|
|
std::string outStreamStr;
|
|
raw_string_ostream os(outStreamStr);
|
|
Block->print(os);
|
|
std::string &outStr = os.str();
|
|
|
|
if (outStr[0] == '\n')
|
|
outStr.erase(outStr.begin());
|
|
|
|
// Process string output to left justify the block.
|
|
for (unsigned i = 0; i != outStr.length(); ++i) {
|
|
if (outStr[i] == '\n') {
|
|
outStr[i] = '\\';
|
|
outStr.insert(outStr.begin() + i + 1, 'l');
|
|
}
|
|
}
|
|
|
|
return outStr;
|
|
}
|
|
|
|
} // end namespace llvm
|
|
|
|
void mlir::viewGraph(Region ®ion, const Twine &name, bool shortNames,
|
|
const Twine &title, llvm::GraphProgram::Name program) {
|
|
llvm::ViewGraph(®ion, name, shortNames, title, program);
|
|
}
|
|
|
|
raw_ostream &mlir::writeGraph(raw_ostream &os, Region ®ion, bool shortNames,
|
|
const Twine &title) {
|
|
return llvm::WriteGraph(os, ®ion, shortNames, title);
|
|
}
|
|
|
|
void mlir::Region::viewGraph(const Twine ®ionName) {
|
|
::mlir::viewGraph(*this, regionName);
|
|
}
|
|
void mlir::Region::viewGraph() { viewGraph("region"); }
|
|
|
|
namespace {
|
|
struct PrintCFGPass : public FunctionPass<PrintCFGPass> {
|
|
/// Include the generated pass utilities.
|
|
#define GEN_PASS_PrintCFG
|
|
#include "mlir/Transforms/Passes.h.inc"
|
|
|
|
PrintCFGPass(raw_ostream &os = llvm::errs(), bool shortNames = false,
|
|
const Twine &title = "")
|
|
: os(os), shortNames(shortNames), title(title.str()) {}
|
|
void runOnFunction() override {
|
|
mlir::writeGraph(os, getFunction().getBody(), shortNames, title);
|
|
}
|
|
|
|
private:
|
|
raw_ostream &os;
|
|
bool shortNames;
|
|
std::string title;
|
|
};
|
|
} // namespace
|
|
|
|
std::unique_ptr<mlir::OpPassBase<mlir::FuncOp>>
|
|
mlir::createPrintCFGGraphPass(raw_ostream &os, bool shortNames,
|
|
const Twine &title) {
|
|
return std::make_unique<PrintCFGPass>(os, shortNames, title);
|
|
}
|