2 * Copyright © 2018 Red Hat.
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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
23 #include "ac_llvm_util.h"
24 #include "ac_llvm_build.h"
25 #include "radv_shader_helper.h"
28 class radv_llvm_per_thread_info
{
30 radv_llvm_per_thread_info(enum radeon_family arg_family
,
31 enum ac_target_machine_options arg_tm_options
,
32 unsigned arg_wave_size
)
33 : family(arg_family
), tm_options(arg_tm_options
),
34 wave_size(arg_wave_size
), passes(NULL
), passes_wave32(NULL
) {}
36 ~radv_llvm_per_thread_info()
38 ac_destroy_llvm_compiler(&llvm_info
);
43 if (!ac_init_llvm_compiler(&llvm_info
,
48 passes
= ac_create_llvm_passes(llvm_info
.tm
);
52 if (llvm_info
.tm_wave32
) {
53 passes_wave32
= ac_create_llvm_passes(llvm_info
.tm_wave32
);
61 bool compile_to_memory_buffer(LLVMModuleRef module
,
62 char **pelf_buffer
, size_t *pelf_size
)
64 struct ac_compiler_passes
*p
= wave_size
== 32 ? passes_wave32
: passes
;
65 return ac_compile_module_to_elf(p
, module
, pelf_buffer
, pelf_size
);
68 bool is_same(enum radeon_family arg_family
,
69 enum ac_target_machine_options arg_tm_options
,
70 unsigned arg_wave_size
) {
71 if (arg_family
== family
&&
72 arg_tm_options
== tm_options
&&
73 arg_wave_size
== wave_size
)
77 struct ac_llvm_compiler llvm_info
;
79 enum radeon_family family
;
80 enum ac_target_machine_options tm_options
;
82 struct ac_compiler_passes
*passes
;
83 struct ac_compiler_passes
*passes_wave32
;
86 /* we have to store a linked list per thread due to the possiblity of multiple gpus being required */
87 static thread_local
std::list
<radv_llvm_per_thread_info
> radv_llvm_per_thread_list
;
89 bool radv_compile_to_elf(struct ac_llvm_compiler
*info
,
91 char **pelf_buffer
, size_t *pelf_size
)
93 radv_llvm_per_thread_info
*thread_info
= nullptr;
95 for (auto &I
: radv_llvm_per_thread_list
) {
96 if (I
.llvm_info
.tm
== info
->tm
) {
103 struct ac_compiler_passes
*passes
= ac_create_llvm_passes(info
->tm
);
104 bool ret
= ac_compile_module_to_elf(passes
, module
, pelf_buffer
, pelf_size
);
105 ac_destroy_llvm_passes(passes
);
109 return thread_info
->compile_to_memory_buffer(module
, pelf_buffer
, pelf_size
);
112 bool radv_init_llvm_compiler(struct ac_llvm_compiler
*info
,
113 bool thread_compiler
,
114 enum radeon_family family
,
115 enum ac_target_machine_options tm_options
,
118 if (thread_compiler
) {
119 for (auto &I
: radv_llvm_per_thread_list
) {
120 if (I
.is_same(family
, tm_options
, wave_size
)) {
126 radv_llvm_per_thread_list
.emplace_back(family
, tm_options
, wave_size
);
127 radv_llvm_per_thread_info
&tinfo
= radv_llvm_per_thread_list
.back();
130 radv_llvm_per_thread_list
.pop_back();
134 *info
= tinfo
.llvm_info
;
138 if (!ac_init_llvm_compiler(info
,
145 void radv_destroy_llvm_compiler(struct ac_llvm_compiler
*info
,
146 bool thread_compiler
)
148 if (!thread_compiler
)
149 ac_destroy_llvm_compiler(info
);