nir/algebraic: Replace a pattern where iand with a Boolean is used as a bcsel
[mesa.git] / src / compiler / nir / nir_opt_algebraic.py
1 #
2 # Copyright (C) 2014 Intel Corporation
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 # Authors:
24 # Jason Ekstrand (jason@jlekstrand.net)
25
26 from __future__ import print_function
27
28 from collections import OrderedDict
29 import nir_algebraic
30 from nir_opcodes import type_sizes
31 import itertools
32
33 # Convenience variables
34 a = 'a'
35 b = 'b'
36 c = 'c'
37 d = 'd'
38 e = 'e'
39
40 # Written in the form (<search>, <replace>) where <search> is an expression
41 # and <replace> is either an expression or a value. An expression is
42 # defined as a tuple of the form ([~]<op>, <src0>, <src1>, <src2>, <src3>)
43 # where each source is either an expression or a value. A value can be
44 # either a numeric constant or a string representing a variable name.
45 #
46 # If the opcode in a search expression is prefixed by a '~' character, this
47 # indicates that the operation is inexact. Such operations will only get
48 # applied to SSA values that do not have the exact bit set. This should be
49 # used by by any optimizations that are not bit-for-bit exact. It should not,
50 # however, be used for backend-requested lowering operations as those need to
51 # happen regardless of precision.
52 #
53 # Variable names are specified as "[#]name[@type][(cond)]" where "#" inicates
54 # that the given variable will only match constants and the type indicates that
55 # the given variable will only match values from ALU instructions with the
56 # given output type, and (cond) specifies an additional condition function
57 # (see nir_search_helpers.h).
58 #
59 # For constants, you have to be careful to make sure that it is the right
60 # type because python is unaware of the source and destination types of the
61 # opcodes.
62 #
63 # All expression types can have a bit-size specified. For opcodes, this
64 # looks like "op@32", for variables it is "a@32" or "a@uint32" to specify a
65 # type and size. In the search half of the expression this indicates that it
66 # should only match that particular bit-size. In the replace half of the
67 # expression this indicates that the constructed value should have that
68 # bit-size.
69
70 optimizations = [
71
72 (('imul', a, '#b@32(is_pos_power_of_two)'), ('ishl', a, ('find_lsb', b))),
73 (('imul', a, '#b@32(is_neg_power_of_two)'), ('ineg', ('ishl', a, ('find_lsb', ('iabs', b))))),
74 (('unpack_64_2x32_split_x', ('imul_2x32_64(is_used_once)', a, b)), ('imul', a, b)),
75 (('unpack_64_2x32_split_x', ('umul_2x32_64(is_used_once)', a, b)), ('imul', a, b)),
76 (('imul_2x32_64', a, b), ('pack_64_2x32_split', ('imul', a, b), ('imul_high', a, b)), 'options->lower_mul_2x32_64'),
77 (('umul_2x32_64', a, b), ('pack_64_2x32_split', ('imul', a, b), ('umul_high', a, b)), 'options->lower_mul_2x32_64'),
78 (('udiv', a, 1), a),
79 (('idiv', a, 1), a),
80 (('umod', a, 1), 0),
81 (('imod', a, 1), 0),
82 (('udiv', a, '#b@32(is_pos_power_of_two)'), ('ushr', a, ('find_lsb', b))),
83 (('idiv', a, '#b@32(is_pos_power_of_two)'), ('imul', ('isign', a), ('ushr', ('iabs', a), ('find_lsb', b))), 'options->lower_idiv'),
84 (('idiv', a, '#b@32(is_neg_power_of_two)'), ('ineg', ('imul', ('isign', a), ('ushr', ('iabs', a), ('find_lsb', ('iabs', b))))), 'options->lower_idiv'),
85 (('umod', a, '#b(is_pos_power_of_two)'), ('iand', a, ('isub', b, 1))),
86
87 (('fneg', ('fneg', a)), a),
88 (('ineg', ('ineg', a)), a),
89 (('fabs', ('fabs', a)), ('fabs', a)),
90 (('fabs', ('fneg', a)), ('fabs', a)),
91 (('fabs', ('u2f', a)), ('u2f', a)),
92 (('iabs', ('iabs', a)), ('iabs', a)),
93 (('iabs', ('ineg', a)), ('iabs', a)),
94 (('f2b', ('fneg', a)), ('f2b', a)),
95 (('i2b', ('ineg', a)), ('i2b', a)),
96 (('~fadd', a, 0.0), a),
97 (('iadd', a, 0), a),
98 (('usadd_4x8', a, 0), a),
99 (('usadd_4x8', a, ~0), ~0),
100 (('~fadd', ('fmul', a, b), ('fmul', a, c)), ('fmul', a, ('fadd', b, c))),
101 (('iadd', ('imul', a, b), ('imul', a, c)), ('imul', a, ('iadd', b, c))),
102 (('~fadd', ('fneg', a), a), 0.0),
103 (('iadd', ('ineg', a), a), 0),
104 (('iadd', ('ineg', a), ('iadd', a, b)), b),
105 (('iadd', a, ('iadd', ('ineg', a), b)), b),
106 (('~fadd', ('fneg', a), ('fadd', a, b)), b),
107 (('~fadd', a, ('fadd', ('fneg', a), b)), b),
108 (('~fmul', a, 0.0), 0.0),
109 (('imul', a, 0), 0),
110 (('umul_unorm_4x8', a, 0), 0),
111 (('umul_unorm_4x8', a, ~0), a),
112 (('fmul', a, 1.0), a),
113 (('imul', a, 1), a),
114 (('fmul', a, -1.0), ('fneg', a)),
115 (('imul', a, -1), ('ineg', a)),
116 # If a < 0: fsign(a)*a*a => -1*a*a => -a*a => abs(a)*a
117 # If a > 0: fsign(a)*a*a => 1*a*a => a*a => abs(a)*a
118 # If a == 0: fsign(a)*a*a => 0*0*0 => abs(0)*0
119 (('fmul', ('fsign', a), ('fmul', a, a)), ('fmul', ('fabs', a), a)),
120 (('fmul', ('fmul', ('fsign', a), a), a), ('fmul', ('fabs', a), a)),
121 (('~ffma', 0.0, a, b), b),
122 (('~ffma', a, 0.0, b), b),
123 (('~ffma', a, b, 0.0), ('fmul', a, b)),
124 (('ffma', a, 1.0, b), ('fadd', a, b)),
125 (('ffma', 1.0, a, b), ('fadd', a, b)),
126 (('~flrp', a, b, 0.0), a),
127 (('~flrp', a, b, 1.0), b),
128 (('~flrp', a, a, b), a),
129 (('~flrp', 0.0, a, b), ('fmul', a, b)),
130 (('~flrp', a, b, ('b2f', 'c@1')), ('bcsel', c, b, a), 'options->lower_flrp32'),
131 (('~flrp', a, 0.0, c), ('fadd', ('fmul', ('fneg', a), c), a)),
132 (('flrp@16', a, b, c), ('fadd', ('fmul', c, ('fsub', b, a)), a), 'options->lower_flrp16'),
133 (('flrp@32', a, b, c), ('fadd', ('fmul', c, ('fsub', b, a)), a), 'options->lower_flrp32'),
134 (('flrp@64', a, b, c), ('fadd', ('fmul', c, ('fsub', b, a)), a), 'options->lower_flrp64'),
135 (('ftrunc', a), ('bcsel', ('flt', a, 0.0), ('fneg', ('ffloor', ('fabs', a))), ('ffloor', ('fabs', a))), 'options->lower_ftrunc'),
136 (('ffloor', a), ('fsub', a, ('ffract', a)), 'options->lower_ffloor'),
137 (('fadd', a, ('fneg', ('ffract', a))), ('ffloor', a), '!options->lower_ffloor'),
138 (('ffract', a), ('fsub', a, ('ffloor', a)), 'options->lower_ffract'),
139 (('fceil', a), ('fneg', ('ffloor', ('fneg', a))), 'options->lower_fceil'),
140 (('~fadd', ('fmul', a, ('fadd', 1.0, ('fneg', ('b2f', 'c@1')))), ('fmul', b, ('b2f', c))), ('bcsel', c, b, a), 'options->lower_flrp32'),
141 (('~fadd@32', ('fmul', a, ('fadd', 1.0, ('fneg', c ))), ('fmul', b, c )), ('flrp', a, b, c), '!options->lower_flrp32'),
142 (('~fadd@64', ('fmul', a, ('fadd', 1.0, ('fneg', c ))), ('fmul', b, c )), ('flrp', a, b, c), '!options->lower_flrp64'),
143 (('~fadd', a, ('fmul', ('b2f', 'c@1'), ('fadd', b, ('fneg', a)))), ('bcsel', c, b, a), 'options->lower_flrp32'),
144 (('~fadd@32', a, ('fmul', c , ('fadd', b, ('fneg', a)))), ('flrp', a, b, c), '!options->lower_flrp32'),
145 (('~fadd@64', a, ('fmul', c , ('fadd', b, ('fneg', a)))), ('flrp', a, b, c), '!options->lower_flrp64'),
146 (('ffma', a, b, c), ('fadd', ('fmul', a, b), c), 'options->lower_ffma'),
147 (('~fadd', ('fmul', a, b), c), ('ffma', a, b, c), 'options->fuse_ffma'),
148
149 (('~fmul', ('fadd', ('iand', ('ineg', ('b2i32', 'a@bool')), ('fmul', b, c)), '#d'), '#e'),
150 ('bcsel', a, ('fmul', ('fadd', ('fmul', b, c), d), e), ('fmul', d, e))),
151
152 (('fdot4', ('vec4', a, b, c, 1.0), d), ('fdph', ('vec3', a, b, c), d)),
153 (('fdot4', ('vec4', a, 0.0, 0.0, 0.0), b), ('fmul', a, b)),
154 (('fdot4', ('vec4', a, b, 0.0, 0.0), c), ('fdot2', ('vec2', a, b), c)),
155 (('fdot4', ('vec4', a, b, c, 0.0), d), ('fdot3', ('vec3', a, b, c), d)),
156
157 (('fdot3', ('vec3', a, 0.0, 0.0), b), ('fmul', a, b)),
158 (('fdot3', ('vec3', a, b, 0.0), c), ('fdot2', ('vec2', a, b), c)),
159
160 # (a * #b + #c) << #d
161 # ((a * #b) << #d) + (#c << #d)
162 # (a * (#b << #d)) + (#c << #d)
163 (('ishl', ('iadd', ('imul', a, '#b'), '#c'), '#d'),
164 ('iadd', ('imul', a, ('ishl', b, d)), ('ishl', c, d))),
165
166 # (a * #b) << #c
167 # a * (#b << #c)
168 (('ishl', ('imul', a, '#b'), '#c'), ('imul', a, ('ishl', b, c))),
169
170 # Comparison simplifications
171 (('~inot', ('flt', a, b)), ('fge', a, b)),
172 (('~inot', ('fge', a, b)), ('flt', a, b)),
173 (('~inot', ('feq', a, b)), ('fne', a, b)),
174 (('~inot', ('fne', a, b)), ('feq', a, b)),
175 (('inot', ('ilt', a, b)), ('ige', a, b)),
176 (('inot', ('ult', a, b)), ('uge', a, b)),
177 (('inot', ('ige', a, b)), ('ilt', a, b)),
178 (('inot', ('uge', a, b)), ('ult', a, b)),
179 (('inot', ('ieq', a, b)), ('ine', a, b)),
180 (('inot', ('ine', a, b)), ('ieq', a, b)),
181
182 # 0.0 >= b2f(a)
183 # b2f(a) <= 0.0
184 # b2f(a) == 0.0 because b2f(a) can only be 0 or 1
185 # inot(a)
186 (('fge', 0.0, ('b2f', 'a@1')), ('inot', a)),
187
188 (('fge', ('fneg', ('b2f', 'a@1')), 0.0), ('inot', a)),
189
190 (('fne', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('ior', a, b)),
191 (('fne', ('fmax', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('ior', a, b)),
192 (('fne', ('bcsel', a, 1.0, ('b2f', 'b@1')) , 0.0), ('ior', a, b)),
193 (('fne', ('b2f', 'a@1'), ('fneg', ('b2f', 'b@1'))), ('ior', a, b)),
194 (('fne', ('fmul', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('iand', a, b)),
195 (('fne', ('fmin', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('iand', a, b)),
196 (('fne', ('bcsel', a, ('b2f', 'b@1'), 0.0) , 0.0), ('iand', a, b)),
197 (('fne', ('fadd', ('b2f', 'a@1'), ('fneg', ('b2f', 'b@1'))), 0.0), ('ixor', a, b)),
198 (('fne', ('b2f', 'a@1') , ('b2f', 'b@1') ), ('ixor', a, b)),
199 (('fne', ('fneg', ('b2f', 'a@1')), ('fneg', ('b2f', 'b@1'))), ('ixor', a, b)),
200 (('feq', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('inot', ('ior', a, b))),
201 (('feq', ('fmax', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('inot', ('ior', a, b))),
202 (('feq', ('bcsel', a, 1.0, ('b2f', 'b@1')) , 0.0), ('inot', ('ior', a, b))),
203 (('feq', ('b2f', 'a@1'), ('fneg', ('b2f', 'b@1'))), ('inot', ('ior', a, b))),
204 (('feq', ('fmul', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('inot', ('iand', a, b))),
205 (('feq', ('fmin', ('b2f', 'a@1'), ('b2f', 'b@1')), 0.0), ('inot', ('iand', a, b))),
206 (('feq', ('bcsel', a, ('b2f', 'b@1'), 0.0) , 0.0), ('inot', ('iand', a, b))),
207 (('feq', ('fadd', ('b2f', 'a@1'), ('fneg', ('b2f', 'b@1'))), 0.0), ('ieq', a, b)),
208 (('feq', ('b2f', 'a@1') , ('b2f', 'b@1') ), ('ieq', a, b)),
209 (('feq', ('fneg', ('b2f', 'a@1')), ('fneg', ('b2f', 'b@1'))), ('ieq', a, b)),
210
211 # -(b2f(a) + b2f(b)) < 0
212 # 0 < b2f(a) + b2f(b)
213 # 0 != b2f(a) + b2f(b) b2f must be 0 or 1, so the sum is non-negative
214 # a || b
215 (('flt', ('fneg', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1'))), 0.0), ('ior', a, b)),
216 (('flt', 0.0, ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1'))), ('ior', a, b)),
217
218 # -(b2f(a) + b2f(b)) >= 0
219 # 0 >= b2f(a) + b2f(b)
220 # 0 == b2f(a) + b2f(b) b2f must be 0 or 1, so the sum is non-negative
221 # !(a || b)
222 (('fge', ('fneg', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1'))), 0.0), ('inot', ('ior', a, b))),
223 (('fge', 0.0, ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1'))), ('inot', ('ior', a, b))),
224
225 # Some optimizations (below) convert things like (a < b || c < b) into
226 # (min(a, c) < b). However, this interfers with the previous optimizations
227 # that try to remove comparisons with negated sums of b2f. This just
228 # breaks that apart.
229 (('flt', ('fmin', c, ('fneg', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1')))), 0.0),
230 ('ior', ('flt', c, 0.0), ('ior', a, b))),
231
232 (('~flt', ('fadd', a, b), a), ('flt', b, 0.0)),
233 (('~fge', ('fadd', a, b), a), ('fge', b, 0.0)),
234 (('~feq', ('fadd', a, b), a), ('feq', b, 0.0)),
235 (('~fne', ('fadd', a, b), a), ('fne', b, 0.0)),
236
237 # Cannot remove the addition from ilt or ige due to overflow.
238 (('ieq', ('iadd', a, b), a), ('ieq', b, 0)),
239 (('ine', ('iadd', a, b), a), ('ine', b, 0)),
240
241 # fmin(-b2f(a), b) >= 0.0
242 # -b2f(a) >= 0.0 && b >= 0.0
243 # -b2f(a) == 0.0 && b >= 0.0 -b2f can only be 0 or -1, never >0
244 # b2f(a) == 0.0 && b >= 0.0
245 # a == False && b >= 0.0
246 # !a && b >= 0.0
247 #
248 # The fge in the second replacement is not a typo. I leave the proof that
249 # "fmin(-b2f(a), b) >= 0 <=> fmin(-b2f(a), b) == 0" as an exercise for the
250 # reader.
251 (('fge', ('fmin', ('fneg', ('b2f', 'a@1')), 'b@1'), 0.0), ('iand', ('inot', a), ('fge', b, 0.0))),
252 (('feq', ('fmin', ('fneg', ('b2f', 'a@1')), 'b@1'), 0.0), ('iand', ('inot', a), ('fge', b, 0.0))),
253
254 (('feq', ('b2f', 'a@1'), 0.0), ('inot', a)),
255 (('fne', ('b2f', 'a@1'), 0.0), a),
256 (('ieq', ('b2i', 'a@1'), 0), ('inot', a)),
257 (('ine', ('b2i', 'a@1'), 0), a),
258
259 (('fne', ('u2f', a), 0.0), ('ine', a, 0)),
260 (('feq', ('u2f', a), 0.0), ('ieq', a, 0)),
261 (('fge', ('u2f', a), 0.0), True),
262 (('fge', 0.0, ('u2f', a)), ('uge', 0, a)), # ieq instead?
263 (('flt', ('u2f', a), 0.0), False),
264 (('flt', 0.0, ('u2f', a)), ('ult', 0, a)), # ine instead?
265 (('fne', ('i2f', a), 0.0), ('ine', a, 0)),
266 (('feq', ('i2f', a), 0.0), ('ieq', a, 0)),
267 (('fge', ('i2f', a), 0.0), ('ige', a, 0)),
268 (('fge', 0.0, ('i2f', a)), ('ige', 0, a)),
269 (('flt', ('i2f', a), 0.0), ('ilt', a, 0)),
270 (('flt', 0.0, ('i2f', a)), ('ilt', 0, a)),
271
272 # 0.0 < fabs(a)
273 # fabs(a) > 0.0
274 # fabs(a) != 0.0 because fabs(a) must be >= 0
275 # a != 0.0
276 (('~flt', 0.0, ('fabs', a)), ('fne', a, 0.0)),
277
278 # -fabs(a) < 0.0
279 # fabs(a) > 0.0
280 (('~flt', ('fneg', ('fabs', a)), 0.0), ('fne', a, 0.0)),
281
282 # 0.0 >= fabs(a)
283 # 0.0 == fabs(a) because fabs(a) must be >= 0
284 # 0.0 == a
285 (('fge', 0.0, ('fabs', a)), ('feq', a, 0.0)),
286
287 # -fabs(a) >= 0.0
288 # 0.0 >= fabs(a)
289 (('fge', ('fneg', ('fabs', a)), 0.0), ('feq', a, 0.0)),
290
291 (('fmax', ('b2f(is_used_once)', 'a@1'), ('b2f', 'b@1')), ('b2f', ('ior', a, b))),
292 (('fmax', ('fneg(is_used_once)', ('b2f(is_used_once)', 'a@1')), ('fneg', ('b2f', 'b@1'))), ('fneg', ('b2f', ('ior', a, b)))),
293 (('fmin', ('b2f(is_used_once)', 'a@1'), ('b2f', 'b@1')), ('b2f', ('iand', a, b))),
294 (('fmin', ('fneg(is_used_once)', ('b2f(is_used_once)', 'a@1')), ('fneg', ('b2f', 'b@1'))), ('fneg', ('b2f', ('iand', a, b)))),
295
296 # fmin(b2f(a), b)
297 # bcsel(a, fmin(b2f(a), b), fmin(b2f(a), b))
298 # bcsel(a, fmin(b2f(True), b), fmin(b2f(False), b))
299 # bcsel(a, fmin(1.0, b), fmin(0.0, b))
300 #
301 # Since b is a constant, constant folding will eliminate the fmin and the
302 # fmax. If b is > 1.0, the bcsel will be replaced with a b2f.
303 (('fmin', ('b2f', 'a@1'), '#b'), ('bcsel', a, ('fmin', b, 1.0), ('fmin', b, 0.0))),
304
305 (('flt', ('fadd(is_used_once)', a, ('fneg', b)), 0.0), ('flt', a, b)),
306
307 (('fge', ('fneg', ('fabs', a)), 0.0), ('feq', a, 0.0)),
308 (('~bcsel', ('flt', b, a), b, a), ('fmin', a, b)),
309 (('~bcsel', ('flt', a, b), b, a), ('fmax', a, b)),
310 (('~bcsel', ('fge', a, b), b, a), ('fmin', a, b)),
311 (('~bcsel', ('fge', b, a), b, a), ('fmax', a, b)),
312 (('bcsel', ('i2b', a), b, c), ('bcsel', ('ine', a, 0), b, c)),
313 (('bcsel', ('inot', a), b, c), ('bcsel', a, c, b)),
314 (('bcsel', a, ('bcsel', a, b, c), d), ('bcsel', a, b, d)),
315 (('bcsel', a, b, ('bcsel', a, c, d)), ('bcsel', a, b, d)),
316 (('bcsel', a, ('bcsel', b, c, d), ('bcsel(is_used_once)', b, c, 'e')), ('bcsel', b, c, ('bcsel', a, d, 'e'))),
317 (('bcsel', a, ('bcsel(is_used_once)', b, c, d), ('bcsel', b, c, 'e')), ('bcsel', b, c, ('bcsel', a, d, 'e'))),
318 (('bcsel', a, ('bcsel', b, c, d), ('bcsel(is_used_once)', b, 'e', d)), ('bcsel', b, ('bcsel', a, c, 'e'), d)),
319 (('bcsel', a, ('bcsel(is_used_once)', b, c, d), ('bcsel', b, 'e', d)), ('bcsel', b, ('bcsel', a, c, 'e'), d)),
320 (('bcsel', a, True, b), ('ior', a, b)),
321 (('bcsel', a, a, b), ('ior', a, b)),
322 (('bcsel', a, b, False), ('iand', a, b)),
323 (('bcsel', a, b, a), ('iand', a, b)),
324 (('fmin', a, a), a),
325 (('fmax', a, a), a),
326 (('imin', a, a), a),
327 (('imax', a, a), a),
328 (('umin', a, a), a),
329 (('umax', a, a), a),
330 (('fmax', ('fmax', a, b), b), ('fmax', a, b)),
331 (('umax', ('umax', a, b), b), ('umax', a, b)),
332 (('imax', ('imax', a, b), b), ('imax', a, b)),
333 (('fmin', ('fmin', a, b), b), ('fmin', a, b)),
334 (('umin', ('umin', a, b), b), ('umin', a, b)),
335 (('imin', ('imin', a, b), b), ('imin', a, b)),
336 (('fmax', a, ('fneg', a)), ('fabs', a)),
337 (('imax', a, ('ineg', a)), ('iabs', a)),
338 (('fmin', a, ('fneg', a)), ('fneg', ('fabs', a))),
339 (('imin', a, ('ineg', a)), ('ineg', ('iabs', a))),
340 (('fmin', a, ('fneg', ('fabs', a))), ('fneg', ('fabs', a))),
341 (('imin', a, ('ineg', ('iabs', a))), ('ineg', ('iabs', a))),
342 (('fmin', a, ('fabs', a)), a),
343 (('imin', a, ('iabs', a)), a),
344 (('fmax', a, ('fneg', ('fabs', a))), a),
345 (('imax', a, ('ineg', ('iabs', a))), a),
346 (('fmax', a, ('fabs', a)), ('fabs', a)),
347 (('imax', a, ('iabs', a)), ('iabs', a)),
348 (('fmax', a, ('fneg', a)), ('fabs', a)),
349 (('imax', a, ('ineg', a)), ('iabs', a)),
350 (('~fmin', ('fmax', a, 0.0), 1.0), ('fsat', a), '!options->lower_fsat'),
351 (('~fmax', ('fmin', a, 1.0), 0.0), ('fsat', a), '!options->lower_fsat'),
352 (('fsat', ('fsign', a)), ('b2f', ('flt', 0.0, a))),
353 (('fsat', ('b2f', a)), ('b2f', a)),
354 (('fsat', a), ('fmin', ('fmax', a, 0.0), 1.0), 'options->lower_fsat'),
355 (('fsat', ('fsat', a)), ('fsat', a)),
356 (('fmin', ('fmax', ('fmin', ('fmax', a, b), c), b), c), ('fmin', ('fmax', a, b), c)),
357 (('imin', ('imax', ('imin', ('imax', a, b), c), b), c), ('imin', ('imax', a, b), c)),
358 (('umin', ('umax', ('umin', ('umax', a, b), c), b), c), ('umin', ('umax', a, b), c)),
359 (('fmax', ('fsat', a), '#b@32(is_zero_to_one)'), ('fsat', ('fmax', a, b))),
360 (('fmin', ('fsat', a), '#b@32(is_zero_to_one)'), ('fsat', ('fmin', a, b))),
361 (('extract_u8', ('imin', ('imax', a, 0), 0xff), 0), ('imin', ('imax', a, 0), 0xff)),
362 (('~ior', ('flt(is_used_once)', a, b), ('flt', a, c)), ('flt', a, ('fmax', b, c))),
363 (('~ior', ('flt(is_used_once)', a, c), ('flt', b, c)), ('flt', ('fmin', a, b), c)),
364 (('~ior', ('fge(is_used_once)', a, b), ('fge', a, c)), ('fge', a, ('fmin', b, c))),
365 (('~ior', ('fge(is_used_once)', a, c), ('fge', b, c)), ('fge', ('fmax', a, b), c)),
366 (('~ior', ('flt', a, '#b'), ('flt', a, '#c')), ('flt', a, ('fmax', b, c))),
367 (('~ior', ('flt', '#a', c), ('flt', '#b', c)), ('flt', ('fmin', a, b), c)),
368 (('~ior', ('fge', a, '#b'), ('fge', a, '#c')), ('fge', a, ('fmin', b, c))),
369 (('~ior', ('fge', '#a', c), ('fge', '#b', c)), ('fge', ('fmax', a, b), c)),
370 (('~iand', ('flt(is_used_once)', a, b), ('flt', a, c)), ('flt', a, ('fmin', b, c))),
371 (('~iand', ('flt(is_used_once)', a, c), ('flt', b, c)), ('flt', ('fmax', a, b), c)),
372 (('~iand', ('fge(is_used_once)', a, b), ('fge', a, c)), ('fge', a, ('fmax', b, c))),
373 (('~iand', ('fge(is_used_once)', a, c), ('fge', b, c)), ('fge', ('fmin', a, b), c)),
374 (('~iand', ('flt', a, '#b'), ('flt', a, '#c')), ('flt', a, ('fmin', b, c))),
375 (('~iand', ('flt', '#a', c), ('flt', '#b', c)), ('flt', ('fmax', a, b), c)),
376 (('~iand', ('fge', a, '#b'), ('fge', a, '#c')), ('fge', a, ('fmax', b, c))),
377 (('~iand', ('fge', '#a', c), ('fge', '#b', c)), ('fge', ('fmin', a, b), c)),
378
379 (('ior', ('ilt(is_used_once)', a, b), ('ilt', a, c)), ('ilt', a, ('imax', b, c))),
380 (('ior', ('ilt(is_used_once)', a, c), ('ilt', b, c)), ('ilt', ('imin', a, b), c)),
381 (('ior', ('ige(is_used_once)', a, b), ('ige', a, c)), ('ige', a, ('imin', b, c))),
382 (('ior', ('ige(is_used_once)', a, c), ('ige', b, c)), ('ige', ('imax', a, b), c)),
383 (('ior', ('ult(is_used_once)', a, b), ('ult', a, c)), ('ult', a, ('umax', b, c))),
384 (('ior', ('ult(is_used_once)', a, c), ('ult', b, c)), ('ult', ('umin', a, b), c)),
385 (('ior', ('uge(is_used_once)', a, b), ('uge', a, c)), ('uge', a, ('umin', b, c))),
386 (('ior', ('uge(is_used_once)', a, c), ('uge', b, c)), ('uge', ('umax', a, b), c)),
387 (('iand', ('ilt(is_used_once)', a, b), ('ilt', a, c)), ('ilt', a, ('imin', b, c))),
388 (('iand', ('ilt(is_used_once)', a, c), ('ilt', b, c)), ('ilt', ('imax', a, b), c)),
389 (('iand', ('ige(is_used_once)', a, b), ('ige', a, c)), ('ige', a, ('imax', b, c))),
390 (('iand', ('ige(is_used_once)', a, c), ('ige', b, c)), ('ige', ('imin', a, b), c)),
391 (('iand', ('ult(is_used_once)', a, b), ('ult', a, c)), ('ult', a, ('umin', b, c))),
392 (('iand', ('ult(is_used_once)', a, c), ('ult', b, c)), ('ult', ('umax', a, b), c)),
393 (('iand', ('uge(is_used_once)', a, b), ('uge', a, c)), ('uge', a, ('umax', b, c))),
394 (('iand', ('uge(is_used_once)', a, c), ('uge', b, c)), ('uge', ('umin', a, b), c)),
395
396 # Common pattern like 'if (i == 0 || i == 1 || ...)'
397 (('ior', ('ieq', a, 0), ('ieq', a, 1)), ('uge', 1, a)),
398 (('ior', ('uge', 1, a), ('ieq', a, 2)), ('uge', 2, a)),
399 (('ior', ('uge', 2, a), ('ieq', a, 3)), ('uge', 3, a)),
400
401 # The (i2f32, ...) part is an open-coded fsign. When that is combined with
402 # the bcsel, it's basically copysign(1.0, a). There is no copysign in NIR,
403 # so emit an open-coded version of that.
404 (('bcsel@32', ('feq', a, 0.0), 1.0, ('i2f32', ('iadd', ('b2i32', ('flt', 0.0, 'a@32')), ('ineg', ('b2i32', ('flt', 'a@32', 0.0)))))),
405 ('ior', 0x3f800000, ('iand', a, 0x80000000))),
406
407 (('ior', a, ('ieq', a, False)), True),
408 (('ior', a, ('inot', a)), -1),
409
410 (('iand', ('ieq', 'a@32', 0), ('ieq', 'b@32', 0)), ('ieq', ('ior', 'a@32', 'b@32'), 0)),
411
412 # These patterns can result when (a < b || a < c) => (a < min(b, c))
413 # transformations occur before constant propagation and loop-unrolling.
414 (('~flt', a, ('fmax', b, a)), ('flt', a, b)),
415 (('~flt', ('fmin', a, b), a), ('flt', b, a)),
416 (('~fge', a, ('fmin', b, a)), True),
417 (('~fge', ('fmax', a, b), a), True),
418 (('~flt', a, ('fmin', b, a)), False),
419 (('~flt', ('fmax', a, b), a), False),
420 (('~fge', a, ('fmax', b, a)), ('fge', a, b)),
421 (('~fge', ('fmin', a, b), a), ('fge', b, a)),
422
423 (('ilt', a, ('imax', b, a)), ('ilt', a, b)),
424 (('ilt', ('imin', a, b), a), ('ilt', b, a)),
425 (('ige', a, ('imin', b, a)), True),
426 (('ige', ('imax', a, b), a), True),
427 (('ult', a, ('umax', b, a)), ('ult', a, b)),
428 (('ult', ('umin', a, b), a), ('ult', b, a)),
429 (('uge', a, ('umin', b, a)), True),
430 (('uge', ('umax', a, b), a), True),
431 (('ilt', a, ('imin', b, a)), False),
432 (('ilt', ('imax', a, b), a), False),
433 (('ige', a, ('imax', b, a)), ('ige', a, b)),
434 (('ige', ('imin', a, b), a), ('ige', b, a)),
435 (('ult', a, ('umin', b, a)), False),
436 (('ult', ('umax', a, b), a), False),
437 (('uge', a, ('umax', b, a)), ('uge', a, b)),
438 (('uge', ('umin', a, b), a), ('uge', b, a)),
439
440 (('ilt', '#a', ('imax', '#b', c)), ('ior', ('ilt', a, b), ('ilt', a, c))),
441 (('ilt', ('imin', '#a', b), '#c'), ('ior', ('ilt', a, c), ('ilt', b, c))),
442 (('ige', '#a', ('imin', '#b', c)), ('ior', ('ige', a, b), ('ige', a, c))),
443 (('ige', ('imax', '#a', b), '#c'), ('ior', ('ige', a, c), ('ige', b, c))),
444 (('ult', '#a', ('umax', '#b', c)), ('ior', ('ult', a, b), ('ult', a, c))),
445 (('ult', ('umin', '#a', b), '#c'), ('ior', ('ult', a, c), ('ult', b, c))),
446 (('uge', '#a', ('umin', '#b', c)), ('ior', ('uge', a, b), ('uge', a, c))),
447 (('uge', ('umax', '#a', b), '#c'), ('ior', ('uge', a, c), ('uge', b, c))),
448 (('ilt', '#a', ('imin', '#b', c)), ('iand', ('ilt', a, b), ('ilt', a, c))),
449 (('ilt', ('imax', '#a', b), '#c'), ('iand', ('ilt', a, c), ('ilt', b, c))),
450 (('ige', '#a', ('imax', '#b', c)), ('iand', ('ige', a, b), ('ige', a, c))),
451 (('ige', ('imin', '#a', b), '#c'), ('iand', ('ige', a, c), ('ige', b, c))),
452 (('ult', '#a', ('umin', '#b', c)), ('iand', ('ult', a, b), ('ult', a, c))),
453 (('ult', ('umax', '#a', b), '#c'), ('iand', ('ult', a, c), ('ult', b, c))),
454 (('uge', '#a', ('umax', '#b', c)), ('iand', ('uge', a, b), ('uge', a, c))),
455 (('uge', ('umin', '#a', b), '#c'), ('iand', ('uge', a, c), ('uge', b, c))),
456
457 # Thanks to sign extension, the ishr(a, b) is negative if and only if a is
458 # negative.
459 (('bcsel', ('ilt', a, 0), ('ineg', ('ishr', a, b)), ('ishr', a, b)),
460 ('iabs', ('ishr', a, b))),
461 (('iabs', ('ishr', ('iabs', a), b)), ('ishr', ('iabs', a), b)),
462
463 (('fabs', ('slt', a, b)), ('slt', a, b)),
464 (('fabs', ('sge', a, b)), ('sge', a, b)),
465 (('fabs', ('seq', a, b)), ('seq', a, b)),
466 (('fabs', ('sne', a, b)), ('sne', a, b)),
467 (('slt', a, b), ('b2f', ('flt', a, b)), 'options->lower_scmp'),
468 (('sge', a, b), ('b2f', ('fge', a, b)), 'options->lower_scmp'),
469 (('seq', a, b), ('b2f', ('feq', a, b)), 'options->lower_scmp'),
470 (('sne', a, b), ('b2f', ('fne', a, b)), 'options->lower_scmp'),
471 (('fne', ('fneg', a), a), ('fne', a, 0.0)),
472 (('feq', ('fneg', a), a), ('feq', a, 0.0)),
473 # Emulating booleans
474 (('imul', ('b2i', 'a@1'), ('b2i', 'b@1')), ('b2i', ('iand', a, b))),
475 (('fmul', ('b2f', 'a@1'), ('b2f', 'b@1')), ('b2f', ('iand', a, b))),
476 (('fsat', ('fadd', ('b2f', 'a@1'), ('b2f', 'b@1'))), ('b2f', ('ior', a, b))),
477 (('iand', 'a@bool32', 1.0), ('b2f', a)),
478 # True/False are ~0 and 0 in NIR. b2i of True is 1, and -1 is ~0 (True).
479 (('ineg', ('b2i32', 'a@32')), a),
480 (('flt', ('fneg', ('b2f', 'a@1')), 0), a), # Generated by TGSI KILL_IF.
481 (('flt', ('fsub', 0.0, ('b2f', 'a@1')), 0), a), # Generated by TGSI KILL_IF.
482 # Comparison with the same args. Note that these are not done for
483 # the float versions because NaN always returns false on float
484 # inequalities.
485 (('ilt', a, a), False),
486 (('ige', a, a), True),
487 (('ieq', a, a), True),
488 (('ine', a, a), False),
489 (('ult', a, a), False),
490 (('uge', a, a), True),
491 # Logical and bit operations
492 (('fand', a, 0.0), 0.0),
493 (('iand', a, a), a),
494 (('iand', a, ~0), a),
495 (('iand', a, 0), 0),
496 (('ior', a, a), a),
497 (('ior', a, 0), a),
498 (('ior', a, True), True),
499 (('fxor', a, a), 0.0),
500 (('ixor', a, a), 0),
501 (('ixor', a, 0), a),
502 (('inot', ('inot', a)), a),
503 (('ior', ('iand', a, b), b), b),
504 (('ior', ('ior', a, b), b), ('ior', a, b)),
505 (('iand', ('ior', a, b), b), b),
506 (('iand', ('iand', a, b), b), ('iand', a, b)),
507 # DeMorgan's Laws
508 (('iand', ('inot', a), ('inot', b)), ('inot', ('ior', a, b))),
509 (('ior', ('inot', a), ('inot', b)), ('inot', ('iand', a, b))),
510 # Shift optimizations
511 (('ishl', 0, a), 0),
512 (('ishl', a, 0), a),
513 (('ishr', 0, a), 0),
514 (('ishr', a, 0), a),
515 (('ushr', 0, a), 0),
516 (('ushr', a, 0), a),
517 (('iand', 0xff, ('ushr@32', a, 24)), ('ushr', a, 24)),
518 (('iand', 0xffff, ('ushr@32', a, 16)), ('ushr', a, 16)),
519 # Exponential/logarithmic identities
520 (('~fexp2', ('flog2', a)), a), # 2^lg2(a) = a
521 (('~flog2', ('fexp2', a)), a), # lg2(2^a) = a
522 (('fpow', a, b), ('fexp2', ('fmul', ('flog2', a), b)), 'options->lower_fpow'), # a^b = 2^(lg2(a)*b)
523 (('~fexp2', ('fmul', ('flog2', a), b)), ('fpow', a, b), '!options->lower_fpow'), # 2^(lg2(a)*b) = a^b
524 (('~fexp2', ('fadd', ('fmul', ('flog2', a), b), ('fmul', ('flog2', c), d))),
525 ('~fmul', ('fpow', a, b), ('fpow', c, d)), '!options->lower_fpow'), # 2^(lg2(a) * b + lg2(c) + d) = a^b * c^d
526 (('~fexp2', ('fmul', ('flog2', a), 2.0)), ('fmul', a, a)),
527 (('~fexp2', ('fmul', ('flog2', a), 4.0)), ('fmul', ('fmul', a, a), ('fmul', a, a))),
528 (('~fpow', a, 1.0), a),
529 (('~fpow', a, 2.0), ('fmul', a, a)),
530 (('~fpow', a, 4.0), ('fmul', ('fmul', a, a), ('fmul', a, a))),
531 (('~fpow', 2.0, a), ('fexp2', a)),
532 (('~fpow', ('fpow', a, 2.2), 0.454545), a),
533 (('~fpow', ('fabs', ('fpow', a, 2.2)), 0.454545), ('fabs', a)),
534 (('~fsqrt', ('fexp2', a)), ('fexp2', ('fmul', 0.5, a))),
535 (('~frcp', ('fexp2', a)), ('fexp2', ('fneg', a))),
536 (('~frsq', ('fexp2', a)), ('fexp2', ('fmul', -0.5, a))),
537 (('~flog2', ('fsqrt', a)), ('fmul', 0.5, ('flog2', a))),
538 (('~flog2', ('frcp', a)), ('fneg', ('flog2', a))),
539 (('~flog2', ('frsq', a)), ('fmul', -0.5, ('flog2', a))),
540 (('~flog2', ('fpow', a, b)), ('fmul', b, ('flog2', a))),
541 (('~fmul', ('fexp2(is_used_once)', a), ('fexp2(is_used_once)', b)), ('fexp2', ('fadd', a, b))),
542 # Division and reciprocal
543 (('~fdiv', 1.0, a), ('frcp', a)),
544 (('fdiv', a, b), ('fmul', a, ('frcp', b)), 'options->lower_fdiv'),
545 (('~frcp', ('frcp', a)), a),
546 (('~frcp', ('fsqrt', a)), ('frsq', a)),
547 (('fsqrt', a), ('frcp', ('frsq', a)), 'options->lower_fsqrt'),
548 (('~frcp', ('frsq', a)), ('fsqrt', a), '!options->lower_fsqrt'),
549 # Boolean simplifications
550 (('i2b32(is_used_by_if)', a), ('ine32', a, 0)),
551 (('i2b1(is_used_by_if)', a), ('ine', a, 0)),
552 (('ieq', a, True), a),
553 (('ine(is_not_used_by_if)', a, True), ('inot', a)),
554 (('ine', a, False), a),
555 (('ieq(is_not_used_by_if)', a, False), ('inot', 'a')),
556 (('bcsel', a, True, False), a),
557 (('bcsel', a, False, True), ('inot', a)),
558 (('bcsel@32', a, 1.0, 0.0), ('b2f', a)),
559 (('bcsel@32', a, 0.0, 1.0), ('b2f', ('inot', a))),
560 (('bcsel@32', a, -1.0, -0.0), ('fneg', ('b2f', a))),
561 (('bcsel@32', a, -0.0, -1.0), ('fneg', ('b2f', ('inot', a)))),
562 (('bcsel', True, b, c), b),
563 (('bcsel', False, b, c), c),
564 (('bcsel', a, ('b2f(is_used_once)', 'b@32'), ('b2f', 'c@32')), ('b2f', ('bcsel', a, b, c))),
565 # The result of this should be hit by constant propagation and, in the
566 # next round of opt_algebraic, get picked up by one of the above two.
567 (('bcsel', '#a', b, c), ('bcsel', ('ine', 'a', 0), b, c)),
568
569 (('bcsel', a, b, b), b),
570 (('fcsel', a, b, b), b),
571
572 # D3D Boolean emulation
573 (('bcsel', a, -1, 0), ('ineg', ('b2i', 'a@1'))),
574 (('bcsel', a, 0, -1), ('ineg', ('b2i', ('inot', a)))),
575 (('iand', ('ineg', ('b2i', 'a@1')), ('ineg', ('b2i', 'b@1'))),
576 ('ineg', ('b2i', ('iand', a, b)))),
577 (('ior', ('ineg', ('b2i','a@1')), ('ineg', ('b2i', 'b@1'))),
578 ('ineg', ('b2i', ('ior', a, b)))),
579 (('ieq', ('ineg', ('b2i', 'a@1')), 0), ('inot', a)),
580 (('ieq', ('ineg', ('b2i', 'a@1')), -1), a),
581 (('ine', ('ineg', ('b2i', 'a@1')), 0), a),
582 (('ine', ('ineg', ('b2i', 'a@1')), -1), ('inot', a)),
583 (('iand', ('ineg', ('b2i', a)), 1.0), ('b2f', a)),
584
585 # SM5 32-bit shifts are defined to use the 5 least significant bits
586 (('ishl', 'a@32', ('iand', 31, b)), ('ishl', a, b)),
587 (('ishr', 'a@32', ('iand', 31, b)), ('ishr', a, b)),
588 (('ushr', 'a@32', ('iand', 31, b)), ('ushr', a, b)),
589
590 # Conversions
591 (('i2b32', ('b2i', 'a@32')), a),
592 (('f2i', ('ftrunc', a)), ('f2i', a)),
593 (('f2u', ('ftrunc', a)), ('f2u', a)),
594 (('i2b', ('ineg', a)), ('i2b', a)),
595 (('i2b', ('iabs', a)), ('i2b', a)),
596 (('fabs', ('b2f', a)), ('b2f', a)),
597 (('iabs', ('b2i', a)), ('b2i', a)),
598 (('inot', ('f2b1', a)), ('feq', a, 0.0)),
599
600 # Ironically, mark these as imprecise because removing the conversions may
601 # preserve more precision than doing the conversions (e.g.,
602 # uint(float(0x81818181u)) == 0x81818200).
603 (('~f2i32', ('i2f', 'a@32')), a),
604 (('~f2i32', ('u2f', 'a@32')), a),
605 (('~f2u32', ('i2f', 'a@32')), a),
606 (('~f2u32', ('u2f', 'a@32')), a),
607
608 # Section 5.4.1 (Conversion and Scalar Constructors) of the GLSL 4.60 spec
609 # says:
610 #
611 # It is undefined to convert a negative floating-point value to an
612 # uint.
613 #
614 # Assuming that (uint)some_float behaves like (uint)(int)some_float allows
615 # some optimizations in the i965 backend to proceed.
616 (('ige', ('f2u', a), b), ('ige', ('f2i', a), b)),
617 (('ige', b, ('f2u', a)), ('ige', b, ('f2i', a))),
618 (('ilt', ('f2u', a), b), ('ilt', ('f2i', a), b)),
619 (('ilt', b, ('f2u', a)), ('ilt', b, ('f2i', a))),
620
621 # Packing and then unpacking does nothing
622 (('unpack_64_2x32_split_x', ('pack_64_2x32_split', a, b)), a),
623 (('unpack_64_2x32_split_y', ('pack_64_2x32_split', a, b)), b),
624 (('pack_64_2x32_split', ('unpack_64_2x32_split_x', a),
625 ('unpack_64_2x32_split_y', a)), a),
626
627 # Byte extraction
628 (('ushr', 'a@16', 8), ('extract_u8', a, 1), '!options->lower_extract_byte'),
629 (('ushr', 'a@32', 24), ('extract_u8', a, 3), '!options->lower_extract_byte'),
630 (('ushr', 'a@64', 56), ('extract_u8', a, 7), '!options->lower_extract_byte'),
631 (('ishr', 'a@16', 8), ('extract_i8', a, 1), '!options->lower_extract_byte'),
632 (('ishr', 'a@32', 24), ('extract_i8', a, 3), '!options->lower_extract_byte'),
633 (('ishr', 'a@64', 56), ('extract_i8', a, 7), '!options->lower_extract_byte'),
634 (('iand', 0xff, a), ('extract_u8', a, 0), '!options->lower_extract_byte')
635 ]
636
637 # After the ('extract_u8', a, 0) pattern, above, triggers, there will be
638 # patterns like those below.
639 for op in ('ushr', 'ishr'):
640 optimizations.extend([(('extract_u8', (op, 'a@16', 8), 0), ('extract_u8', a, 1))])
641 optimizations.extend([(('extract_u8', (op, 'a@32', 8 * i), 0), ('extract_u8', a, i)) for i in range(1, 4)])
642 optimizations.extend([(('extract_u8', (op, 'a@64', 8 * i), 0), ('extract_u8', a, i)) for i in range(1, 8)])
643
644 optimizations.extend([(('extract_u8', ('extract_u16', a, 1), 0), ('extract_u8', a, 2))])
645
646 # After the ('extract_[iu]8', a, 3) patterns, above, trigger, there will be
647 # patterns like those below.
648 for op in ('extract_u8', 'extract_i8'):
649 optimizations.extend([((op, ('ishl', 'a@16', 8), 1), (op, a, 0))])
650 optimizations.extend([((op, ('ishl', 'a@32', 24 - 8 * i), 3), (op, a, i)) for i in range(2, -1, -1)])
651 optimizations.extend([((op, ('ishl', 'a@64', 56 - 8 * i), 7), (op, a, i)) for i in range(6, -1, -1)])
652
653 optimizations.extend([
654 # Word extraction
655 (('ushr', ('ishl', 'a@32', 16), 16), ('extract_u16', a, 0), '!options->lower_extract_word'),
656 (('ushr', 'a@32', 16), ('extract_u16', a, 1), '!options->lower_extract_word'),
657 (('ishr', ('ishl', 'a@32', 16), 16), ('extract_i16', a, 0), '!options->lower_extract_word'),
658 (('ishr', 'a@32', 16), ('extract_i16', a, 1), '!options->lower_extract_word'),
659 (('iand', 0xffff, a), ('extract_u16', a, 0), '!options->lower_extract_word'),
660
661 # Subtracts
662 (('~fsub', a, ('fsub', 0.0, b)), ('fadd', a, b)),
663 (('isub', a, ('isub', 0, b)), ('iadd', a, b)),
664 (('ussub_4x8', a, 0), a),
665 (('ussub_4x8', a, ~0), 0),
666 (('fsub', a, b), ('fadd', a, ('fneg', b)), 'options->lower_sub'),
667 (('isub', a, b), ('iadd', a, ('ineg', b)), 'options->lower_sub'),
668 (('fneg', a), ('fsub', 0.0, a), 'options->lower_negate'),
669 (('ineg', a), ('isub', 0, a), 'options->lower_negate'),
670 (('~fadd', a, ('fsub', 0.0, b)), ('fsub', a, b)),
671 (('iadd', a, ('isub', 0, b)), ('isub', a, b)),
672 (('fabs', ('fsub', 0.0, a)), ('fabs', a)),
673 (('iabs', ('isub', 0, a)), ('iabs', a)),
674
675 # Propagate negation up multiplication chains
676 (('fmul', ('fneg', a), b), ('fneg', ('fmul', a, b))),
677 (('imul', ('ineg', a), b), ('ineg', ('imul', a, b))),
678
679 # Propagate constants up multiplication chains
680 (('~fmul(is_used_once)', ('fmul(is_used_once)', 'a(is_not_const)', 'b(is_not_const)'), '#c'), ('fmul', ('fmul', a, c), b)),
681 (('imul(is_used_once)', ('imul(is_used_once)', 'a(is_not_const)', 'b(is_not_const)'), '#c'), ('imul', ('imul', a, c), b)),
682 (('~fadd(is_used_once)', ('fadd(is_used_once)', 'a(is_not_const)', 'b(is_not_const)'), '#c'), ('fadd', ('fadd', a, c), b)),
683 (('iadd(is_used_once)', ('iadd(is_used_once)', 'a(is_not_const)', 'b(is_not_const)'), '#c'), ('iadd', ('iadd', a, c), b)),
684
685 # Reassociate constants in add/mul chains so they can be folded together.
686 # For now, we mostly only handle cases where the constants are separated by
687 # a single non-constant. We could do better eventually.
688 (('~fmul', '#a', ('fmul', 'b(is_not_const)', '#c')), ('fmul', ('fmul', a, c), b)),
689 (('imul', '#a', ('imul', 'b(is_not_const)', '#c')), ('imul', ('imul', a, c), b)),
690 (('~fadd', '#a', ('fadd', 'b(is_not_const)', '#c')), ('fadd', ('fadd', a, c), b)),
691 (('~fadd', '#a', ('fneg', ('fadd', 'b(is_not_const)', '#c'))), ('fadd', ('fadd', a, ('fneg', c)), ('fneg', b))),
692 (('iadd', '#a', ('iadd', 'b(is_not_const)', '#c')), ('iadd', ('iadd', a, c), b)),
693
694 # By definition...
695 (('bcsel', ('ige', ('find_lsb', a), 0), ('find_lsb', a), -1), ('find_lsb', a)),
696 (('bcsel', ('ige', ('ifind_msb', a), 0), ('ifind_msb', a), -1), ('ifind_msb', a)),
697 (('bcsel', ('ige', ('ufind_msb', a), 0), ('ufind_msb', a), -1), ('ufind_msb', a)),
698
699 (('bcsel', ('ine', a, 0), ('find_lsb', a), -1), ('find_lsb', a)),
700 (('bcsel', ('ine', a, 0), ('ifind_msb', a), -1), ('ifind_msb', a)),
701 (('bcsel', ('ine', a, 0), ('ufind_msb', a), -1), ('ufind_msb', a)),
702
703 (('bcsel', ('ine', a, -1), ('ifind_msb', a), -1), ('ifind_msb', a)),
704
705 # Misc. lowering
706 (('fmod@16', a, b), ('fsub', a, ('fmul', b, ('ffloor', ('fdiv', a, b)))), 'options->lower_fmod16'),
707 (('fmod@32', a, b), ('fsub', a, ('fmul', b, ('ffloor', ('fdiv', a, b)))), 'options->lower_fmod32'),
708 (('fmod@64', a, b), ('fsub', a, ('fmul', b, ('ffloor', ('fdiv', a, b)))), 'options->lower_fmod64'),
709 (('frem', a, b), ('fsub', a, ('fmul', b, ('ftrunc', ('fdiv', a, b)))), 'options->lower_fmod32'),
710 (('uadd_carry@32', a, b), ('b2i', ('ult', ('iadd', a, b), a)), 'options->lower_uadd_carry'),
711 (('usub_borrow@32', a, b), ('b2i', ('ult', a, b)), 'options->lower_usub_borrow'),
712
713 (('bitfield_insert', 'base', 'insert', 'offset', 'bits'),
714 ('bcsel', ('ilt', 31, 'bits'), 'insert',
715 ('bfi', ('bfm', 'bits', 'offset'), 'insert', 'base')),
716 'options->lower_bitfield_insert'),
717 (('ihadd', a, b), ('iadd', ('iand', a, b), ('ishr', ('ixor', a, b), 1)), 'options->lower_hadd'),
718 (('uhadd', a, b), ('iadd', ('iand', a, b), ('ushr', ('ixor', a, b), 1)), 'options->lower_hadd'),
719 (('irhadd', a, b), ('isub', ('ior', a, b), ('ishr', ('ixor', a, b), 1)), 'options->lower_hadd'),
720 (('urhadd', a, b), ('isub', ('ior', a, b), ('ushr', ('ixor', a, b), 1)), 'options->lower_hadd'),
721 (('uadd_sat', a, b), ('bcsel', ('ult', ('iadd', a, b), a), -1, ('iadd', a, b)), 'options->lower_add_sat'),
722 (('usub_sat', a, b), ('bcsel', ('ult', a, b), 0, ('isub', a, b)), 'options->lower_add_sat'),
723
724 # Alternative lowering that doesn't rely on bfi.
725 (('bitfield_insert', 'base', 'insert', 'offset', 'bits'),
726 ('bcsel', ('ilt', 31, 'bits'),
727 'insert',
728 ('ior',
729 ('iand', 'base', ('inot', ('bfm', 'bits', 'offset'))),
730 ('iand', ('ishl', 'insert', 'offset'), ('bfm', 'bits', 'offset')))),
731 'options->lower_bitfield_insert_to_shifts'),
732
733 # bfm lowering -- note that the NIR opcode is undefined if either arg is 32.
734 (('bfm', 'bits', 'offset'),
735 ('ishl', ('isub', ('ishl', 1, 'bits'), 1), 'offset'),
736 'options->lower_bfm'),
737
738 (('ibitfield_extract', 'value', 'offset', 'bits'),
739 ('bcsel', ('ilt', 31, 'bits'), 'value',
740 ('ibfe', 'value', 'offset', 'bits')),
741 'options->lower_bitfield_extract'),
742
743 (('ubitfield_extract', 'value', 'offset', 'bits'),
744 ('bcsel', ('ult', 31, 'bits'), 'value',
745 ('ubfe', 'value', 'offset', 'bits')),
746 'options->lower_bitfield_extract'),
747
748 (('ibitfield_extract', 'value', 'offset', 'bits'),
749 ('bcsel', ('ieq', 0, 'bits'),
750 0,
751 ('ishr',
752 ('ishl', 'value', ('isub', ('isub', 32, 'bits'), 'offset')),
753 ('isub', 32, 'bits'))),
754 'options->lower_bitfield_extract_to_shifts'),
755
756 (('ubitfield_extract', 'value', 'offset', 'bits'),
757 ('iand',
758 ('ushr', 'value', 'offset'),
759 ('bcsel', ('ieq', 'bits', 32),
760 0xffffffff,
761 ('bfm', 'bits', 0))),
762 'options->lower_bitfield_extract_to_shifts'),
763
764 (('ifind_msb', 'value'),
765 ('ufind_msb', ('bcsel', ('ilt', 'value', 0), ('inot', 'value'), 'value')),
766 'options->lower_ifind_msb'),
767
768 (('find_lsb', 'value'),
769 ('ufind_msb', ('iand', 'value', ('ineg', 'value'))),
770 'options->lower_find_lsb'),
771
772 (('extract_i8', a, 'b@32'),
773 ('ishr', ('ishl', a, ('imul', ('isub', 3, b), 8)), 24),
774 'options->lower_extract_byte'),
775
776 (('extract_u8', a, 'b@32'),
777 ('iand', ('ushr', a, ('imul', b, 8)), 0xff),
778 'options->lower_extract_byte'),
779
780 (('extract_i16', a, 'b@32'),
781 ('ishr', ('ishl', a, ('imul', ('isub', 1, b), 16)), 16),
782 'options->lower_extract_word'),
783
784 (('extract_u16', a, 'b@32'),
785 ('iand', ('ushr', a, ('imul', b, 16)), 0xffff),
786 'options->lower_extract_word'),
787
788 (('pack_unorm_2x16', 'v'),
789 ('pack_uvec2_to_uint',
790 ('f2u32', ('fround_even', ('fmul', ('fsat', 'v'), 65535.0)))),
791 'options->lower_pack_unorm_2x16'),
792
793 (('pack_unorm_4x8', 'v'),
794 ('pack_uvec4_to_uint',
795 ('f2u32', ('fround_even', ('fmul', ('fsat', 'v'), 255.0)))),
796 'options->lower_pack_unorm_4x8'),
797
798 (('pack_snorm_2x16', 'v'),
799 ('pack_uvec2_to_uint',
800 ('f2i32', ('fround_even', ('fmul', ('fmin', 1.0, ('fmax', -1.0, 'v')), 32767.0)))),
801 'options->lower_pack_snorm_2x16'),
802
803 (('pack_snorm_4x8', 'v'),
804 ('pack_uvec4_to_uint',
805 ('f2i32', ('fround_even', ('fmul', ('fmin', 1.0, ('fmax', -1.0, 'v')), 127.0)))),
806 'options->lower_pack_snorm_4x8'),
807
808 (('unpack_unorm_2x16', 'v'),
809 ('fdiv', ('u2f32', ('vec2', ('extract_u16', 'v', 0),
810 ('extract_u16', 'v', 1))),
811 65535.0),
812 'options->lower_unpack_unorm_2x16'),
813
814 (('unpack_unorm_4x8', 'v'),
815 ('fdiv', ('u2f32', ('vec4', ('extract_u8', 'v', 0),
816 ('extract_u8', 'v', 1),
817 ('extract_u8', 'v', 2),
818 ('extract_u8', 'v', 3))),
819 255.0),
820 'options->lower_unpack_unorm_4x8'),
821
822 (('unpack_snorm_2x16', 'v'),
823 ('fmin', 1.0, ('fmax', -1.0, ('fdiv', ('i2f', ('vec2', ('extract_i16', 'v', 0),
824 ('extract_i16', 'v', 1))),
825 32767.0))),
826 'options->lower_unpack_snorm_2x16'),
827
828 (('unpack_snorm_4x8', 'v'),
829 ('fmin', 1.0, ('fmax', -1.0, ('fdiv', ('i2f', ('vec4', ('extract_i8', 'v', 0),
830 ('extract_i8', 'v', 1),
831 ('extract_i8', 'v', 2),
832 ('extract_i8', 'v', 3))),
833 127.0))),
834 'options->lower_unpack_snorm_4x8'),
835
836 (('isign', a), ('imin', ('imax', a, -1), 1), 'options->lower_isign'),
837 ])
838
839 # bit_size dependent lowerings
840 for bit_size in [8, 16, 32, 64]:
841 # convenience constants
842 intmax = (1 << (bit_size - 1)) - 1
843 intmin = 1 << (bit_size - 1)
844
845 optimizations += [
846 (('iadd_sat@' + str(bit_size), a, b),
847 ('bcsel', ('ige', b, 1), ('bcsel', ('ilt', ('iadd', a, b), a), intmax, ('iadd', a, b)),
848 ('bcsel', ('ilt', a, ('iadd', a, b)), intmin, ('iadd', a, b))), 'options->lower_add_sat'),
849 (('isub_sat@' + str(bit_size), a, b),
850 ('bcsel', ('ilt', b, 0), ('bcsel', ('ilt', ('isub', a, b), a), intmax, ('isub', a, b)),
851 ('bcsel', ('ilt', a, ('isub', a, b)), intmin, ('isub', a, b))), 'options->lower_add_sat'),
852 ]
853
854 invert = OrderedDict([('feq', 'fne'), ('fne', 'feq'), ('fge', 'flt'), ('flt', 'fge')])
855
856 for left, right in itertools.combinations_with_replacement(invert.keys(), 2):
857 optimizations.append((('inot', ('ior(is_used_once)', (left, a, b), (right, c, d))),
858 ('iand', (invert[left], a, b), (invert[right], c, d))))
859 optimizations.append((('inot', ('iand(is_used_once)', (left, a, b), (right, c, d))),
860 ('ior', (invert[left], a, b), (invert[right], c, d))))
861
862 # Optimize x2bN(b2x(x)) -> x
863 for size in type_sizes('bool'):
864 aN = 'a@' + str(size)
865 f2bN = 'f2b' + str(size)
866 i2bN = 'i2b' + str(size)
867 optimizations.append(((f2bN, ('b2f', aN)), a))
868 optimizations.append(((i2bN, ('b2i', aN)), a))
869
870 # Optimize x2yN(b2x(x)) -> b2y
871 for x, y in itertools.product(['f', 'u', 'i'], ['f', 'u', 'i']):
872 if x != 'f' and y != 'f' and x != y:
873 continue
874
875 b2x = 'b2f' if x == 'f' else 'b2i'
876 b2y = 'b2f' if y == 'f' else 'b2i'
877 x2yN = '{}2{}'.format(x, y)
878 optimizations.append(((x2yN, (b2x, a)), (b2y, a)))
879
880 def fexp2i(exp, bits):
881 # We assume that exp is already in the right range.
882 if bits == 16:
883 return ('i2i16', ('ishl', ('iadd', exp, 15), 10))
884 elif bits == 32:
885 return ('ishl', ('iadd', exp, 127), 23)
886 elif bits == 64:
887 return ('pack_64_2x32_split', 0, ('ishl', ('iadd', exp, 1023), 20))
888 else:
889 assert False
890
891 def ldexp(f, exp, bits):
892 # First, we clamp exp to a reasonable range. The maximum possible range
893 # for a normal exponent is [-126, 127] and, throwing in denormals, you get
894 # a maximum range of [-149, 127]. This means that we can potentially have
895 # a swing of +-276. If you start with FLT_MAX, you actually have to do
896 # ldexp(FLT_MAX, -278) to get it to flush all the way to zero. The GLSL
897 # spec, on the other hand, only requires that we handle an exponent value
898 # in the range [-126, 128]. This implementation is *mostly* correct; it
899 # handles a range on exp of [-252, 254] which allows you to create any
900 # value (including denorms if the hardware supports it) and to adjust the
901 # exponent of any normal value to anything you want.
902 if bits == 16:
903 exp = ('imin', ('imax', exp, -28), 30)
904 elif bits == 32:
905 exp = ('imin', ('imax', exp, -252), 254)
906 elif bits == 64:
907 exp = ('imin', ('imax', exp, -2044), 2046)
908 else:
909 assert False
910
911 # Now we compute two powers of 2, one for exp/2 and one for exp-exp/2.
912 # (We use ishr which isn't the same for -1, but the -1 case still works
913 # since we use exp-exp/2 as the second exponent.) While the spec
914 # technically defines ldexp as f * 2.0^exp, simply multiplying once doesn't
915 # work with denormals and doesn't allow for the full swing in exponents
916 # that you can get with normalized values. Instead, we create two powers
917 # of two and multiply by them each in turn. That way the effective range
918 # of our exponent is doubled.
919 pow2_1 = fexp2i(('ishr', exp, 1), bits)
920 pow2_2 = fexp2i(('isub', exp, ('ishr', exp, 1)), bits)
921 return ('fmul', ('fmul', f, pow2_1), pow2_2)
922
923 optimizations += [
924 (('ldexp@16', 'x', 'exp'), ldexp('x', 'exp', 16), 'options->lower_ldexp'),
925 (('ldexp@32', 'x', 'exp'), ldexp('x', 'exp', 32), 'options->lower_ldexp'),
926 (('ldexp@64', 'x', 'exp'), ldexp('x', 'exp', 64), 'options->lower_ldexp'),
927 ]
928
929 # Unreal Engine 4 demo applications open-codes bitfieldReverse()
930 def bitfield_reverse(u):
931 step1 = ('ior', ('ishl', u, 16), ('ushr', u, 16))
932 step2 = ('ior', ('ishl', ('iand', step1, 0x00ff00ff), 8), ('ushr', ('iand', step1, 0xff00ff00), 8))
933 step3 = ('ior', ('ishl', ('iand', step2, 0x0f0f0f0f), 4), ('ushr', ('iand', step2, 0xf0f0f0f0), 4))
934 step4 = ('ior', ('ishl', ('iand', step3, 0x33333333), 2), ('ushr', ('iand', step3, 0xcccccccc), 2))
935 step5 = ('ior', ('ishl', ('iand', step4, 0x55555555), 1), ('ushr', ('iand', step4, 0xaaaaaaaa), 1))
936
937 return step5
938
939 optimizations += [(bitfield_reverse('x@32'), ('bitfield_reverse', 'x'))]
940
941 # For any float comparison operation, "cmp", if you have "a == a && a cmp b"
942 # then the "a == a" is redundant because it's equivalent to "a is not NaN"
943 # and, if a is a NaN then the second comparison will fail anyway.
944 for op in ['flt', 'fge', 'feq']:
945 optimizations += [
946 (('iand', ('feq', a, a), (op, a, b)), (op, a, b)),
947 (('iand', ('feq', a, a), (op, b, a)), (op, b, a)),
948 ]
949
950 # Add optimizations to handle the case where the result of a ternary is
951 # compared to a constant. This way we can take things like
952 #
953 # (a ? 0 : 1) > 0
954 #
955 # and turn it into
956 #
957 # a ? (0 > 0) : (1 > 0)
958 #
959 # which constant folding will eat for lunch. The resulting ternary will
960 # further get cleaned up by the boolean reductions above and we will be
961 # left with just the original variable "a".
962 for op in ['flt', 'fge', 'feq', 'fne',
963 'ilt', 'ige', 'ieq', 'ine', 'ult', 'uge']:
964 optimizations += [
965 ((op, ('bcsel', 'a', '#b', '#c'), '#d'),
966 ('bcsel', 'a', (op, 'b', 'd'), (op, 'c', 'd'))),
967 ((op, '#d', ('bcsel', a, '#b', '#c')),
968 ('bcsel', 'a', (op, 'd', 'b'), (op, 'd', 'c'))),
969 ]
970
971
972 # For example, this converts things like
973 #
974 # 1 + mix(0, a - 1, condition)
975 #
976 # into
977 #
978 # mix(1, (a-1)+1, condition)
979 #
980 # Other optimizations will rearrange the constants.
981 for op in ['fadd', 'fmul', 'iadd', 'imul']:
982 optimizations += [
983 ((op, ('bcsel(is_used_once)', a, '#b', c), '#d'), ('bcsel', a, (op, b, d), (op, c, d)))
984 ]
985
986 # For derivatives in compute shaders, GLSL_NV_compute_shader_derivatives
987 # states:
988 #
989 # If neither layout qualifier is specified, derivatives in compute shaders
990 # return zero, which is consistent with the handling of built-in texture
991 # functions like texture() in GLSL 4.50 compute shaders.
992 for op in ['fddx', 'fddx_fine', 'fddx_coarse',
993 'fddy', 'fddy_fine', 'fddy_coarse']:
994 optimizations += [
995 ((op, 'a'), 0.0, 'info->stage == MESA_SHADER_COMPUTE && info->cs.derivative_group == DERIVATIVE_GROUP_NONE')
996 ]
997
998 # This section contains "late" optimizations that should be run before
999 # creating ffmas and calling regular optimizations for the final time.
1000 # Optimizations should go here if they help code generation and conflict
1001 # with the regular optimizations.
1002 before_ffma_optimizations = [
1003 # Propagate constants down multiplication chains
1004 (('~fmul(is_used_once)', ('fmul(is_used_once)', 'a(is_not_const)', '#b'), 'c(is_not_const)'), ('fmul', ('fmul', a, c), b)),
1005 (('imul(is_used_once)', ('imul(is_used_once)', 'a(is_not_const)', '#b'), 'c(is_not_const)'), ('imul', ('imul', a, c), b)),
1006 (('~fadd(is_used_once)', ('fadd(is_used_once)', 'a(is_not_const)', '#b'), 'c(is_not_const)'), ('fadd', ('fadd', a, c), b)),
1007 (('iadd(is_used_once)', ('iadd(is_used_once)', 'a(is_not_const)', '#b'), 'c(is_not_const)'), ('iadd', ('iadd', a, c), b)),
1008
1009 (('~fadd', ('fmul', a, b), ('fmul', a, c)), ('fmul', a, ('fadd', b, c))),
1010 (('iadd', ('imul', a, b), ('imul', a, c)), ('imul', a, ('iadd', b, c))),
1011 (('~fadd', ('fneg', a), a), 0.0),
1012 (('iadd', ('ineg', a), a), 0),
1013 (('iadd', ('ineg', a), ('iadd', a, b)), b),
1014 (('iadd', a, ('iadd', ('ineg', a), b)), b),
1015 (('~fadd', ('fneg', a), ('fadd', a, b)), b),
1016 (('~fadd', a, ('fadd', ('fneg', a), b)), b),
1017 ]
1018
1019 # This section contains "late" optimizations that should be run after the
1020 # regular optimizations have finished. Optimizations should go here if
1021 # they help code generation but do not necessarily produce code that is
1022 # more easily optimizable.
1023 late_optimizations = [
1024 # Most of these optimizations aren't quite safe when you get infinity or
1025 # Nan involved but the first one should be fine.
1026 (('flt', ('fadd', a, b), 0.0), ('flt', a, ('fneg', b))),
1027 (('flt', ('fneg', ('fadd', a, b)), 0.0), ('flt', ('fneg', a), b)),
1028 (('~fge', ('fadd', a, b), 0.0), ('fge', a, ('fneg', b))),
1029 (('~fge', ('fneg', ('fadd', a, b)), 0.0), ('fge', ('fneg', a), b)),
1030 (('~feq', ('fadd', a, b), 0.0), ('feq', a, ('fneg', b))),
1031 (('~fne', ('fadd', a, b), 0.0), ('fne', a, ('fneg', b))),
1032
1033 (('~fge', ('fmin(is_used_once)', ('fadd(is_used_once)', a, b), ('fadd', c, d)), 0.0), ('iand', ('fge', a, ('fneg', b)), ('fge', c, ('fneg', d)))),
1034
1035 (('fdot2', a, b), ('fdot_replicated2', a, b), 'options->fdot_replicates'),
1036 (('fdot3', a, b), ('fdot_replicated3', a, b), 'options->fdot_replicates'),
1037 (('fdot4', a, b), ('fdot_replicated4', a, b), 'options->fdot_replicates'),
1038 (('fdph', a, b), ('fdph_replicated', a, b), 'options->fdot_replicates'),
1039
1040 (('b2f(is_used_more_than_once)', ('inot', 'a@1')), ('bcsel', a, 0.0, 1.0)),
1041 (('fneg(is_used_more_than_once)', ('b2f', ('inot', 'a@1'))), ('bcsel', a, -0.0, -1.0)),
1042
1043 # we do these late so that we don't get in the way of creating ffmas
1044 (('fmin', ('fadd(is_used_once)', '#c', a), ('fadd(is_used_once)', '#c', b)), ('fadd', c, ('fmin', a, b))),
1045 (('fmax', ('fadd(is_used_once)', '#c', a), ('fadd(is_used_once)', '#c', b)), ('fadd', c, ('fmax', a, b))),
1046
1047 (('bcsel', a, 0, ('b2f32', ('inot', 'b@bool'))), ('b2f32', ('inot', ('ior', a, b)))),
1048 ]
1049
1050 print(nir_algebraic.AlgebraicPass("nir_opt_algebraic", optimizations).render())
1051 print(nir_algebraic.AlgebraicPass("nir_opt_algebraic_before_ffma",
1052 before_ffma_optimizations).render())
1053 print(nir_algebraic.AlgebraicPass("nir_opt_algebraic_late",
1054 late_optimizations).render())