7ef7e6aacd
This should be uniqued when linking, but right now it creates a lot of metadata spam listing the same version. This should also probably be reporting the compiled version of the user program, which may differ from the library. Currently the library IR files report 1.0 while 1.1/1.2 are the default for user programs. llvm-svn: 279692
99 lines
2.9 KiB
C++
99 lines
2.9 KiB
C++
#include "llvm/Bitcode/ReaderWriter.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/ErrorOr.h"
|
|
#include "llvm/Support/ToolOutputFile.h"
|
|
#include "llvm/Config/llvm-config.h"
|
|
|
|
#include <system_error>
|
|
|
|
using namespace llvm;
|
|
|
|
static cl::opt<std::string>
|
|
InputFilename(cl::Positional, cl::desc("<input bitcode>"), cl::init("-"));
|
|
|
|
static cl::opt<std::string>
|
|
OutputFilename("o", cl::desc("Output filename"),
|
|
cl::value_desc("filename"));
|
|
|
|
int main(int argc, char **argv) {
|
|
LLVMContext Context;
|
|
llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
|
|
|
|
cl::ParseCommandLineOptions(argc, argv, "libclc builtin preparation tool\n");
|
|
|
|
std::string ErrorMessage;
|
|
Module *M = nullptr;
|
|
|
|
{
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
|
|
MemoryBuffer::getFile(InputFilename);
|
|
std::unique_ptr<MemoryBuffer> &BufferPtr = BufferOrErr.get();
|
|
if (std::error_code ec = BufferOrErr.getError())
|
|
ErrorMessage = ec.message();
|
|
else {
|
|
ErrorOr<std::unique_ptr<Module>> ModuleOrErr =
|
|
parseBitcodeFile(BufferPtr.get()->getMemBufferRef(), Context);
|
|
if (std::error_code ec = ModuleOrErr.getError())
|
|
ErrorMessage = ec.message();
|
|
|
|
M = ModuleOrErr.get().release();
|
|
}
|
|
}
|
|
|
|
if (!M) {
|
|
errs() << argv[0] << ": ";
|
|
if (ErrorMessage.size())
|
|
errs() << ErrorMessage << "\n";
|
|
else
|
|
errs() << "bitcode didn't read correctly.\n";
|
|
return 1;
|
|
}
|
|
|
|
// Strip the OpenCL version metadata. There are a lot of linked
|
|
// modules in the library build, each spamming the same
|
|
// version. This may also report a different version than the user
|
|
// program is using. This should probably be uniqued when linking.
|
|
if (NamedMDNode *OCLVersion = M->getNamedMetadata("opencl.ocl.version"))
|
|
M->eraseNamedMetadata(OCLVersion);
|
|
|
|
// Set linkage of every external definition to linkonce_odr.
|
|
for (Module::iterator i = M->begin(), e = M->end(); i != e; ++i) {
|
|
if (!i->isDeclaration() && i->getLinkage() == GlobalValue::ExternalLinkage)
|
|
i->setLinkage(GlobalValue::LinkOnceODRLinkage);
|
|
}
|
|
|
|
for (Module::global_iterator i = M->global_begin(), e = M->global_end();
|
|
i != e; ++i) {
|
|
if (!i->isDeclaration() && i->getLinkage() == GlobalValue::ExternalLinkage)
|
|
i->setLinkage(GlobalValue::LinkOnceODRLinkage);
|
|
}
|
|
|
|
if (OutputFilename.empty()) {
|
|
errs() << "no output file\n";
|
|
return 1;
|
|
}
|
|
|
|
std::error_code EC;
|
|
std::unique_ptr<tool_output_file> Out
|
|
(new tool_output_file(OutputFilename, EC, sys::fs::F_None));
|
|
if (EC) {
|
|
errs() << EC.message() << '\n';
|
|
exit(1);
|
|
}
|
|
|
|
WriteBitcodeToFile(M, Out->os());
|
|
|
|
// Declare success.
|
|
Out->keep();
|
|
return 0;
|
|
}
|
|
|