radv_llvm_helper.cpp revision ed98bd31 
1/*
2 * Copyright © 2018 Red Hat.
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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21 * IN THE SOFTWARE.
22 */
23#include "ac_llvm_util.h"
24#include "ac_llvm_build.h"
25#include "radv_shader_helper.h"
26
27#include <list>
28class radv_llvm_per_thread_info {
29public:
30	radv_llvm_per_thread_info(enum radeon_family arg_family,
31				enum ac_target_machine_options arg_tm_options)
32		: family(arg_family), tm_options(arg_tm_options), passes(NULL) {}
33
34	~radv_llvm_per_thread_info()
35	{
36		ac_destroy_llvm_passes(passes);
37		ac_destroy_llvm_compiler(&llvm_info);
38	}
39
40	bool init(void)
41	{
42		if (!ac_init_llvm_compiler(&llvm_info,
43					  family,
44					  tm_options))
45			return false;
46
47		passes = ac_create_llvm_passes(llvm_info.tm);
48		if (!passes)
49			return false;
50
51		return true;
52	}
53
54	bool compile_to_memory_buffer(LLVMModuleRef module,
55				      struct ac_shader_binary *binary)
56	{
57		return ac_compile_module_to_binary(passes, module, binary);
58	}
59
60	bool is_same(enum radeon_family arg_family,
61		     enum ac_target_machine_options arg_tm_options) {
62		if (arg_family == family &&
63		    arg_tm_options == tm_options)
64			return true;
65		return false;
66	}
67	struct ac_llvm_compiler llvm_info;
68private:
69	enum radeon_family family;
70	enum ac_target_machine_options tm_options;
71	struct ac_compiler_passes *passes;
72};
73
74/* we have to store a linked list per thread due to the possiblity of multiple gpus being required */
75static thread_local std::list<radv_llvm_per_thread_info> radv_llvm_per_thread_list;
76
77bool radv_compile_to_binary(struct ac_llvm_compiler *info,
78			    LLVMModuleRef module,
79			    struct ac_shader_binary *binary)
80{
81	radv_llvm_per_thread_info *thread_info = nullptr;
82
83	for (auto &I : radv_llvm_per_thread_list) {
84		if (I.llvm_info.tm == info->tm) {
85			thread_info = &I;
86			break;
87		}
88	}
89
90	if (!thread_info) {
91		struct ac_compiler_passes *passes = ac_create_llvm_passes(info->tm);
92		bool ret = ac_compile_module_to_binary(passes, module, binary);
93		ac_destroy_llvm_passes(passes);
94		return ret;
95	}
96
97	return thread_info->compile_to_memory_buffer(module, binary);
98}
99
100bool radv_init_llvm_compiler(struct ac_llvm_compiler *info,
101			     bool thread_compiler,
102			     enum radeon_family family,
103			     enum ac_target_machine_options tm_options)
104{
105	if (thread_compiler) {
106		for (auto &I : radv_llvm_per_thread_list) {
107			if (I.is_same(family, tm_options)) {
108				*info = I.llvm_info;
109				return true;
110			}
111		}
112
113		radv_llvm_per_thread_list.emplace_back(family, tm_options);
114		radv_llvm_per_thread_info &tinfo = radv_llvm_per_thread_list.back();
115
116		if (!tinfo.init()) {
117			radv_llvm_per_thread_list.pop_back();
118			return false;
119		}
120
121		*info = tinfo.llvm_info;
122		return true;
123	}
124
125	if (!ac_init_llvm_compiler(info,
126				   family,
127				   tm_options))
128		return false;
129	return true;
130}
131
132void radv_destroy_llvm_compiler(struct ac_llvm_compiler *info,
133				bool thread_compiler)
134{
135	if (!thread_compiler)
136		ac_destroy_llvm_compiler(info);
137}
138