src/gallium/drivers/radeon/radeon_llvm_util.c

   1 /*
   2  * Copyright 2012, 2013 Advanced Micro Devices, Inc.
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * 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 OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  *
  23  * Authors: Tom Stellard <thomas.stellard@amd.com>
  24  *
  25  */
  26
  27 #include "radeon_llvm_util.h"
  28 #include "util/u_memory.h"
  29
  30 #include <llvm-c/BitReader.h>
  31 #include <llvm-c/Core.h>
  32 #include <llvm-c/Target.h>
  33 #include <llvm-c/Transforms/PassManagerBuilder.h>
  34
  35 LLVMModuleRef radeon_llvm_parse_bitcode(const unsigned char * bitcode,
  36                                                         unsigned bitcode_len)
  37 {
  38         LLVMMemoryBufferRef buf;
  39         LLVMContextRef ctx = LLVMContextCreate();
  40         LLVMModuleRef module;
  41
  42         buf = LLVMCreateMemoryBufferWithMemoryRangeCopy((const char*)bitcode,
  43                                                         bitcode_len, "radeon");
  44         LLVMParseBitcodeInContext(ctx, buf, &module, NULL);
  45         LLVMDisposeMemoryBuffer(buf);
  46         return module;
  47 }
  48
  49 unsigned radeon_llvm_get_num_kernels(const unsigned char *bitcode,
  50                                 unsigned bitcode_len)
  51 {
  52         LLVMModuleRef mod = radeon_llvm_parse_bitcode(bitcode, bitcode_len);
  53         return LLVMGetNamedMetadataNumOperands(mod, "opencl.kernels");
  54 }
  55
  56 static void radeon_llvm_optimize(LLVMModuleRef mod)
  57 {
  58         const char *data_layout = LLVMGetDataLayout(mod);
  59         LLVMTargetDataRef TD = LLVMCreateTargetData(data_layout);
  60         LLVMPassManagerBuilderRef builder = LLVMPassManagerBuilderCreate();
  61         LLVMPassManagerRef pass_manager = LLVMCreatePassManager();
  62         LLVMAddTargetData(TD, pass_manager);
  63
  64         LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 1000000000);
  65         LLVMPassManagerBuilderPopulateModulePassManager(builder, pass_manager);
  66
  67         LLVMRunPassManager(pass_manager, mod);
  68         LLVMPassManagerBuilderDispose(builder);
  69         LLVMDisposePassManager(pass_manager);
  70 }
  71
  72 LLVMModuleRef radeon_llvm_get_kernel_module(unsigned index,
  73                 const unsigned char *bitcode, unsigned bitcode_len)
  74 {
  75         LLVMModuleRef mod;
  76         unsigned num_kernels;
  77         LLVMValueRef *kernel_metadata;
  78         unsigned i;
  79
  80         mod = radeon_llvm_parse_bitcode(bitcode, bitcode_len);
  81         num_kernels = LLVMGetNamedMetadataNumOperands(mod, "opencl.kernels");
  82         kernel_metadata = MALLOC(num_kernels * sizeof(LLVMValueRef));
  83         LLVMGetNamedMetadataOperands(mod, "opencl.kernels", kernel_metadata);
  84         for (i = 0; i < num_kernels; i++) {
  85                 LLVMValueRef kernel_signature, kernel_function;
  86                 if (i == index) {
  87                         continue;
  88                 }
  89                 kernel_signature = kernel_metadata[i];
  90                 LLVMGetMDNodeOperands(kernel_signature, &kernel_function);
  91                 LLVMDeleteFunction(kernel_function);
  92         }
  93         FREE(kernel_metadata);
  94         radeon_llvm_optimize(mod);
  95         return mod;
  96 }