1 //===- LTO.cpp ------------------------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "LTO.h" 10 #include "Config.h" 11 #include "InputFiles.h" 12 #include "Symbols.h" 13 #include "lld/Common/Args.h" 14 #include "lld/Common/ErrorHandler.h" 15 #include "lld/Common/Strings.h" 16 #include "lld/Common/TargetOptionsCommandFlags.h" 17 #include "llvm/ADT/STLExtras.h" 18 #include "llvm/ADT/SmallString.h" 19 #include "llvm/ADT/StringRef.h" 20 #include "llvm/ADT/Twine.h" 21 #include "llvm/Bitcode/BitcodeWriter.h" 22 #include "llvm/IR/DiagnosticPrinter.h" 23 #include "llvm/LTO/Caching.h" 24 #include "llvm/LTO/Config.h" 25 #include "llvm/LTO/LTO.h" 26 #include "llvm/Object/SymbolicFile.h" 27 #include "llvm/Support/CodeGen.h" 28 #include "llvm/Support/Error.h" 29 #include "llvm/Support/FileSystem.h" 30 #include "llvm/Support/MemoryBuffer.h" 31 #include "llvm/Support/raw_ostream.h" 32 #include <algorithm> 33 #include <cstddef> 34 #include <memory> 35 #include <string> 36 #include <system_error> 37 #include <vector> 38 39 using namespace llvm; 40 using namespace llvm::object; 41 using namespace lld; 42 using namespace lld::coff; 43 44 // Creates an empty file to and returns a raw_fd_ostream to write to it. 45 static std::unique_ptr<raw_fd_ostream> openFile(StringRef file) { 46 std::error_code ec; 47 auto ret = 48 std::make_unique<raw_fd_ostream>(file, ec, sys::fs::OpenFlags::OF_None); 49 if (ec) { 50 error("cannot open " + file + ": " + ec.message()); 51 return nullptr; 52 } 53 return ret; 54 } 55 56 static std::string getThinLTOOutputFile(StringRef path) { 57 return lto::getThinLTOOutputFile( 58 std::string(path), std::string(config->thinLTOPrefixReplace.first), 59 std::string(config->thinLTOPrefixReplace.second)); 60 } 61 62 static lto::Config createConfig() { 63 lto::Config c; 64 c.Options = initTargetOptionsFromCodeGenFlags(); 65 66 // Always emit a section per function/datum with LTO. LLVM LTO should get most 67 // of the benefit of linker GC, but there are still opportunities for ICF. 68 c.Options.FunctionSections = true; 69 c.Options.DataSections = true; 70 71 // Use static reloc model on 32-bit x86 because it usually results in more 72 // compact code, and because there are also known code generation bugs when 73 // using the PIC model (see PR34306). 74 if (config->machine == COFF::IMAGE_FILE_MACHINE_I386) 75 c.RelocModel = Reloc::Static; 76 else 77 c.RelocModel = Reloc::PIC_; 78 c.DisableVerify = true; 79 c.DiagHandler = diagnosticHandler; 80 c.OptLevel = config->ltoo; 81 c.CPU = getCPUStr(); 82 c.MAttrs = getMAttrs(); 83 c.CGOptLevel = args::getCGOptLevel(config->ltoo); 84 c.AlwaysEmitRegularLTOObj = !config->ltoObjPath.empty(); 85 c.UseNewPM = config->ltoNewPassManager; 86 c.DebugPassManager = config->ltoDebugPassManager; 87 c.CSIRProfile = std::string(config->ltoCSProfileFile); 88 c.RunCSIRInstr = config->ltoCSProfileGenerate; 89 90 if (config->saveTemps) 91 checkError(c.addSaveTemps(std::string(config->outputFile) + ".", 92 /*UseInputModulePath*/ true)); 93 return c; 94 } 95 96 BitcodeCompiler::BitcodeCompiler() { 97 // Initialize indexFile. 98 if (!config->thinLTOIndexOnlyArg.empty()) 99 indexFile = openFile(config->thinLTOIndexOnlyArg); 100 101 // Initialize ltoObj. 102 lto::ThinBackend backend; 103 if (config->thinLTOIndexOnly) { 104 auto OnIndexWrite = [&](StringRef S) { thinIndices.erase(S); }; 105 backend = lto::createWriteIndexesThinBackend( 106 std::string(config->thinLTOPrefixReplace.first), 107 std::string(config->thinLTOPrefixReplace.second), 108 config->thinLTOEmitImportsFiles, indexFile.get(), OnIndexWrite); 109 } else { 110 backend = lto::createInProcessThinBackend( 111 llvm::heavyweight_hardware_concurrency(config->thinLTOJobs)); 112 } 113 114 ltoObj = std::make_unique<lto::LTO>(createConfig(), backend, 115 config->ltoPartitions); 116 } 117 118 BitcodeCompiler::~BitcodeCompiler() = default; 119 120 static void undefine(Symbol *s) { replaceSymbol<Undefined>(s, s->getName()); } 121 122 void BitcodeCompiler::add(BitcodeFile &f) { 123 lto::InputFile &obj = *f.obj; 124 unsigned symNum = 0; 125 std::vector<Symbol *> symBodies = f.getSymbols(); 126 std::vector<lto::SymbolResolution> resols(symBodies.size()); 127 128 if (config->thinLTOIndexOnly) 129 thinIndices.insert(obj.getName()); 130 131 // Provide a resolution to the LTO API for each symbol. 132 for (const lto::InputFile::Symbol &objSym : obj.symbols()) { 133 Symbol *sym = symBodies[symNum]; 134 lto::SymbolResolution &r = resols[symNum]; 135 ++symNum; 136 137 // Ideally we shouldn't check for SF_Undefined but currently IRObjectFile 138 // reports two symbols for module ASM defined. Without this check, lld 139 // flags an undefined in IR with a definition in ASM as prevailing. 140 // Once IRObjectFile is fixed to report only one symbol this hack can 141 // be removed. 142 r.Prevailing = !objSym.isUndefined() && sym->getFile() == &f; 143 r.VisibleToRegularObj = sym->isUsedInRegularObj; 144 if (r.Prevailing) 145 undefine(sym); 146 147 // We tell LTO to not apply interprocedural optimization for wrapped 148 // (with -wrap) symbols because otherwise LTO would inline them while 149 // their values are still not final. 150 r.LinkerRedefined = !sym->canInline; 151 } 152 checkError(ltoObj->add(std::move(f.obj), resols)); 153 } 154 155 // Merge all the bitcode files we have seen, codegen the result 156 // and return the resulting objects. 157 std::vector<InputFile *> BitcodeCompiler::compile() { 158 unsigned maxTasks = ltoObj->getMaxTasks(); 159 buf.resize(maxTasks); 160 files.resize(maxTasks); 161 162 // The /lldltocache option specifies the path to a directory in which to cache 163 // native object files for ThinLTO incremental builds. If a path was 164 // specified, configure LTO to use it as the cache directory. 165 lto::NativeObjectCache cache; 166 if (!config->ltoCache.empty()) 167 cache = check(lto::localCache( 168 config->ltoCache, [&](size_t task, std::unique_ptr<MemoryBuffer> mb) { 169 files[task] = std::move(mb); 170 })); 171 172 checkError(ltoObj->run( 173 [&](size_t task) { 174 return std::make_unique<lto::NativeObjectStream>( 175 std::make_unique<raw_svector_ostream>(buf[task])); 176 }, 177 cache)); 178 179 // Emit empty index files for non-indexed files 180 for (StringRef s : thinIndices) { 181 std::string path = getThinLTOOutputFile(s); 182 openFile(path + ".thinlto.bc"); 183 if (config->thinLTOEmitImportsFiles) 184 openFile(path + ".imports"); 185 } 186 187 // ThinLTO with index only option is required to generate only the index 188 // files. After that, we exit from linker and ThinLTO backend runs in a 189 // distributed environment. 190 if (config->thinLTOIndexOnly) { 191 if (!config->ltoObjPath.empty()) 192 saveBuffer(buf[0], config->ltoObjPath); 193 if (indexFile) 194 indexFile->close(); 195 return {}; 196 } 197 198 if (!config->ltoCache.empty()) 199 pruneCache(config->ltoCache, config->ltoCachePolicy); 200 201 std::vector<InputFile *> ret; 202 for (unsigned i = 0; i != maxTasks; ++i) { 203 // Assign unique names to LTO objects. This ensures they have unique names 204 // in the PDB if one is produced. The names should look like: 205 // - foo.exe.lto.obj 206 // - foo.exe.lto.1.obj 207 // - ... 208 StringRef ltoObjName = 209 saver.save(Twine(config->outputFile) + ".lto" + 210 (i == 0 ? Twine("") : Twine('.') + Twine(i)) + ".obj"); 211 212 // Get the native object contents either from the cache or from memory. Do 213 // not use the cached MemoryBuffer directly, or the PDB will not be 214 // deterministic. 215 StringRef objBuf; 216 if (files[i]) 217 objBuf = files[i]->getBuffer(); 218 else 219 objBuf = buf[i]; 220 if (objBuf.empty()) 221 continue; 222 223 if (config->saveTemps) 224 saveBuffer(buf[i], ltoObjName); 225 ret.push_back(make<ObjFile>(MemoryBufferRef(objBuf, ltoObjName))); 226 } 227 228 return ret; 229 } 230