ThreadSafeModule cloneToNewContext(ThreadSafeModule &TSM, GVPredicate ShouldCloneDef, GVModifier UpdateClonedDefSource) { assert(TSM && "Can not clone null module"); if (!ShouldCloneDef) ShouldCloneDef = [](const GlobalValue &) { return true; }; auto Lock = TSM.getContextLock(); SmallVector<char, 1> ClonedModuleBuffer; { std::set<GlobalValue *> ClonedDefsInSrc; ValueToValueMapTy VMap; auto Tmp = CloneModule(*TSM.getModule(), VMap, [&](const GlobalValue *GV) { if (ShouldCloneDef(*GV)) { ClonedDefsInSrc.insert(const_cast<GlobalValue *>(GV)); return true; } return false; }); if (UpdateClonedDefSource) for (auto *GV : ClonedDefsInSrc) UpdateClonedDefSource(*GV); BitcodeWriter BCWriter(ClonedModuleBuffer); BCWriter.writeModule(*Tmp); BCWriter.writeSymtab(); BCWriter.writeStrtab(); } MemoryBufferRef ClonedModuleBufferRef( StringRef(ClonedModuleBuffer.data(), ClonedModuleBuffer.size()), "cloned module buffer"); ThreadSafeContext NewTSCtx(llvm::make_unique<LLVMContext>()); auto ClonedModule = cantFail(parseBitcodeFile(ClonedModuleBufferRef, *NewTSCtx.getContext())); ClonedModule->setModuleIdentifier(TSM.getModule()->getName()); return ThreadSafeModule(std::move(ClonedModule), std::move(NewTSCtx)); }
// Simple helper to load a module from bitcode std::unique_ptr<Module> loadModuleFromBuffer(const MemoryBufferRef &Buffer, LLVMContext &Context, bool Lazy) { SMDiagnostic Err; ErrorOr<std::unique_ptr<Module>> ModuleOrErr(nullptr); if (Lazy) { ModuleOrErr = getLazyBitcodeModule(MemoryBuffer::getMemBuffer(Buffer, false), Context, /* ShouldLazyLoadMetadata */ Lazy); } else { ModuleOrErr = parseBitcodeFile(Buffer, Context); } if (std::error_code EC = ModuleOrErr.getError()) { Err = SMDiagnostic(Buffer.getBufferIdentifier(), SourceMgr::DK_Error, EC.message()); Err.print("ThinLTO", errs()); report_fatal_error("Can't load module, abort."); } return std::move(ModuleOrErr.get()); }