src/gallium/state_trackers/clover/llvm/codegen/bitcode.cpp

   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 #include <llvm/Config/llvm-config.h>
  42 #if LLVM_VERSION_MAJOR < 4
  43 #include <llvm/Bitcode/ReaderWriter.h>
  44 #else
  45 #include <llvm/Bitcode/BitcodeReader.h>
  46 #include <llvm/Bitcode/BitcodeWriter.h>
  47 #endif
  48 #include <llvm/Support/raw_ostream.h>
  49
  50 using namespace clover;
  51 using namespace clover::llvm;
  52
  53 namespace {
  54    std::vector<char>
  55    emit_code(const ::llvm::Module &mod) {
  56       ::llvm::SmallVector<char, 1024> data;
  57       ::llvm::raw_svector_ostream os { data };
  58       compat::write_bitcode_to_file(mod, os);
  59       return { os.str().begin(), os.str().end() };
  60    }
  61 }
  62
  63 std::string
  64 clover::llvm::print_module_bitcode(const ::llvm::Module &mod) {
  65    std::string s;
  66    ::llvm::raw_string_ostream os { s };
  67    mod.print(os, NULL);
  68    return os.str();
  69 }
  70
  71 module
  72 clover::llvm::build_module_library(const ::llvm::Module &mod,
  73                                    enum module::section::type section_type) {
  74    module m;
  75    const auto code = emit_code(mod);
  76    m.secs.emplace_back(0, section_type, code.size(), code);
  77    return m;
  78 }
  79
  80 std::unique_ptr< ::llvm::Module>
  81 clover::llvm::parse_module_library(const module &m, ::llvm::LLVMContext &ctx,
  82                                    std::string &r_log) {
  83    auto mod = ::llvm::parseBitcodeFile(::llvm::MemoryBufferRef(
  84                                         as_string(m.secs[0].data), " "), ctx);
  85
  86    compat::handle_module_error(mod, [&](const std::string &s) {
  87          fail(r_log, error(CL_INVALID_PROGRAM), s);
  88       });
  89
  90    return std::unique_ptr< ::llvm::Module>(std::move(*mod));
  91 }