2e39b57837
This reapplication changes debug intrinsic declaration removal to only take place when printing final IR, so that the processing format of the Module does not affect the output. This reverts commit d128448efdd4e2bf3c9bc9a5b43ae642aa78026f.
93 lines
3.3 KiB
C++
93 lines
3.3 KiB
C++
//===--- IRPrintingPasses.cpp - Module and Function printing passes -------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// PrintModulePass and PrintFunctionPass implementations.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/IRPrinter/IRPrintingPasses.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Analysis/ModuleSummaryAnalysis.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/PrintPasses.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
|
|
extern cl::opt<bool> WriteNewDbgInfoFormat;
|
|
|
|
PrintModulePass::PrintModulePass() : OS(dbgs()) {}
|
|
PrintModulePass::PrintModulePass(raw_ostream &OS, const std::string &Banner,
|
|
bool ShouldPreserveUseListOrder,
|
|
bool EmitSummaryIndex)
|
|
: OS(OS), Banner(Banner),
|
|
ShouldPreserveUseListOrder(ShouldPreserveUseListOrder),
|
|
EmitSummaryIndex(EmitSummaryIndex) {}
|
|
|
|
PreservedAnalyses PrintModulePass::run(Module &M, ModuleAnalysisManager &AM) {
|
|
// RemoveDIs: Regardless of the format we've processed this module in, use
|
|
// `WriteNewDbgInfoFormat` to determine which format we use to write it.
|
|
ScopedDbgInfoFormatSetter FormatSetter(M, WriteNewDbgInfoFormat);
|
|
// Remove intrinsic declarations when printing in the new format.
|
|
// TODO: Move this into Module::setIsNewDbgInfoFormat when we're ready to
|
|
// update test output.
|
|
if (WriteNewDbgInfoFormat)
|
|
M.removeDebugIntrinsicDeclarations();
|
|
|
|
if (llvm::isFunctionInPrintList("*")) {
|
|
if (!Banner.empty())
|
|
OS << Banner << "\n";
|
|
M.print(OS, nullptr, ShouldPreserveUseListOrder);
|
|
} else {
|
|
bool BannerPrinted = false;
|
|
for (const auto &F : M.functions()) {
|
|
if (llvm::isFunctionInPrintList(F.getName())) {
|
|
if (!BannerPrinted && !Banner.empty()) {
|
|
OS << Banner << "\n";
|
|
BannerPrinted = true;
|
|
}
|
|
F.print(OS);
|
|
}
|
|
}
|
|
}
|
|
|
|
ModuleSummaryIndex *Index =
|
|
EmitSummaryIndex ? &(AM.getResult<ModuleSummaryIndexAnalysis>(M))
|
|
: nullptr;
|
|
if (Index) {
|
|
if (Index->modulePaths().empty())
|
|
Index->addModule("");
|
|
Index->print(OS);
|
|
}
|
|
|
|
return PreservedAnalyses::all();
|
|
}
|
|
|
|
PrintFunctionPass::PrintFunctionPass() : OS(dbgs()) {}
|
|
PrintFunctionPass::PrintFunctionPass(raw_ostream &OS, const std::string &Banner)
|
|
: OS(OS), Banner(Banner) {}
|
|
|
|
PreservedAnalyses PrintFunctionPass::run(Function &F,
|
|
FunctionAnalysisManager &) {
|
|
// RemoveDIs: Regardless of the format we've processed this function in, use
|
|
// `WriteNewDbgInfoFormat` to determine which format we use to write it.
|
|
ScopedDbgInfoFormatSetter FormatSetter(F, WriteNewDbgInfoFormat);
|
|
|
|
if (isFunctionInPrintList(F.getName())) {
|
|
if (forcePrintModuleIR())
|
|
OS << Banner << " (function: " << F.getName() << ")\n" << *F.getParent();
|
|
else
|
|
OS << Banner << '\n' << static_cast<Value &>(F);
|
|
}
|
|
|
|
return PreservedAnalyses::all();
|
|
}
|