2 * Copyright 2012, 2013 Advanced Micro Devices, Inc.
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:
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
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
23 * Authors: Tom Stellard <thomas.stellard@amd.com>
27 #include "radeon_llvm_util.h"
28 #include "util/u_memory.h"
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>
35 LLVMModuleRef
radeon_llvm_parse_bitcode(const unsigned char * bitcode
,
38 LLVMMemoryBufferRef buf
;
39 LLVMContextRef ctx
= LLVMContextCreate();
42 buf
= LLVMCreateMemoryBufferWithMemoryRangeCopy((const char*)bitcode
,
43 bitcode_len
, "radeon");
44 LLVMParseBitcodeInContext(ctx
, buf
, &module
, NULL
);
45 LLVMDisposeMemoryBuffer(buf
);
49 unsigned radeon_llvm_get_num_kernels(const unsigned char *bitcode
,
52 LLVMModuleRef mod
= radeon_llvm_parse_bitcode(bitcode
, bitcode_len
);
53 return LLVMGetNamedMetadataNumOperands(mod
, "opencl.kernels");
56 static void radeon_llvm_optimize(LLVMModuleRef mod
)
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
);
64 LLVMPassManagerBuilderUseInlinerWithThreshold(builder
, 1000000000);
65 LLVMPassManagerBuilderPopulateModulePassManager(builder
, pass_manager
);
67 LLVMRunPassManager(pass_manager
, mod
);
68 LLVMPassManagerBuilderDispose(builder
);
69 LLVMDisposePassManager(pass_manager
);
72 LLVMModuleRef
radeon_llvm_get_kernel_module(unsigned index
,
73 const unsigned char *bitcode
, unsigned bitcode_len
)
77 LLVMValueRef
*kernel_metadata
;
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
;
89 kernel_signature
= kernel_metadata
[i
];
90 LLVMGetMDNodeOperands(kernel_signature
, &kernel_function
);
91 LLVMDeleteFunction(kernel_function
);
93 FREE(kernel_metadata
);
94 radeon_llvm_optimize(mod
);