diff options
author | Peter Collingbourne <peter@pcc.me.uk> | 2013-09-19 22:15:52 +0000 |
---|---|---|
committer | Peter Collingbourne <peter@pcc.me.uk> | 2013-09-19 22:15:52 +0000 |
commit | 88fae0edcce84920ba7c5685c36f6bd6cfb9b86d (patch) | |
tree | c1956347a81b84b10b8310d7dbc7e28443d33c05 /tools/llvm-lto/llvm-lto.cpp | |
parent | 617305a0eb086f34a1005c8be2af3eea4a2a715d (diff) | |
download | llvm-88fae0edcce84920ba7c5685c36f6bd6cfb9b86d.tar.gz llvm-88fae0edcce84920ba7c5685c36f6bd6cfb9b86d.tar.bz2 llvm-88fae0edcce84920ba7c5685c36f6bd6cfb9b86d.tar.xz |
Fix LTO handling of module-level assembly (PR14152).
Patch by Tom Roeder!
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@191042 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'tools/llvm-lto/llvm-lto.cpp')
-rw-r--r-- | tools/llvm-lto/llvm-lto.cpp | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/tools/llvm-lto/llvm-lto.cpp b/tools/llvm-lto/llvm-lto.cpp new file mode 100644 index 0000000000..f25037c37a --- /dev/null +++ b/tools/llvm-lto/llvm-lto.cpp @@ -0,0 +1,103 @@ +//===-- llvm-lto: a simple command-line program to link modules with LTO --===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// This program takes in a list of bitcode files, links them, performs link-time +// optimization, and outputs an object file. +// +//===----------------------------------------------------------------------===// + +#include "llvm-c/lto.h" +#include "llvm/Support/CommandLine.h" +#include "llvm/Support/ManagedStatic.h" +#include "llvm/Support/PrettyStackTrace.h" +#include "llvm/Support/Signals.h" +#include "llvm/Support/raw_ostream.h" + +using namespace llvm; + +static cl::list<std::string> InputFilenames(cl::Positional, cl::OneOrMore, + cl::desc("<input bitcode files>")); + +static cl::opt<std::string> OutputFilename("o", + cl::desc("Override output filename"), + cl::init(""), + cl::value_desc("filename")); + +int main(int argc, char **argv) { + // Print a stack trace if we signal out. + sys::PrintStackTraceOnErrorSignal(); + PrettyStackTraceProgram X(argc, argv); + + llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. + cl::ParseCommandLineOptions(argc, argv, "llvm LTO linker\n"); + + unsigned BaseArg = 0; + std::string ErrorMessage; + + lto_code_gen_t code_gen = lto_codegen_create(); + if (code_gen == NULL) + errs() << argv[0] << ": error creating a code generation module: " + << lto_get_error_message() << "\n"; + + lto_codegen_set_pic_model(code_gen, LTO_CODEGEN_PIC_MODEL_DYNAMIC); + lto_codegen_set_debug_model(code_gen, LTO_DEBUG_MODEL_DWARF); + + for (unsigned i = BaseArg; i < InputFilenames.size(); ++i) { + lto_module_t BitcodeModule = lto_module_create(InputFilenames[i].c_str()); + if (BitcodeModule == NULL) { + errs() << argv[0] << ": error loading file '" << InputFilenames[i] + << "': " << lto_get_error_message() << "\n"; + return 1; + } + + if (lto_codegen_add_module(code_gen, BitcodeModule)) { + errs() << argv[0] << ": error adding file '" << InputFilenames[i] + << "': " << lto_get_error_message() << "\n"; + lto_module_dispose(BitcodeModule); + return 1; + } + + lto_module_dispose(BitcodeModule); + } + + if (!OutputFilename.empty()) { + size_t len = 0; + const void *Code = lto_codegen_compile(code_gen, &len); + if (Code == NULL) { + errs() << argv[0] + << ": error compiling the code: " << lto_get_error_message() + << "\n"; + return 1; + } + + std::string ErrorInfo; + raw_fd_ostream FileStream(OutputFilename.c_str(), ErrorInfo); + if (!ErrorInfo.empty()) { + errs() << argv[0] << ": error opening the file '" << OutputFilename + << "': " << ErrorInfo << "\n"; + return 1; + } + + FileStream.write(reinterpret_cast<const char *>(Code), len); + } else { + const char *OutputName = NULL; + if (lto_codegen_compile_to_file(code_gen, &OutputName)) { + errs() << argv[0] + << ": error compiling the code: " << lto_get_error_message() + << "\n"; + return 1; + } + + outs() << "Wrote native object file '" << OutputName << "'\n"; + } + + lto_codegen_dispose(code_gen); + + return 0; +} |