libclc/utils/prepare-builtins.cpp

   1 #if HAVE_LLVM > 0x0390
   2 #include "llvm/Bitcode/BitcodeReader.h"
   3 #include "llvm/Bitcode/BitcodeWriter.h"
   4 #else
   5 #include "llvm/Bitcode/ReaderWriter.h"
   6 #endif
   7
   8 #include "llvm/IR/Function.h"
   9 #include "llvm/IR/GlobalVariable.h"
  10 #include "llvm/IR/LLVMContext.h"
  11 #include "llvm/IR/Module.h"
  12 #include "llvm/Support/CommandLine.h"
  13 #include "llvm/Support/ManagedStatic.h"
  14 #include "llvm/Support/MemoryBuffer.h"
  15 #include "llvm/Support/FileSystem.h"
  16 #include "llvm/Support/raw_ostream.h"
  17 #include "llvm/Support/ErrorOr.h"
  18 #include "llvm/Support/ToolOutputFile.h"
  19 #include "llvm/Config/llvm-config.h"
  20
  21 #include <system_error>
  22
  23 using namespace llvm;
  24
  25 static cl::opt<std::string>
  26 InputFilename(cl::Positional, cl::desc("<input bitcode>"), cl::init("-"));
  27
  28 static cl::opt<std::string>
  29 OutputFilename("o", cl::desc("Output filename"),
  30                cl::value_desc("filename"));
  31
  32 int main(int argc, char **argv) {
  33   LLVMContext Context;
  34   llvm_shutdown_obj Y;  // Call llvm_shutdown() on exit.
  35
  36   cl::ParseCommandLineOptions(argc, argv, "libclc builtin preparation tool\n");
  37
  38   std::string ErrorMessage;
  39   Module *M = nullptr;
  40
  41   {
  42     ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
  43       MemoryBuffer::getFile(InputFilename);
  44     if (std::error_code  ec = BufferOrErr.getError()) {
  45       ErrorMessage = ec.message();
  46     } else {
  47       std::unique_ptr<MemoryBuffer> &BufferPtr = BufferOrErr.get();
  48       ErrorOr<std::unique_ptr<Module>> ModuleOrErr =
  49 #if HAVE_LLVM > 0x0390
  50           expectedToErrorOrAndEmitErrors(Context,
  51           parseBitcodeFile(BufferPtr.get()->getMemBufferRef(), Context));
  52 #else
  53           parseBitcodeFile(BufferPtr.get()->getMemBufferRef(), Context);
  54 #endif
  55       if (std::error_code ec = ModuleOrErr.getError())
  56         ErrorMessage = ec.message();
  57
  58       M = ModuleOrErr.get().release();
  59     }
  60   }
  61
  62   if (!M) {
  63     errs() << argv[0] << ": ";
  64     if (ErrorMessage.size())
  65       errs() << ErrorMessage << "\n";
  66     else
  67       errs() << "bitcode didn't read correctly.\n";
  68     return 1;
  69   }
  70
  71   // Strip the OpenCL version metadata. There are a lot of linked
  72   // modules in the library build, each spamming the same
  73   // version. This may also report a different version than the user
  74   // program is using. This should probably be uniqued when linking.
  75   if (NamedMDNode *OCLVersion = M->getNamedMetadata("opencl.ocl.version"))
  76       M->eraseNamedMetadata(OCLVersion);
  77
  78   // Set linkage of every external definition to linkonce_odr.
  79   for (Module::iterator i = M->begin(), e = M->end(); i != e; ++i) {
  80     if (!i->isDeclaration() && i->getLinkage() == GlobalValue::ExternalLinkage)
  81       i->setLinkage(GlobalValue::LinkOnceODRLinkage);
  82   }
  83
  84   for (Module::global_iterator i = M->global_begin(), e = M->global_end();
  85        i != e; ++i) {
  86     if (!i->isDeclaration() && i->getLinkage() == GlobalValue::ExternalLinkage)
  87       i->setLinkage(GlobalValue::LinkOnceODRLinkage);
  88   }
  89
  90   if (OutputFilename.empty()) {
  91     errs() << "no output file\n";
  92     return 1;
  93   }
  94
  95   std::error_code EC;
  96 #if HAVE_LLVM >= 0x0600
  97   std::unique_ptr<ToolOutputFile> Out(
  98       new ToolOutputFile(OutputFilename, EC, sys::fs::OF_None));
  99 #else
 100   std::unique_ptr<tool_output_file> Out(
 101       new tool_output_file(OutputFilename, EC, sys::fs::OF_None));
 102 #endif
 103   if (EC) {
 104     errs() << EC.message() << '\n';
 105     exit(1);
 106   }
 107
 108 #if HAVE_LLVM >= 0x0700
 109   WriteBitcodeToFile(*M, Out->os());
 110 #else
 111   WriteBitcodeToFile(M, Out->os());
 112 #endif
 113
 114   // Declare success.
 115   Out->keep();
 116   return 0;
 117 }
 118