teak-llvm/llvm/lib/Transforms/Utils/NameAnonGlobals.cpp
Reid Kleckner 05da2fe521 Sink all InitializePasses.h includes
This file lists every pass in LLVM, and is included by Pass.h, which is
very popular. Every time we add, remove, or rename a pass in LLVM, it
caused lots of recompilation.

I found this fact by looking at this table, which is sorted by the
number of times a file was changed over the last 100,000 git commits
multiplied by the number of object files that depend on it in the
current checkout:
  recompiles    touches affected_files  header
  342380        95      3604    llvm/include/llvm/ADT/STLExtras.h
  314730        234     1345    llvm/include/llvm/InitializePasses.h
  307036        118     2602    llvm/include/llvm/ADT/APInt.h
  213049        59      3611    llvm/include/llvm/Support/MathExtras.h
  170422        47      3626    llvm/include/llvm/Support/Compiler.h
  162225        45      3605    llvm/include/llvm/ADT/Optional.h
  158319        63      2513    llvm/include/llvm/ADT/Triple.h
  140322        39      3598    llvm/include/llvm/ADT/StringRef.h
  137647        59      2333    llvm/include/llvm/Support/Error.h
  131619        73      1803    llvm/include/llvm/Support/FileSystem.h

Before this change, touching InitializePasses.h would cause 1345 files
to recompile. After this change, touching it only causes 550 compiles in
an incremental rebuild.

Reviewers: bkramer, asbirlea, bollu, jdoerfert

Differential Revision: https://reviews.llvm.org/D70211
2019-11-13 16:34:37 -08:00

121 lines
3.3 KiB
C++

//===- NameAnonGlobals.cpp - ThinLTO Support: Name Unnamed Globals --------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This file implements naming anonymous globals to make sure they can be
// referred to by ThinLTO.
//
//===----------------------------------------------------------------------===//
#include "llvm/Transforms/Utils/NameAnonGlobals.h"
#include "llvm/ADT/SmallString.h"
#include "llvm/IR/Module.h"
#include "llvm/InitializePasses.h"
#include "llvm/Support/MD5.h"
#include "llvm/Transforms/Utils/ModuleUtils.h"
using namespace llvm;
namespace {
// Compute a "unique" hash for the module based on the name of the public
// globals.
class ModuleHasher {
Module &TheModule;
std::string TheHash;
public:
ModuleHasher(Module &M) : TheModule(M) {}
/// Return the lazily computed hash.
std::string &get() {
if (!TheHash.empty())
// Cache hit :)
return TheHash;
MD5 Hasher;
for (auto &F : TheModule) {
if (F.isDeclaration() || F.hasLocalLinkage() || !F.hasName())
continue;
auto Name = F.getName();
Hasher.update(Name);
}
for (auto &GV : TheModule.globals()) {
if (GV.isDeclaration() || GV.hasLocalLinkage() || !GV.hasName())
continue;
auto Name = GV.getName();
Hasher.update(Name);
}
// Now return the result.
MD5::MD5Result Hash;
Hasher.final(Hash);
SmallString<32> Result;
MD5::stringifyResult(Hash, Result);
TheHash = Result.str();
return TheHash;
}
};
} // end anonymous namespace
// Rename all the anon globals in the module
bool llvm::nameUnamedGlobals(Module &M) {
bool Changed = false;
ModuleHasher ModuleHash(M);
int count = 0;
auto RenameIfNeed = [&](GlobalValue &GV) {
if (GV.hasName())
return;
GV.setName(Twine("anon.") + ModuleHash.get() + "." + Twine(count++));
Changed = true;
};
for (auto &GO : M.global_objects())
RenameIfNeed(GO);
for (auto &GA : M.aliases())
RenameIfNeed(GA);
return Changed;
}
namespace {
// Legacy pass that provides a name to every anon globals.
class NameAnonGlobalLegacyPass : public ModulePass {
public:
/// Pass identification, replacement for typeid
static char ID;
/// Specify pass name for debug output
StringRef getPassName() const override { return "Name Anon Globals"; }
explicit NameAnonGlobalLegacyPass() : ModulePass(ID) {}
bool runOnModule(Module &M) override { return nameUnamedGlobals(M); }
};
char NameAnonGlobalLegacyPass::ID = 0;
} // anonymous namespace
PreservedAnalyses NameAnonGlobalPass::run(Module &M,
ModuleAnalysisManager &AM) {
if (!nameUnamedGlobals(M))
return PreservedAnalyses::all();
return PreservedAnalyses::none();
}
INITIALIZE_PASS_BEGIN(NameAnonGlobalLegacyPass, "name-anon-globals",
"Provide a name to nameless globals", false, false)
INITIALIZE_PASS_END(NameAnonGlobalLegacyPass, "name-anon-globals",
"Provide a name to nameless globals", false, false)
namespace llvm {
ModulePass *createNameAnonGlobalPass() {
return new NameAnonGlobalLegacyPass();
}
}