diff options
| -rw-r--r-- | include/llvm/LinkTimeOptimizer.h | 6 | ||||
| -rw-r--r-- | tools/lto/lto.cpp | 19 |
2 files changed, 24 insertions, 1 deletions
diff --git a/include/llvm/LinkTimeOptimizer.h b/include/llvm/LinkTimeOptimizer.h index d20d21e..76a7af6 100644 --- a/include/llvm/LinkTimeOptimizer.h +++ b/include/llvm/LinkTimeOptimizer.h @@ -83,6 +83,8 @@ namespace llvm { public: typedef hash_map<const char*, LLVMSymbol*, hash<const char*>, string_compare> NameToSymbolMap; + typedef hash_map<const char*, Module*, hash<const char*>, + string_compare> NameToModuleMap; enum LTOStatus readLLVMObjectFile(const std::string &InputFilename, NameToSymbolMap &symbols, @@ -92,8 +94,12 @@ namespace llvm { std::string &targetTriple); private: + Module *getModule (const std::string &InputFilename); + + private: std::vector<Module *> modules; NameToSymbolMap allSymbols; + NameToModuleMap allModules; }; } // End llvm namespace diff --git a/tools/lto/lto.cpp b/tools/lto/lto.cpp index e49364e..96b449e 100644 --- a/tools/lto/lto.cpp +++ b/tools/lto/lto.cpp @@ -99,6 +99,23 @@ findExternalRefs(Value *value, std::set<std::string> &references, findExternalRefs(c->getOperand(i), references, mangler); } +/// InputFilename is a LLVM bytecode file. If Module with InputFilename is +/// available then return it. Otherwise parseInputFilename. +Module * +LinkTimeOptimizer::getModule(const std::string &InputFilename) +{ + Module *m = NULL; + + NameToModuleMap::iterator pos = allModules.find(InputFilename.c_str()); + if (pos != allModules.end()) + m = allModules[InputFilename.c_str()]; + else { + m = ParseBytecodeFile(InputFilename); + allModules[InputFilename.c_str()] = m; + } + return m; +} + /// InputFilename is a LLVM bytecode file. Read it using bytecode reader. /// Collect global functions and symbol names in symbols vector. /// Collect external references in references vector. @@ -108,7 +125,7 @@ LinkTimeOptimizer::readLLVMObjectFile(const std::string &InputFilename, NameToSymbolMap &symbols, std::set<std::string> &references) { - Module *m = ParseBytecodeFile(InputFilename); + Module *m = getModule(InputFilename); if (!m) return LTO_READ_FAILURE; |