|
15016
|
1 /* |
|
|
2 |
|
|
3 Copyright (C) 2012 Max Brister <max@2bass.com> |
|
|
4 |
|
|
5 This file is part of Octave. |
|
|
6 |
|
|
7 Octave is free software; you can redistribute it and/or modify it |
|
|
8 under the terms of the GNU General Public License as published by the |
|
|
9 Free Software Foundation; either version 3 of the License, or (at your |
|
|
10 option) any later version. |
|
|
11 |
|
|
12 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
|
13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
|
14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
|
15 for more details. |
|
|
16 |
|
|
17 You should have received a copy of the GNU General Public License |
|
|
18 along with Octave; see the file COPYING. If not, see |
|
|
19 <http://www.gnu.org/licenses/>. |
|
|
20 |
|
|
21 */ |
|
|
22 |
|
|
23 // Some utility classes and functions used throughout jit |
|
|
24 |
|
|
25 #if !defined (octave_jit_util_h) |
|
|
26 #define octave_jit_util_h 1 |
|
|
27 |
|
|
28 #ifdef HAVE_LLVM |
|
|
29 |
|
|
30 #include <stdexcept> |
|
|
31 |
|
|
32 // we don't want to include llvm headers here, as they require |
|
|
33 // __STDC_LIMIT_MACROS and __STDC_CONSTANT_MACROS be defined in the entire |
|
|
34 // compilation unit |
|
|
35 namespace llvm |
|
|
36 { |
|
|
37 class Value; |
|
|
38 class Module; |
|
|
39 class FunctionPassManager; |
|
|
40 class PassManager; |
|
|
41 class ExecutionEngine; |
|
|
42 class Function; |
|
|
43 class BasicBlock; |
|
|
44 class LLVMContext; |
|
|
45 class Type; |
|
|
46 class StructType; |
|
|
47 class Twine; |
|
|
48 class GlobalVariable; |
|
|
49 class TerminatorInst; |
|
|
50 class PHINode; |
|
|
51 |
|
|
52 class ConstantFolder; |
|
|
53 |
|
|
54 template <bool preserveNames> |
|
|
55 class IRBuilderDefaultInserter; |
|
|
56 |
|
|
57 template <bool preserveNames, typename T, typename Inserter> |
|
|
58 class IRBuilder; |
|
|
59 |
|
|
60 typedef IRBuilder<true, ConstantFolder, IRBuilderDefaultInserter<true> > |
|
|
61 IRBuilderD; |
|
|
62 } |
|
|
63 |
|
|
64 class octave_base_value; |
|
|
65 class octave_builtin; |
|
|
66 class octave_value; |
|
|
67 class tree; |
|
|
68 class tree_expression; |
|
|
69 |
|
|
70 // thrown when we should give up on JIT and interpret |
|
|
71 class jit_fail_exception : public std::runtime_error |
|
|
72 { |
|
|
73 public: |
|
|
74 jit_fail_exception (void) : std::runtime_error ("unknown"), mknown (false) {} |
|
|
75 jit_fail_exception (const std::string& reason) : std::runtime_error (reason), |
|
|
76 mknown (true) |
|
|
77 {} |
|
|
78 |
|
|
79 bool known (void) const { return mknown; } |
|
|
80 private: |
|
|
81 bool mknown; |
|
|
82 }; |
|
|
83 |
|
|
84 // llvm doesn't provide this, and it's really useful for debugging |
|
|
85 std::ostream& operator<< (std::ostream& os, const llvm::Value& v); |
|
|
86 |
|
|
87 template <typename HOLDER_T, typename SUB_T> |
|
|
88 class jit_internal_node; |
|
|
89 |
|
|
90 // jit_internal_list and jit_internal_node implement generic embedded doubly |
|
|
91 // linked lists. List items extend from jit_internal_list, and can be placed |
|
|
92 // in nodes of type jit_internal_node. We use CRTP twice. |
|
|
93 template <typename LIST_T, typename NODE_T> |
|
|
94 class |
|
|
95 jit_internal_list |
|
|
96 { |
|
|
97 friend class jit_internal_node<LIST_T, NODE_T>; |
|
|
98 public: |
|
|
99 jit_internal_list (void) : use_head (0), use_tail (0), muse_count (0) {} |
|
|
100 |
|
|
101 virtual ~jit_internal_list (void) |
|
|
102 { |
|
|
103 while (use_head) |
|
|
104 use_head->stash_value (0); |
|
|
105 } |
|
|
106 |
|
|
107 NODE_T *first_use (void) const { return use_head; } |
|
|
108 |
|
|
109 size_t use_count (void) const { return muse_count; } |
|
|
110 private: |
|
|
111 NODE_T *use_head; |
|
|
112 NODE_T *use_tail; |
|
|
113 size_t muse_count; |
|
|
114 }; |
|
|
115 |
|
|
116 // a node for internal linked lists |
|
|
117 template <typename LIST_T, typename NODE_T> |
|
|
118 class |
|
|
119 jit_internal_node |
|
|
120 { |
|
|
121 public: |
|
|
122 typedef jit_internal_list<LIST_T, NODE_T> jit_ilist; |
|
|
123 |
|
|
124 jit_internal_node (void) : mvalue (0), mnext (0), mprev (0) {} |
|
|
125 |
|
|
126 ~jit_internal_node (void) { remove (); } |
|
|
127 |
|
|
128 LIST_T *value (void) const { return mvalue; } |
|
|
129 |
|
|
130 void stash_value (LIST_T *avalue) |
|
|
131 { |
|
|
132 remove (); |
|
|
133 |
|
|
134 mvalue = avalue; |
|
|
135 |
|
|
136 if (mvalue) |
|
|
137 { |
|
|
138 jit_ilist *ilist = mvalue; |
|
|
139 NODE_T *sthis = static_cast<NODE_T *> (this); |
|
|
140 if (ilist->use_head) |
|
|
141 { |
|
|
142 ilist->use_tail->mnext = sthis; |
|
|
143 mprev = ilist->use_tail; |
|
|
144 } |
|
|
145 else |
|
|
146 ilist->use_head = sthis; |
|
|
147 |
|
|
148 ilist->use_tail = sthis; |
|
|
149 ++ilist->muse_count; |
|
|
150 } |
|
|
151 } |
|
|
152 |
|
|
153 NODE_T *next (void) const { return mnext; } |
|
|
154 |
|
|
155 NODE_T *prev (void) const { return mprev; } |
|
|
156 private: |
|
|
157 void remove () |
|
|
158 { |
|
|
159 if (mvalue) |
|
|
160 { |
|
|
161 jit_ilist *ilist = mvalue; |
|
|
162 if (mprev) |
|
|
163 mprev->mnext = mnext; |
|
|
164 else |
|
|
165 // we are the use_head |
|
|
166 ilist->use_head = mnext; |
|
|
167 |
|
|
168 if (mnext) |
|
|
169 mnext->mprev = mprev; |
|
|
170 else |
|
|
171 // we are the use tail |
|
|
172 ilist->use_tail = mprev; |
|
|
173 |
|
|
174 mnext = mprev = 0; |
|
|
175 --ilist->muse_count; |
|
|
176 mvalue = 0; |
|
|
177 } |
|
|
178 } |
|
|
179 |
|
|
180 LIST_T *mvalue; |
|
|
181 NODE_T *mnext; |
|
|
182 NODE_T *mprev; |
|
|
183 }; |
|
|
184 |
|
|
185 // Use like: isa<jit_phi> (value) |
|
|
186 // basically just a short cut type typing dyanmic_cast. |
|
|
187 template <typename T, typename U> |
|
|
188 bool isa (U *value) |
|
|
189 { |
|
|
190 return dynamic_cast<T *> (value); |
|
|
191 } |
|
|
192 |
|
|
193 #define JIT_ASSIGN_ARG(i) the_args[i] = arg ## i; |
|
|
194 #define JIT_EXPAND(ret, fname, type, isconst, N) \ |
|
|
195 ret fname (JIT_PARAM_ARGS OCT_MAKE_DECL_LIST (type, arg, N)) isconst \ |
|
|
196 { \ |
|
|
197 std::vector<type> the_args (N); \ |
|
|
198 OCT_ITERATE_MACRO (JIT_ASSIGN_ARG, N); \ |
|
|
199 return fname (JIT_PARAMS the_args); \ |
|
|
200 } |
|
|
201 |
|
|
202 #endif |
|
|
203 #endif |
