2 * Copyright © 2019 Google, 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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21 * DEALINGS IN THE SOFTWARE.
25 * \file lower_precision.cpp
28 #include "main/macros.h"
29 #include "main/mtypes.h"
30 #include "compiler/glsl_types.h"
32 #include "ir_builder.h"
33 #include "ir_optimization.h"
34 #include "ir_rvalue_visitor.h"
35 #include "util/half_float.h"
37 #include "util/hash_table.h"
42 class find_precision_visitor
: public ir_rvalue_enter_visitor
{
44 find_precision_visitor(const struct gl_shader_compiler_options
*options
);
45 ~find_precision_visitor();
47 virtual void handle_rvalue(ir_rvalue
**rvalue
);
48 virtual ir_visitor_status
visit_enter(ir_call
*ir
);
50 ir_function_signature
*map_builtin(ir_function_signature
*sig
);
52 /* Set of rvalues that can be lowered. This will be filled in by
53 * find_lowerable_rvalues_visitor. Only the root node of a lowerable section
54 * will be added to this set.
56 struct set
*lowerable_rvalues
;
59 * A mapping of builtin signature functions to lowered versions. This is
60 * filled in lazily when a lowered version is needed.
62 struct hash_table
*lowered_builtins
;
64 * A temporary hash table only used in order to clone functions.
66 struct hash_table
*clone_ht
;
68 void *lowered_builtin_mem_ctx
;
70 const struct gl_shader_compiler_options
*options
;
73 class find_lowerable_rvalues_visitor
: public ir_hierarchical_visitor
{
75 enum can_lower_state
{
81 enum parent_relation
{
82 /* The parent performs a further operation involving the result from the
83 * child and can be lowered along with it.
86 /* The parent instruction’s operation is independent of the child type so
87 * the child should be lowered separately.
89 INDEPENDENT_OPERATION
,
93 ir_instruction
*instr
;
94 enum can_lower_state state
;
95 /* List of child rvalues that can be lowered. When this stack entry is
96 * popped, if this node itself can’t be lowered than all of the children
97 * are root nodes to lower so we will add them to lowerable_rvalues.
98 * Otherwise if this node can also be lowered then we won’t add the
99 * children because we only want to add the topmost lowerable nodes to
100 * lowerable_rvalues and the children will be lowered as part of lowering
103 std::vector
<ir_instruction
*> lowerable_children
;
106 find_lowerable_rvalues_visitor(struct set
*result
,
107 const struct gl_shader_compiler_options
*options
);
109 static void stack_enter(class ir_instruction
*ir
, void *data
);
110 static void stack_leave(class ir_instruction
*ir
, void *data
);
112 virtual ir_visitor_status
visit(ir_constant
*ir
);
113 virtual ir_visitor_status
visit(ir_dereference_variable
*ir
);
115 virtual ir_visitor_status
visit_enter(ir_dereference_record
*ir
);
116 virtual ir_visitor_status
visit_enter(ir_dereference_array
*ir
);
117 virtual ir_visitor_status
visit_enter(ir_texture
*ir
);
118 virtual ir_visitor_status
visit_enter(ir_expression
*ir
);
120 virtual ir_visitor_status
visit_leave(ir_assignment
*ir
);
121 virtual ir_visitor_status
visit_leave(ir_call
*ir
);
123 can_lower_state
handle_precision(const glsl_type
*type
,
124 int precision
) const;
126 static parent_relation
get_parent_relation(ir_instruction
*parent
,
127 ir_instruction
*child
);
129 std::vector
<stack_entry
> stack
;
130 struct set
*lowerable_rvalues
;
131 const struct gl_shader_compiler_options
*options
;
133 void pop_stack_entry();
134 void add_lowerable_children(const stack_entry
&entry
);
137 class lower_precision_visitor
: public ir_rvalue_visitor
{
139 virtual void handle_rvalue(ir_rvalue
**rvalue
);
140 virtual ir_visitor_status
visit_enter(ir_dereference_array
*);
141 virtual ir_visitor_status
visit_enter(ir_dereference_record
*);
142 virtual ir_visitor_status
visit_enter(ir_call
*ir
);
143 virtual ir_visitor_status
visit_enter(ir_texture
*ir
);
144 virtual ir_visitor_status
visit_leave(ir_expression
*);
148 can_lower_type(const struct gl_shader_compiler_options
*options
,
149 const glsl_type
*type
)
151 /* Don’t lower any expressions involving non-float types except bool and
152 * texture samplers. This will rule out operations that change the type such
153 * as conversion to ints. Instead it will end up lowering the arguments
154 * instead and adding a final conversion to float32. We want to handle
155 * boolean types so that it will do comparisons as 16-bit.
158 switch (type
->without_array()->base_type
) {
159 /* TODO: should we do anything for these two with regard to Int16 vs FP16
163 case GLSL_TYPE_SAMPLER
:
164 case GLSL_TYPE_IMAGE
:
167 case GLSL_TYPE_FLOAT
:
168 return options
->LowerPrecisionFloat16
;
172 return options
->LowerPrecisionInt16
;
179 find_lowerable_rvalues_visitor::find_lowerable_rvalues_visitor(struct set
*res
,
180 const struct gl_shader_compiler_options
*opts
)
182 lowerable_rvalues
= res
;
184 callback_enter
= stack_enter
;
185 callback_leave
= stack_leave
;
191 find_lowerable_rvalues_visitor::stack_enter(class ir_instruction
*ir
,
194 find_lowerable_rvalues_visitor
*state
=
195 (find_lowerable_rvalues_visitor
*) data
;
197 /* Add a new stack entry for this instruction */
201 entry
.state
= state
->in_assignee
? CANT_LOWER
: UNKNOWN
;
203 state
->stack
.push_back(entry
);
207 find_lowerable_rvalues_visitor::add_lowerable_children(const stack_entry
&entry
)
209 /* We can’t lower this node so if there were any pending children then they
210 * are all root lowerable nodes and we should add them to the set.
212 for (auto &it
: entry
.lowerable_children
)
213 _mesa_set_add(lowerable_rvalues
, it
);
217 find_lowerable_rvalues_visitor::pop_stack_entry()
219 const stack_entry
&entry
= stack
.back();
221 if (stack
.size() >= 2) {
222 /* Combine this state into the parent state, unless the parent operation
223 * doesn’t have any relation to the child operations
225 stack_entry
&parent
= stack
.end()[-2];
226 parent_relation rel
= get_parent_relation(parent
.instr
, entry
.instr
);
228 if (rel
== COMBINED_OPERATION
) {
229 switch (entry
.state
) {
231 parent
.state
= CANT_LOWER
;
234 if (parent
.state
== UNKNOWN
)
235 parent
.state
= SHOULD_LOWER
;
243 if (entry
.state
== SHOULD_LOWER
) {
244 ir_rvalue
*rv
= entry
.instr
->as_rvalue();
247 add_lowerable_children(entry
);
248 } else if (stack
.size() >= 2) {
249 stack_entry
&parent
= stack
.end()[-2];
251 switch (get_parent_relation(parent
.instr
, rv
)) {
252 case COMBINED_OPERATION
:
253 /* We only want to add the toplevel lowerable instructions to the
254 * lowerable set. Therefore if there is a parent then instead of
255 * adding this instruction to the set we will queue depending on
256 * the result of the parent instruction.
258 parent
.lowerable_children
.push_back(entry
.instr
);
260 case INDEPENDENT_OPERATION
:
261 _mesa_set_add(lowerable_rvalues
, rv
);
265 /* This is a toplevel node so add it directly to the lowerable
268 _mesa_set_add(lowerable_rvalues
, rv
);
270 } else if (entry
.state
== CANT_LOWER
) {
271 add_lowerable_children(entry
);
278 find_lowerable_rvalues_visitor::stack_leave(class ir_instruction
*ir
,
281 find_lowerable_rvalues_visitor
*state
=
282 (find_lowerable_rvalues_visitor
*) data
;
284 state
->pop_stack_entry();
287 enum find_lowerable_rvalues_visitor::can_lower_state
288 find_lowerable_rvalues_visitor::handle_precision(const glsl_type
*type
,
291 if (!can_lower_type(options
, type
))
295 case GLSL_PRECISION_NONE
:
297 case GLSL_PRECISION_HIGH
:
299 case GLSL_PRECISION_MEDIUM
:
300 case GLSL_PRECISION_LOW
:
307 enum find_lowerable_rvalues_visitor::parent_relation
308 find_lowerable_rvalues_visitor::get_parent_relation(ir_instruction
*parent
,
309 ir_instruction
*child
)
311 /* If the parent is a dereference instruction then the only child could be
312 * for example an array dereference and that should be lowered independently
315 if (parent
->as_dereference())
316 return INDEPENDENT_OPERATION
;
318 /* The precision of texture sampling depend on the precision of the sampler.
319 * The rest of the arguments don’t matter so we can treat it as an
320 * independent operation.
322 if (parent
->as_texture())
323 return INDEPENDENT_OPERATION
;
325 return COMBINED_OPERATION
;
329 find_lowerable_rvalues_visitor::visit(ir_constant
*ir
)
331 stack_enter(ir
, this);
333 if (!can_lower_type(options
, ir
->type
))
334 stack
.back().state
= CANT_LOWER
;
336 stack_leave(ir
, this);
338 return visit_continue
;
342 find_lowerable_rvalues_visitor::visit(ir_dereference_variable
*ir
)
344 stack_enter(ir
, this);
346 if (stack
.back().state
== UNKNOWN
)
347 stack
.back().state
= handle_precision(ir
->type
, ir
->precision());
349 stack_leave(ir
, this);
351 return visit_continue
;
355 find_lowerable_rvalues_visitor::visit_enter(ir_dereference_record
*ir
)
357 ir_hierarchical_visitor::visit_enter(ir
);
359 if (stack
.back().state
== UNKNOWN
)
360 stack
.back().state
= handle_precision(ir
->type
, ir
->precision());
362 return visit_continue
;
366 find_lowerable_rvalues_visitor::visit_enter(ir_dereference_array
*ir
)
368 ir_hierarchical_visitor::visit_enter(ir
);
370 if (stack
.back().state
== UNKNOWN
)
371 stack
.back().state
= handle_precision(ir
->type
, ir
->precision());
373 return visit_continue
;
377 find_lowerable_rvalues_visitor::visit_enter(ir_texture
*ir
)
379 ir_hierarchical_visitor::visit_enter(ir
);
381 /* The precision of the sample value depends on the precision of the
384 stack
.back().state
= handle_precision(ir
->type
,
385 ir
->sampler
->precision());
386 return visit_continue
;
390 find_lowerable_rvalues_visitor::visit_enter(ir_expression
*ir
)
392 ir_hierarchical_visitor::visit_enter(ir
);
394 if (!can_lower_type(options
, ir
->type
))
395 stack
.back().state
= CANT_LOWER
;
397 /* Don't lower precision for derivative calculations */
398 if (!options
->LowerPrecisionDerivatives
&&
399 (ir
->operation
== ir_unop_dFdx
||
400 ir
->operation
== ir_unop_dFdx_coarse
||
401 ir
->operation
== ir_unop_dFdx_fine
||
402 ir
->operation
== ir_unop_dFdy
||
403 ir
->operation
== ir_unop_dFdy_coarse
||
404 ir
->operation
== ir_unop_dFdy_fine
)) {
405 stack
.back().state
= CANT_LOWER
;
408 return visit_continue
;
412 function_always_returns_mediump_or_lowp(const char *name
)
414 return !strcmp(name
, "bitCount") ||
415 !strcmp(name
, "findLSB") ||
416 !strcmp(name
, "findMSB") ||
417 !strcmp(name
, "unpackHalf2x16") ||
418 !strcmp(name
, "unpackUnorm4x8") ||
419 !strcmp(name
, "unpackSnorm4x8");
423 is_lowerable_builtin(ir_call
*ir
,
424 const struct set
*lowerable_rvalues
)
426 /* The intrinsic call is inside the wrapper imageLoad function that will
427 * be inlined. We have to handle both of them.
429 if (ir
->callee
->intrinsic_id
== ir_intrinsic_image_load
||
430 (ir
->callee
->is_builtin() &&
431 !strcmp(ir
->callee_name(), "imageLoad"))) {
432 ir_rvalue
*param
= (ir_rvalue
*)ir
->actual_parameters
.get_head();
433 ir_variable
*resource
= param
->variable_referenced();
435 assert(ir
->callee
->return_precision
== GLSL_PRECISION_NONE
);
436 assert(resource
->type
->without_array()->is_image());
438 /* GLSL ES 3.20 requires that images have a precision modifier, but if
439 * you set one, it doesn't do anything, because all intrinsics are
440 * defined with highp. This seems to be a spec bug.
442 * In theory we could set the return value to mediump if the image
443 * format has a lower precision. This appears to be the most sensible
446 const struct util_format_description
*desc
=
447 util_format_description(resource
->data
.image_format
);
449 util_format_get_first_non_void_channel(resource
->data
.image_format
);
452 if (desc
->channel
[i
].pure_integer
||
453 desc
->channel
[i
].type
== UTIL_FORMAT_TYPE_FLOAT
)
454 return desc
->channel
[i
].size
<= 16;
456 return desc
->channel
[i
].size
<= 10; /* unorm/snorm */
459 /* Handle special calls. */
460 if (ir
->callee
->is_builtin() && ir
->actual_parameters
.length()) {
461 ir_rvalue
*param
= (ir_rvalue
*)ir
->actual_parameters
.get_head();
462 ir_variable
*var
= param
->variable_referenced();
464 /* Handle builtin wrappers around ir_texture opcodes. These wrappers will
465 * be inlined by lower_precision() if we return true here, so that we can
466 * get to ir_texture later and do proper lowering.
468 * We should lower the type of the return value if the sampler type
469 * uses lower precision. The function parameters don't matter.
471 if (var
&& var
->type
->without_array()->is_sampler()) {
472 /* textureSize always returns highp. */
473 if (!strcmp(ir
->callee_name(), "textureSize"))
476 return var
->data
.precision
== GLSL_PRECISION_MEDIUM
||
477 var
->data
.precision
== GLSL_PRECISION_LOW
;
481 if (!ir
->callee
->is_builtin() ||
482 /* Parameters are always highp: */
483 !strcmp(ir
->callee_name(), "floatBitsToInt") ||
484 !strcmp(ir
->callee_name(), "floatBitsToUint") ||
485 !strcmp(ir
->callee_name(), "intBitsToFloat") ||
486 !strcmp(ir
->callee_name(), "uintBitsToFloat") ||
487 !strcmp(ir
->callee_name(), "bitfieldReverse") ||
488 !strcmp(ir
->callee_name(), "frexp") ||
489 !strcmp(ir
->callee_name(), "ldexp") ||
490 /* Parameters and outputs are always highp: */
491 /* TODO: The operations are highp, but carry and borrow outputs are lowp. */
492 !strcmp(ir
->callee_name(), "uaddCarry") ||
493 !strcmp(ir
->callee_name(), "usubBorrow") ||
494 !strcmp(ir
->callee_name(), "imulExtended") ||
495 !strcmp(ir
->callee_name(), "umulExtended") ||
496 !strcmp(ir
->callee_name(), "unpackUnorm2x16") ||
497 !strcmp(ir
->callee_name(), "unpackSnorm2x16") ||
498 /* Outputs are highp: */
499 !strcmp(ir
->callee_name(), "packUnorm2x16") ||
500 !strcmp(ir
->callee_name(), "packSnorm2x16") ||
501 /* Parameters are mediump and outputs are highp. The parameters should
502 * be optimized in NIR, not here, e.g:
503 * - packHalf2x16 can just be a bitcast from f16vec2 to uint32
504 * - Other opcodes don't have to convert parameters to highp if the hw
505 * has f16 versions. Optimize in NIR accordingly.
507 !strcmp(ir
->callee_name(), "packHalf2x16") ||
508 !strcmp(ir
->callee_name(), "packUnorm4x8") ||
509 !strcmp(ir
->callee_name(), "packSnorm4x8") ||
510 /* Atomic functions are not lowered. */
511 strstr(ir
->callee_name(), "atomic") == ir
->callee_name())
514 assert(ir
->callee
->return_precision
== GLSL_PRECISION_NONE
);
516 /* Number of parameters to check if they are lowerable. */
517 unsigned check_parameters
= ir
->actual_parameters
.length();
519 /* Interpolation functions only consider the precision of the interpolant. */
520 /* Bitfield functions ignore the precision of "offset" and "bits". */
521 if (!strcmp(ir
->callee_name(), "interpolateAtOffset") ||
522 !strcmp(ir
->callee_name(), "interpolateAtSample") ||
523 !strcmp(ir
->callee_name(), "bitfieldExtract")) {
524 check_parameters
= 1;
525 } else if (!strcmp(ir
->callee_name(), "bitfieldInsert")) {
526 check_parameters
= 2;
527 } if (function_always_returns_mediump_or_lowp(ir
->callee_name())) {
528 /* These only lower the return value. Parameters keep their precision,
529 * which is preserved in map_builtin.
531 check_parameters
= 0;
534 foreach_in_list(ir_rvalue
, param
, &ir
->actual_parameters
) {
535 if (!check_parameters
)
538 if (!param
->as_constant() &&
539 _mesa_set_search(lowerable_rvalues
, param
) == NULL
)
549 find_lowerable_rvalues_visitor::visit_leave(ir_call
*ir
)
551 ir_hierarchical_visitor::visit_leave(ir
);
553 /* Special case for handling temporary variables generated by the compiler
554 * for function calls. If we assign to one of these using a function call
555 * that has a lowerable return type then we can assume the temporary
556 * variable should have a medium precision too.
559 /* Do nothing if the return type is void. */
560 if (!ir
->return_deref
)
561 return visit_continue
;
563 ir_variable
*var
= ir
->return_deref
->variable_referenced();
565 assert(var
->data
.mode
== ir_var_temporary
);
567 unsigned return_precision
= ir
->callee
->return_precision
;
569 /* If the call is to a builtin, then the function won’t have a return
570 * precision and we should determine it from the precision of the arguments.
572 if (is_lowerable_builtin(ir
, lowerable_rvalues
))
573 return_precision
= GLSL_PRECISION_MEDIUM
;
575 can_lower_state lower_state
=
576 handle_precision(var
->type
, return_precision
);
578 if (lower_state
== SHOULD_LOWER
) {
579 /* There probably shouldn’t be any situations where multiple ir_call
580 * instructions write to the same temporary?
582 assert(var
->data
.precision
== GLSL_PRECISION_NONE
);
583 var
->data
.precision
= GLSL_PRECISION_MEDIUM
;
585 var
->data
.precision
= GLSL_PRECISION_HIGH
;
588 return visit_continue
;
592 find_lowerable_rvalues_visitor::visit_leave(ir_assignment
*ir
)
594 ir_hierarchical_visitor::visit_leave(ir
);
596 /* Special case for handling temporary variables generated by the compiler.
597 * If we assign to one of these using a lowered precision then we can assume
598 * the temporary variable should have a medium precision too.
600 ir_variable
*var
= ir
->lhs
->variable_referenced();
602 if (var
->data
.mode
== ir_var_temporary
) {
603 if (_mesa_set_search(lowerable_rvalues
, ir
->rhs
)) {
604 /* Only override the precision if this is the first assignment. For
605 * temporaries such as the ones generated for the ?: operator there
606 * can be multiple assignments with different precisions. This way we
607 * get the highest precision of all of the assignments.
609 if (var
->data
.precision
== GLSL_PRECISION_NONE
)
610 var
->data
.precision
= GLSL_PRECISION_MEDIUM
;
611 } else if (!ir
->rhs
->as_constant()) {
612 var
->data
.precision
= GLSL_PRECISION_HIGH
;
616 return visit_continue
;
620 find_lowerable_rvalues(const struct gl_shader_compiler_options
*options
,
621 exec_list
*instructions
,
624 find_lowerable_rvalues_visitor
v(result
, options
);
626 visit_list_elements(&v
, instructions
);
628 assert(v
.stack
.empty());
631 static const glsl_type
*
632 convert_type(bool up
, const glsl_type
*type
)
634 if (type
->is_array()) {
635 return glsl_type::get_array_instance(convert_type(up
, type
->fields
.array
),
637 type
->explicit_stride
);
640 glsl_base_type new_base_type
;
643 switch (type
->base_type
) {
644 case GLSL_TYPE_FLOAT16
:
645 new_base_type
= GLSL_TYPE_FLOAT
;
647 case GLSL_TYPE_INT16
:
648 new_base_type
= GLSL_TYPE_INT
;
650 case GLSL_TYPE_UINT16
:
651 new_base_type
= GLSL_TYPE_UINT
;
654 unreachable("invalid type");
658 switch (type
->base_type
) {
659 case GLSL_TYPE_FLOAT
:
660 new_base_type
= GLSL_TYPE_FLOAT16
;
663 new_base_type
= GLSL_TYPE_INT16
;
666 new_base_type
= GLSL_TYPE_UINT16
;
669 unreachable("invalid type");
674 return glsl_type::get_instance(new_base_type
,
675 type
->vector_elements
,
676 type
->matrix_columns
,
677 type
->explicit_stride
,
678 type
->interface_row_major
);
681 static const glsl_type
*
682 lower_glsl_type(const glsl_type
*type
)
684 return convert_type(false, type
);
688 convert_precision(bool up
, ir_rvalue
*ir
)
693 switch (ir
->type
->base_type
) {
694 case GLSL_TYPE_FLOAT16
:
697 case GLSL_TYPE_INT16
:
700 case GLSL_TYPE_UINT16
:
704 unreachable("invalid type");
708 switch (ir
->type
->base_type
) {
709 case GLSL_TYPE_FLOAT
:
719 unreachable("invalid type");
724 const glsl_type
*desired_type
= convert_type(up
, ir
->type
);
725 void *mem_ctx
= ralloc_parent(ir
);
726 return new(mem_ctx
) ir_expression(op
, desired_type
, ir
, NULL
);
730 lower_precision_visitor::handle_rvalue(ir_rvalue
**rvalue
)
732 ir_rvalue
*ir
= *rvalue
;
737 if (ir
->as_dereference()) {
738 if (!ir
->type
->is_boolean())
739 *rvalue
= convert_precision(false, ir
);
740 } else if (ir
->type
->is_32bit()) {
741 ir
->type
= lower_glsl_type(ir
->type
);
743 ir_constant
*const_ir
= ir
->as_constant();
746 ir_constant_data value
;
748 if (ir
->type
->base_type
== GLSL_TYPE_FLOAT16
) {
749 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.f16
); i
++)
750 value
.f16
[i
] = _mesa_float_to_half(const_ir
->value
.f
[i
]);
751 } else if (ir
->type
->base_type
== GLSL_TYPE_INT16
) {
752 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.i16
); i
++)
753 value
.i16
[i
] = const_ir
->value
.i
[i
];
754 } else if (ir
->type
->base_type
== GLSL_TYPE_UINT16
) {
755 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.u16
); i
++)
756 value
.u16
[i
] = const_ir
->value
.u
[i
];
758 unreachable("invalid type");
761 const_ir
->value
= value
;
767 lower_precision_visitor::visit_enter(ir_dereference_record
*ir
)
769 /* We don’t want to lower the variable */
770 return visit_continue_with_parent
;
774 lower_precision_visitor::visit_enter(ir_dereference_array
*ir
)
776 /* We don’t want to convert the array index or the variable. If the array
777 * index itself is lowerable that will be handled separately.
779 return visit_continue_with_parent
;
783 lower_precision_visitor::visit_enter(ir_call
*ir
)
785 /* We don’t want to convert the arguments. These will be handled separately.
787 return visit_continue_with_parent
;
791 lower_precision_visitor::visit_enter(ir_texture
*ir
)
793 /* We don’t want to convert the arguments. These will be handled separately.
795 return visit_continue_with_parent
;
799 lower_precision_visitor::visit_leave(ir_expression
*ir
)
801 ir_rvalue_visitor::visit_leave(ir
);
803 /* If the expression is a conversion operation to or from bool then fix the
806 switch (ir
->operation
) {
808 ir
->operation
= ir_unop_b2f16
;
811 ir
->operation
= ir_unop_f162b
;
815 /* Nothing to do - they both support int16. */
821 return visit_continue
;
825 find_precision_visitor::handle_rvalue(ir_rvalue
**rvalue
)
827 /* Checking the precision of rvalue can be lowered first throughout
828 * find_lowerable_rvalues_visitor.
829 * Once it found the precision of rvalue can be lowered, then we can
830 * add conversion f2fmp, etc. through lower_precision_visitor.
835 struct set_entry
*entry
= _mesa_set_search(lowerable_rvalues
, *rvalue
);
840 _mesa_set_remove(lowerable_rvalues
, entry
);
842 /* If the entire expression is just a variable dereference then trying to
843 * lower it will just directly add pointless to and from conversions without
844 * any actual operation in-between. Although these will eventually get
845 * optimised out, avoiding generating them here also avoids breaking inout
846 * parameters to functions.
848 if ((*rvalue
)->as_dereference())
851 lower_precision_visitor v
;
853 (*rvalue
)->accept(&v
);
854 v
.handle_rvalue(rvalue
);
856 /* We don’t need to add the final conversion if the final type has been
859 if ((*rvalue
)->type
->base_type
!= GLSL_TYPE_BOOL
) {
860 *rvalue
= convert_precision(true, *rvalue
);
865 find_precision_visitor::visit_enter(ir_call
*ir
)
867 ir_rvalue_enter_visitor::visit_enter(ir
);
869 ir_variable
*return_var
=
870 ir
->return_deref
? ir
->return_deref
->variable_referenced() : NULL
;
872 /* Don't do anything for image_load here. We have only changed the return
873 * value to mediump/lowp, so that following instructions can use reduced
876 * The return value type of the intrinsic itself isn't changed here, but
877 * can be changed in NIR if all users use the *2*mp opcode.
879 if (ir
->callee
->intrinsic_id
== ir_intrinsic_image_load
)
880 return visit_continue
;
882 /* If this is a call to a builtin and the find_lowerable_rvalues_visitor
883 * overrode the precision of the temporary return variable, then we can
884 * replace the builtin implementation with a lowered version.
887 if (!ir
->callee
->is_builtin() ||
888 ir
->callee
->is_intrinsic() ||
889 return_var
== NULL
||
890 (return_var
->data
.precision
!= GLSL_PRECISION_MEDIUM
&&
891 return_var
->data
.precision
!= GLSL_PRECISION_LOW
))
892 return visit_continue
;
894 ir
->callee
= map_builtin(ir
->callee
);
895 ir
->generate_inline(ir
);
898 return visit_continue_with_parent
;
901 ir_function_signature
*
902 find_precision_visitor::map_builtin(ir_function_signature
*sig
)
904 if (lowered_builtins
== NULL
) {
905 lowered_builtins
= _mesa_pointer_hash_table_create(NULL
);
906 clone_ht
=_mesa_pointer_hash_table_create(NULL
);
907 lowered_builtin_mem_ctx
= ralloc_context(NULL
);
909 struct hash_entry
*entry
= _mesa_hash_table_search(lowered_builtins
, sig
);
911 return (ir_function_signature
*) entry
->data
;
914 ir_function_signature
*lowered_sig
=
915 sig
->clone(lowered_builtin_mem_ctx
, clone_ht
);
917 /* Functions that always return mediump or lowp should keep their
918 * parameters intact, because they can be highp. NIR can lower
919 * the up-conversion for parameters if needed.
921 if (!function_always_returns_mediump_or_lowp(sig
->function_name())) {
922 foreach_in_list(ir_variable
, param
, &lowered_sig
->parameters
) {
923 param
->data
.precision
= GLSL_PRECISION_MEDIUM
;
927 lower_precision(options
, &lowered_sig
->body
);
929 _mesa_hash_table_clear(clone_ht
, NULL
);
931 _mesa_hash_table_insert(lowered_builtins
, sig
, lowered_sig
);
936 find_precision_visitor::find_precision_visitor(const struct gl_shader_compiler_options
*options
)
937 : lowerable_rvalues(_mesa_pointer_set_create(NULL
)),
938 lowered_builtins(NULL
),
940 lowered_builtin_mem_ctx(NULL
),
945 find_precision_visitor::~find_precision_visitor()
947 _mesa_set_destroy(lowerable_rvalues
, NULL
);
949 if (lowered_builtins
) {
950 _mesa_hash_table_destroy(lowered_builtins
, NULL
);
951 _mesa_hash_table_destroy(clone_ht
, NULL
);
952 ralloc_free(lowered_builtin_mem_ctx
);
956 /* Lowering opcodes to 16 bits is not enough for programs with control flow
957 * (and the ?: operator, which is represented by if-then-else in the IR),
958 * because temporary variables, which are used for passing values between
959 * code blocks, are not lowered, resulting in 32-bit phis in NIR.
961 * First change the variable types to 16 bits, then change all ir_dereference
964 class lower_variables_visitor
: public ir_rvalue_enter_visitor
{
966 lower_variables_visitor(const struct gl_shader_compiler_options
*options
)
968 lower_vars
= _mesa_pointer_set_create(NULL
);
971 virtual ~lower_variables_visitor()
973 _mesa_set_destroy(lower_vars
, NULL
);
976 virtual ir_visitor_status
visit(ir_variable
*var
);
977 virtual ir_visitor_status
visit_enter(ir_assignment
*ir
);
978 virtual ir_visitor_status
visit_enter(ir_return
*ir
);
979 virtual ir_visitor_status
visit_enter(ir_call
*ir
);
980 virtual void handle_rvalue(ir_rvalue
**rvalue
);
982 void fix_types_in_deref_chain(ir_dereference
*ir
);
983 void convert_split_assignment(ir_dereference
*lhs
, ir_rvalue
*rhs
,
986 const struct gl_shader_compiler_options
*options
;
991 lower_constant(ir_constant
*ir
)
993 if (ir
->type
->is_array()) {
994 for (int i
= 0; i
< ir
->type
->array_size(); i
++)
995 lower_constant(ir
->get_array_element(i
));
997 ir
->type
= lower_glsl_type(ir
->type
);
1001 ir
->type
= lower_glsl_type(ir
->type
);
1002 ir_constant_data value
;
1004 if (ir
->type
->base_type
== GLSL_TYPE_FLOAT16
) {
1005 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.f16
); i
++)
1006 value
.f16
[i
] = _mesa_float_to_half(ir
->value
.f
[i
]);
1007 } else if (ir
->type
->base_type
== GLSL_TYPE_INT16
) {
1008 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.i16
); i
++)
1009 value
.i16
[i
] = ir
->value
.i
[i
];
1010 } else if (ir
->type
->base_type
== GLSL_TYPE_UINT16
) {
1011 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.u16
); i
++)
1012 value
.u16
[i
] = ir
->value
.u
[i
];
1014 unreachable("invalid type");
1021 lower_variables_visitor::visit(ir_variable
*var
)
1023 if ((var
->data
.mode
!= ir_var_temporary
&&
1024 var
->data
.mode
!= ir_var_auto
) ||
1025 !var
->type
->without_array()->is_32bit() ||
1026 (var
->data
.precision
!= GLSL_PRECISION_MEDIUM
&&
1027 var
->data
.precision
!= GLSL_PRECISION_LOW
) ||
1028 !can_lower_type(options
, var
->type
))
1029 return visit_continue
;
1031 /* Lower constant initializers. */
1032 if (var
->constant_value
&&
1033 var
->type
== var
->constant_value
->type
) {
1034 if (!options
->LowerPrecisionConstants
)
1035 return visit_continue
;
1036 var
->constant_value
=
1037 var
->constant_value
->clone(ralloc_parent(var
), NULL
);
1038 lower_constant(var
->constant_value
);
1041 if (var
->constant_initializer
&&
1042 var
->type
== var
->constant_initializer
->type
) {
1043 if (!options
->LowerPrecisionConstants
)
1044 return visit_continue
;
1045 var
->constant_initializer
=
1046 var
->constant_initializer
->clone(ralloc_parent(var
), NULL
);
1047 lower_constant(var
->constant_initializer
);
1050 var
->type
= lower_glsl_type(var
->type
);
1051 _mesa_set_add(lower_vars
, var
);
1053 return visit_continue
;
1057 lower_variables_visitor::fix_types_in_deref_chain(ir_dereference
*ir
)
1059 assert(ir
->type
->without_array()->is_32bit());
1060 assert(_mesa_set_search(lower_vars
, ir
->variable_referenced()));
1062 /* Fix the type in the dereference node. */
1063 ir
->type
= lower_glsl_type(ir
->type
);
1065 /* If it's an array, fix the types in the whole dereference chain. */
1066 for (ir_dereference_array
*deref_array
= ir
->as_dereference_array();
1068 deref_array
= deref_array
->array
->as_dereference_array()) {
1069 assert(deref_array
->array
->type
->without_array()->is_32bit());
1070 deref_array
->array
->type
= lower_glsl_type(deref_array
->array
->type
);
1075 lower_variables_visitor::convert_split_assignment(ir_dereference
*lhs
,
1079 void *mem_ctx
= ralloc_parent(lhs
);
1081 if (lhs
->type
->is_array()) {
1082 for (unsigned i
= 0; i
< lhs
->type
->length
; i
++) {
1083 ir_dereference
*l
, *r
;
1085 l
= new(mem_ctx
) ir_dereference_array(lhs
->clone(mem_ctx
, NULL
),
1086 new(mem_ctx
) ir_constant(i
));
1087 r
= new(mem_ctx
) ir_dereference_array(rhs
->clone(mem_ctx
, NULL
),
1088 new(mem_ctx
) ir_constant(i
));
1089 convert_split_assignment(l
, r
, insert_before
);
1094 assert(lhs
->type
->is_16bit() || lhs
->type
->is_32bit());
1095 assert(rhs
->type
->is_16bit() || rhs
->type
->is_32bit());
1096 assert(lhs
->type
->is_16bit() != rhs
->type
->is_16bit());
1098 ir_assignment
*assign
=
1099 new(mem_ctx
) ir_assignment(lhs
, convert_precision(lhs
->type
->is_32bit(), rhs
));
1102 base_ir
->insert_before(assign
);
1104 base_ir
->insert_after(assign
);
1108 lower_variables_visitor::visit_enter(ir_assignment
*ir
)
1110 ir_dereference
*lhs
= ir
->lhs
;
1111 ir_variable
*var
= lhs
->variable_referenced();
1112 ir_dereference
*rhs_deref
= ir
->rhs
->as_dereference();
1113 ir_variable
*rhs_var
= rhs_deref
? rhs_deref
->variable_referenced() : NULL
;
1114 ir_constant
*rhs_const
= ir
->rhs
->as_constant();
1116 /* Legalize array assignments between lowered and non-lowered variables. */
1117 if (lhs
->type
->is_array() &&
1118 (rhs_var
|| rhs_const
) &&
1121 var
->type
->without_array()->is_16bit() !=
1122 rhs_var
->type
->without_array()->is_16bit())) &&
1125 var
->type
->without_array()->is_16bit() &&
1126 rhs_const
->type
->without_array()->is_32bit()))) {
1127 assert(ir
->rhs
->type
->is_array());
1129 /* Fix array assignments from lowered to non-lowered. */
1130 if (rhs_var
&& _mesa_set_search(lower_vars
, rhs_var
)) {
1131 fix_types_in_deref_chain(rhs_deref
);
1132 /* Convert to 32 bits for LHS. */
1133 convert_split_assignment(lhs
, rhs_deref
, true);
1135 return visit_continue
;
1138 /* Fix array assignments from non-lowered to lowered. */
1140 _mesa_set_search(lower_vars
, var
) &&
1141 ir
->rhs
->type
->without_array()->is_32bit()) {
1142 fix_types_in_deref_chain(lhs
);
1143 /* Convert to 16 bits for LHS. */
1144 convert_split_assignment(lhs
, ir
->rhs
, true);
1146 return visit_continue
;
1150 /* Fix assignment types. */
1152 _mesa_set_search(lower_vars
, var
)) {
1153 /* Fix the LHS type. */
1154 if (lhs
->type
->without_array()->is_32bit())
1155 fix_types_in_deref_chain(lhs
);
1157 /* Fix the RHS type if it's a lowered variable. */
1159 _mesa_set_search(lower_vars
, rhs_var
) &&
1160 rhs_deref
->type
->without_array()->is_32bit())
1161 fix_types_in_deref_chain(rhs_deref
);
1163 /* Fix the RHS type if it's a non-array expression. */
1164 if (ir
->rhs
->type
->is_32bit()) {
1165 ir_expression
*expr
= ir
->rhs
->as_expression();
1167 /* Convert the RHS to the LHS type. */
1169 (expr
->operation
== ir_unop_f162f
||
1170 expr
->operation
== ir_unop_i2i
||
1171 expr
->operation
== ir_unop_u2u
) &&
1172 expr
->operands
[0]->type
->is_16bit()) {
1173 /* If there is an "up" conversion, just remove it.
1174 * This is optional. We could as well execute the else statement and
1175 * let NIR eliminate the up+down conversions.
1177 ir
->rhs
= expr
->operands
[0];
1179 /* Add a "down" conversion operation to fix the type of RHS. */
1180 ir
->rhs
= convert_precision(false, ir
->rhs
);
1185 return ir_rvalue_enter_visitor::visit_enter(ir
);
1189 lower_variables_visitor::visit_enter(ir_return
*ir
)
1191 void *mem_ctx
= ralloc_parent(ir
);
1193 ir_dereference
*deref
= ir
->value
? ir
->value
->as_dereference() : NULL
;
1195 ir_variable
*var
= deref
->variable_referenced();
1197 /* Fix the type of the return value. */
1199 _mesa_set_search(lower_vars
, var
) &&
1200 deref
->type
->without_array()->is_32bit()) {
1201 /* Create a 32-bit temporary variable. */
1202 ir_variable
*new_var
=
1203 new(mem_ctx
) ir_variable(deref
->type
, "lowerp", ir_var_temporary
);
1204 base_ir
->insert_before(new_var
);
1206 /* Fix types in dereferences. */
1207 fix_types_in_deref_chain(deref
);
1209 /* Convert to 32 bits for the return value. */
1210 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(new_var
),
1212 ir
->value
= new(mem_ctx
) ir_dereference_variable(new_var
);
1216 return ir_rvalue_enter_visitor::visit_enter(ir
);
1219 void lower_variables_visitor::handle_rvalue(ir_rvalue
**rvalue
)
1221 ir_rvalue
*ir
= *rvalue
;
1223 if (in_assignee
|| ir
== NULL
)
1226 ir_expression
*expr
= ir
->as_expression();
1227 ir_dereference
*expr_op0_deref
= expr
? expr
->operands
[0]->as_dereference() : NULL
;
1229 /* Remove f2fmp(float16). Same for int16 and uint16. */
1232 (expr
->operation
== ir_unop_f2fmp
||
1233 expr
->operation
== ir_unop_i2imp
||
1234 expr
->operation
== ir_unop_u2ump
||
1235 expr
->operation
== ir_unop_f2f16
||
1236 expr
->operation
== ir_unop_i2i
||
1237 expr
->operation
== ir_unop_u2u
) &&
1238 expr
->type
->without_array()->is_16bit() &&
1239 expr_op0_deref
->type
->without_array()->is_32bit() &&
1240 expr_op0_deref
->variable_referenced() &&
1241 _mesa_set_search(lower_vars
, expr_op0_deref
->variable_referenced())) {
1242 fix_types_in_deref_chain(expr_op0_deref
);
1244 /* Remove f2fmp/i2imp/u2ump. */
1245 *rvalue
= expr_op0_deref
;
1249 ir_dereference
*deref
= ir
->as_dereference();
1252 ir_variable
*var
= deref
->variable_referenced();
1254 /* var can be NULL if we are dereferencing ir_constant. */
1256 _mesa_set_search(lower_vars
, var
) &&
1257 deref
->type
->without_array()->is_32bit()) {
1258 void *mem_ctx
= ralloc_parent(ir
);
1260 /* Create a 32-bit temporary variable. */
1261 ir_variable
*new_var
=
1262 new(mem_ctx
) ir_variable(deref
->type
, "lowerp", ir_var_temporary
);
1263 base_ir
->insert_before(new_var
);
1265 /* Fix types in dereferences. */
1266 fix_types_in_deref_chain(deref
);
1268 /* Convert to 32 bits for the rvalue. */
1269 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(new_var
),
1271 *rvalue
= new(mem_ctx
) ir_dereference_variable(new_var
);
1277 lower_variables_visitor::visit_enter(ir_call
*ir
)
1279 void *mem_ctx
= ralloc_parent(ir
);
1281 /* We can't pass 16-bit variables as 32-bit inout/out parameters. */
1282 foreach_two_lists(formal_node
, &ir
->callee
->parameters
,
1283 actual_node
, &ir
->actual_parameters
) {
1284 ir_dereference
*param_deref
=
1285 ((ir_rvalue
*)actual_node
)->as_dereference();
1286 ir_variable
*param
= (ir_variable
*)formal_node
;
1291 ir_variable
*var
= param_deref
->variable_referenced();
1293 /* var can be NULL if we are dereferencing ir_constant. */
1295 _mesa_set_search(lower_vars
, var
) &&
1296 param
->type
->without_array()->is_32bit()) {
1297 fix_types_in_deref_chain(param_deref
);
1299 /* Create a 32-bit temporary variable for the parameter. */
1300 ir_variable
*new_var
=
1301 new(mem_ctx
) ir_variable(param
->type
, "lowerp", ir_var_temporary
);
1302 base_ir
->insert_before(new_var
);
1304 /* Replace the parameter. */
1305 actual_node
->replace_with(new(mem_ctx
) ir_dereference_variable(new_var
));
1307 if (param
->data
.mode
== ir_var_function_in
||
1308 param
->data
.mode
== ir_var_function_inout
) {
1309 /* Convert to 32 bits for passing in. */
1310 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(new_var
),
1311 param_deref
->clone(mem_ctx
, NULL
), true);
1313 if (param
->data
.mode
== ir_var_function_out
||
1314 param
->data
.mode
== ir_var_function_inout
) {
1315 /* Convert to 16 bits after returning. */
1316 convert_split_assignment(param_deref
,
1317 new(mem_ctx
) ir_dereference_variable(new_var
),
1323 /* Fix the type of return value dereferencies. */
1324 ir_dereference_variable
*ret_deref
= ir
->return_deref
;
1325 ir_variable
*ret_var
= ret_deref
? ret_deref
->variable_referenced() : NULL
;
1328 _mesa_set_search(lower_vars
, ret_var
) &&
1329 ret_deref
->type
->without_array()->is_32bit()) {
1330 /* Create a 32-bit temporary variable. */
1331 ir_variable
*new_var
=
1332 new(mem_ctx
) ir_variable(ir
->callee
->return_type
, "lowerp",
1334 base_ir
->insert_before(new_var
);
1336 /* Replace the return variable. */
1337 ret_deref
->var
= new_var
;
1339 /* Convert to 16 bits after returning. */
1340 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(ret_var
),
1341 new(mem_ctx
) ir_dereference_variable(new_var
),
1345 return ir_rvalue_enter_visitor::visit_enter(ir
);
1351 lower_precision(const struct gl_shader_compiler_options
*options
,
1352 exec_list
*instructions
)
1354 find_precision_visitor
v(options
);
1355 find_lowerable_rvalues(options
, instructions
, v
.lowerable_rvalues
);
1356 visit_list_elements(&v
, instructions
);
1358 lower_variables_visitor
vars(options
);
1359 visit_list_elements(&vars
, instructions
);