Subversion Repositories Kolibri OS

Rev

Go to most recent revision | Blame | Last modification | View Log | RSS feed

  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/Transforms/PassManagerBuilder.h>
  33.  
  34. LLVMModuleRef radeon_llvm_parse_bitcode(const unsigned char * bitcode,
  35.                                                         unsigned bitcode_len)
  36. {
  37.         LLVMMemoryBufferRef buf;
  38.         LLVMContextRef ctx = LLVMContextCreate();
  39.         LLVMModuleRef module;
  40.  
  41.         buf = LLVMCreateMemoryBufferWithMemoryRangeCopy((const char*)bitcode,
  42.                                                         bitcode_len, "radeon");
  43.         LLVMParseBitcodeInContext(ctx, buf, &module, NULL);
  44.         return module;
  45. }
  46.  
  47. unsigned radeon_llvm_get_num_kernels(const unsigned char *bitcode,
  48.                                 unsigned bitcode_len)
  49. {
  50.         LLVMModuleRef mod = radeon_llvm_parse_bitcode(bitcode, bitcode_len);
  51.         return LLVMGetNamedMetadataNumOperands(mod, "opencl.kernels");
  52. }
  53.  
  54. static void radeon_llvm_optimize(LLVMModuleRef mod)
  55. {
  56.         LLVMPassManagerBuilderRef builder = LLVMPassManagerBuilderCreate();
  57.         LLVMPassManagerRef pass_manager = LLVMCreatePassManager();
  58.  
  59.         LLVMPassManagerBuilderUseInlinerWithThreshold(builder, 1000000000);
  60.         LLVMPassManagerBuilderPopulateModulePassManager(builder, pass_manager);
  61.  
  62.         LLVMRunPassManager(pass_manager, mod);
  63.         LLVMPassManagerBuilderDispose(builder);
  64.         LLVMDisposePassManager(pass_manager);
  65. }
  66.  
  67. LLVMModuleRef radeon_llvm_get_kernel_module(unsigned index,
  68.                 const unsigned char *bitcode, unsigned bitcode_len)
  69. {
  70.         LLVMModuleRef mod;
  71.         unsigned num_kernels;
  72.         LLVMValueRef *kernel_metadata;
  73.         unsigned i;
  74.  
  75.         mod = radeon_llvm_parse_bitcode(bitcode, bitcode_len);
  76.         num_kernels = LLVMGetNamedMetadataNumOperands(mod, "opencl.kernels");
  77.         kernel_metadata = MALLOC(num_kernels * sizeof(LLVMValueRef));
  78.         LLVMGetNamedMetadataOperands(mod, "opencl.kernels", kernel_metadata);
  79.         for (i = 0; i < num_kernels; i++) {
  80.                 LLVMValueRef kernel_signature, kernel_function;
  81.                 if (i == index) {
  82.                         continue;
  83.                 }
  84.                 kernel_signature = kernel_metadata[i];
  85.                 LLVMGetMDNodeOperands(kernel_signature, &kernel_function);
  86.                 LLVMDeleteFunction(kernel_function);
  87.         }
  88.         FREE(kernel_metadata);
  89.         radeon_llvm_optimize(mod);
  90.         return mod;
  91. }
  92.