mirror of
https://github.com/intel/llvm.git
synced 2026-02-03 02:26:27 +08:00
c64770506b
In the previous state, we were relying on forcing the linker to include all libraries in the final binary and the global initializer to self-register every piece of the system. This change help moving away from this model, and allow users to compose pieces more freely. The current change is only "fixing" the dialect registration and avoiding relying on "whole link" for the passes. The translation is still relying on the global registry, and some refactoring is needed to make this all more convenient. Differential Revision: https://reviews.llvm.org/D74461
90 lines
2.9 KiB
C++
90 lines
2.9 KiB
C++
//===- OpStats.cpp - Prints stats of operations in module -----------------===//
|
|
//
|
|
// 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/IR/Module.h"
|
|
#include "mlir/IR/Operation.h"
|
|
#include "mlir/IR/OperationSupport.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Transforms/Passes.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/Support/Format.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
struct PrintOpStatsPass : public ModulePass<PrintOpStatsPass> {
|
|
explicit PrintOpStatsPass(raw_ostream &os = llvm::errs()) : os(os) {}
|
|
|
|
// Prints the resultant operation statistics post iterating over the module.
|
|
void runOnModule() override;
|
|
|
|
// Print summary of op stats.
|
|
void printSummary();
|
|
|
|
private:
|
|
llvm::StringMap<int64_t> opCount;
|
|
raw_ostream &os;
|
|
};
|
|
} // namespace
|
|
|
|
void PrintOpStatsPass::runOnModule() {
|
|
opCount.clear();
|
|
|
|
// Compute the operation statistics for each function in the module.
|
|
for (auto &op : getModule())
|
|
op.walk([&](Operation *op) { ++opCount[op->getName().getStringRef()]; });
|
|
printSummary();
|
|
}
|
|
|
|
void PrintOpStatsPass::printSummary() {
|
|
os << "Operations encountered:\n";
|
|
os << "-----------------------\n";
|
|
SmallVector<StringRef, 64> sorted(opCount.keys());
|
|
llvm::sort(sorted);
|
|
|
|
// Split an operation name from its dialect prefix.
|
|
auto splitOperationName = [](StringRef opName) {
|
|
auto splitName = opName.split('.');
|
|
return splitName.second.empty() ? std::make_pair("", splitName.first)
|
|
: splitName;
|
|
};
|
|
|
|
// Compute the largest dialect and operation name.
|
|
StringRef dialectName, opName;
|
|
size_t maxLenOpName = 0, maxLenDialect = 0;
|
|
for (const auto &key : sorted) {
|
|
std::tie(dialectName, opName) = splitOperationName(key);
|
|
maxLenDialect = std::max(maxLenDialect, dialectName.size());
|
|
maxLenOpName = std::max(maxLenOpName, opName.size());
|
|
}
|
|
|
|
for (const auto &key : sorted) {
|
|
std::tie(dialectName, opName) = splitOperationName(key);
|
|
|
|
// Left-align the names (aligning on the dialect) and right-align the count
|
|
// below. The alignment is for readability and does not affect CSV/FileCheck
|
|
// parsing.
|
|
if (dialectName.empty())
|
|
os.indent(maxLenDialect + 3);
|
|
else
|
|
os << llvm::right_justify(dialectName, maxLenDialect + 2) << '.';
|
|
|
|
// Left justify the operation name.
|
|
os << llvm::left_justify(opName, maxLenOpName) << " , " << opCount[key]
|
|
<< '\n';
|
|
}
|
|
}
|
|
|
|
std::unique_ptr<OpPassBase<ModuleOp>> mlir::createPrintOpStatsPass() {
|
|
return std::make_unique<PrintOpStatsPass>();
|
|
}
|
|
|
|
static PassRegistration<PrintOpStatsPass>
|
|
pass("print-op-stats", "Print statistics of operations");
|