1 // 2 // Copyright 2012-2016 Francisco Jerez 3 // Copyright 2012-2016 Advanced Micro Devices, Inc. 4 // 5 // Permission is hereby granted, free of charge, to any person obtaining a 6 // copy of this software and associated documentation files (the "Software"), 7 // to deal in the Software without restriction, including without limitation 8 // the rights to use, copy, modify, merge, publish, distribute, sublicense, 9 // and/or sell copies of the Software, and to permit persons to whom the 10 // Software is furnished to do so, subject to the following conditions: 11 // 12 // The above copyright notice and this permission notice shall be included in 13 // all copies or substantial portions of the Software. 14 // 15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 18 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR 19 // OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 20 // ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 21 // OTHER DEALINGS IN THE SOFTWARE. 22 // 23 24 /// 25 /// \file 26 /// Trivial codegen back-end that simply passes through the existing LLVM IR 27 /// and either formats it so it can be consumed by pipe drivers (if 28 /// build_module_bitcode() is used) or serializes so it can be deserialized at 29 /// a later point and passed to the actual codegen back-end (if 30 /// build_module_library() / parse_module_library() is used), potentially 31 /// after linking against other bitcode object files. 32 /// 33 34 #include "llvm/codegen.hpp" 35 #include "llvm/compat.hpp" 36 #include "llvm/metadata.hpp" 37 #include "core/error.hpp" 38 #include "util/algorithm.hpp" 39 40 #include <map> 41 #if HAVE_LLVM < 0x0400 42 #include <llvm/Bitcode/ReaderWriter.h> 43 #else 44 #include <llvm/Bitcode/BitcodeReader.h> 45 #include <llvm/Bitcode/BitcodeWriter.h> 46 #endif 47 #include <llvm/Support/raw_ostream.h> 48 49 using namespace clover; 50 using namespace clover::llvm; 51 52 namespace { 53 std::map<std::string, unsigned> 54 get_symbol_offsets(const ::llvm::Module &mod) { 55 std::map<std::string, unsigned> offsets; 56 unsigned i = 0; 57 58 for (const auto &name : map(std::mem_fn(&::llvm::Function::getName), 59 get_kernels(mod))) 60 offsets[name] = i++; 61 62 return offsets; 63 } 64 65 std::vector<char> 66 emit_code(const ::llvm::Module &mod) { 67 ::llvm::SmallVector<char, 1024> data; 68 ::llvm::raw_svector_ostream os { data }; 69 WriteBitcodeToFile(&mod, os); 70 return { os.str().begin(), os.str().end() }; 71 } 72 } 73 74 module 75 clover::llvm::build_module_bitcode(const ::llvm::Module &mod, 76 const clang::CompilerInstance &c) { 77 return build_module_common(mod, emit_code(mod), get_symbol_offsets(mod), c); 78 } 79 80 std::string 81 clover::llvm::print_module_bitcode(const ::llvm::Module &mod) { 82 std::string s; 83 ::llvm::raw_string_ostream os { s }; 84 mod.print(os, NULL); 85 return os.str(); 86 } 87 88 module 89 clover::llvm::build_module_library(const ::llvm::Module &mod, 90 enum module::section::type section_type) { 91 module m; 92 const auto code = emit_code(mod); 93 m.secs.emplace_back(0, section_type, code.size(), code); 94 return m; 95 } 96 97 std::unique_ptr<::llvm::Module> 98 clover::llvm::parse_module_library(const module &m, ::llvm::LLVMContext &ctx, 99 std::string &r_log) { 100 auto mod = ::llvm::parseBitcodeFile(::llvm::MemoryBufferRef( 101 as_string(m.secs[0].data), " "), ctx); 102 103 compat::handle_module_error(mod, [&](const std::string &s) { 104 fail(r_log, error(CL_INVALID_PROGRAM), s); 105 }); 106 107 return std::unique_ptr<::llvm::Module>(std::move(*mod)); 108 } 109
