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
);
451 if (desc
->channel
[i
].pure_integer
||
452 desc
->channel
[i
].type
== UTIL_FORMAT_TYPE_FLOAT
)
453 return desc
->channel
[i
].size
<= 16;
455 return desc
->channel
[i
].size
<= 10; /* unorm/snorm */
458 /* Handle special calls. */
459 if (ir
->callee
->is_builtin() && ir
->actual_parameters
.length()) {
460 ir_rvalue
*param
= (ir_rvalue
*)ir
->actual_parameters
.get_head();
461 ir_variable
*var
= param
->variable_referenced();
463 /* Handle builtin wrappers around ir_texture opcodes. These wrappers will
464 * be inlined by lower_precision() if we return true here, so that we can
465 * get to ir_texture later and do proper lowering.
467 * We should lower the type of the return value if the sampler type
468 * uses lower precision. The function parameters don't matter.
470 if (var
&& var
->type
->without_array()->is_sampler()) {
471 /* textureSize always returns highp. */
472 if (!strcmp(ir
->callee_name(), "textureSize"))
475 return var
->data
.precision
== GLSL_PRECISION_MEDIUM
||
476 var
->data
.precision
== GLSL_PRECISION_LOW
;
480 if (!ir
->callee
->is_builtin() ||
481 /* Parameters are always highp: */
482 !strcmp(ir
->callee_name(), "floatBitsToInt") ||
483 !strcmp(ir
->callee_name(), "floatBitsToUint") ||
484 !strcmp(ir
->callee_name(), "intBitsToFloat") ||
485 !strcmp(ir
->callee_name(), "uintBitsToFloat") ||
486 !strcmp(ir
->callee_name(), "bitfieldReverse") ||
487 !strcmp(ir
->callee_name(), "frexp") ||
488 !strcmp(ir
->callee_name(), "ldexp") ||
489 /* Parameters and outputs are always highp: */
490 /* TODO: The operations are highp, but carry and borrow outputs are lowp. */
491 !strcmp(ir
->callee_name(), "uaddCarry") ||
492 !strcmp(ir
->callee_name(), "usubBorrow") ||
493 !strcmp(ir
->callee_name(), "imulExtended") ||
494 !strcmp(ir
->callee_name(), "umulExtended") ||
495 !strcmp(ir
->callee_name(), "unpackUnorm2x16") ||
496 !strcmp(ir
->callee_name(), "unpackSnorm2x16") ||
497 /* Outputs are highp: */
498 !strcmp(ir
->callee_name(), "packUnorm2x16") ||
499 !strcmp(ir
->callee_name(), "packSnorm2x16") ||
500 /* Parameters are mediump and outputs are highp. The parameters should
501 * be optimized in NIR, not here, e.g:
502 * - packHalf2x16 can just be a bitcast from f16vec2 to uint32
503 * - Other opcodes don't have to convert parameters to highp if the hw
504 * has f16 versions. Optimize in NIR accordingly.
506 !strcmp(ir
->callee_name(), "packHalf2x16") ||
507 !strcmp(ir
->callee_name(), "packUnorm4x8") ||
508 !strcmp(ir
->callee_name(), "packSnorm4x8"))
511 assert(ir
->callee
->return_precision
== GLSL_PRECISION_NONE
);
513 /* Number of parameters to check if they are lowerable. */
514 unsigned check_parameters
= ir
->actual_parameters
.length();
516 /* Interpolation functions only consider the precision of the interpolant. */
517 /* Bitfield functions ignore the precision of "offset" and "bits". */
518 if (!strcmp(ir
->callee_name(), "interpolateAtOffset") ||
519 !strcmp(ir
->callee_name(), "interpolateAtSample") ||
520 !strcmp(ir
->callee_name(), "bitfieldExtract")) {
521 check_parameters
= 1;
522 } else if (!strcmp(ir
->callee_name(), "bitfieldInsert")) {
523 check_parameters
= 2;
524 } if (function_always_returns_mediump_or_lowp(ir
->callee_name())) {
525 /* These only lower the return value. Parameters keep their precision,
526 * which is preserved in map_builtin.
528 check_parameters
= 0;
531 foreach_in_list(ir_rvalue
, param
, &ir
->actual_parameters
) {
532 if (!check_parameters
)
535 if (!param
->as_constant() &&
536 _mesa_set_search(lowerable_rvalues
, param
) == NULL
)
546 find_lowerable_rvalues_visitor::visit_leave(ir_call
*ir
)
548 ir_hierarchical_visitor::visit_leave(ir
);
550 /* Special case for handling temporary variables generated by the compiler
551 * for function calls. If we assign to one of these using a function call
552 * that has a lowerable return type then we can assume the temporary
553 * variable should have a medium precision too.
556 /* Do nothing if the return type is void. */
557 if (!ir
->return_deref
)
558 return visit_continue
;
560 ir_variable
*var
= ir
->return_deref
->variable_referenced();
562 assert(var
->data
.mode
== ir_var_temporary
);
564 unsigned return_precision
= ir
->callee
->return_precision
;
566 /* If the call is to a builtin, then the function won’t have a return
567 * precision and we should determine it from the precision of the arguments.
569 if (is_lowerable_builtin(ir
, lowerable_rvalues
))
570 return_precision
= GLSL_PRECISION_MEDIUM
;
572 can_lower_state lower_state
=
573 handle_precision(var
->type
, return_precision
);
575 if (lower_state
== SHOULD_LOWER
) {
576 /* There probably shouldn’t be any situations where multiple ir_call
577 * instructions write to the same temporary?
579 assert(var
->data
.precision
== GLSL_PRECISION_NONE
);
580 var
->data
.precision
= GLSL_PRECISION_MEDIUM
;
582 var
->data
.precision
= GLSL_PRECISION_HIGH
;
585 return visit_continue
;
589 find_lowerable_rvalues_visitor::visit_leave(ir_assignment
*ir
)
591 ir_hierarchical_visitor::visit_leave(ir
);
593 /* Special case for handling temporary variables generated by the compiler.
594 * If we assign to one of these using a lowered precision then we can assume
595 * the temporary variable should have a medium precision too.
597 ir_variable
*var
= ir
->lhs
->variable_referenced();
599 if (var
->data
.mode
== ir_var_temporary
) {
600 if (_mesa_set_search(lowerable_rvalues
, ir
->rhs
)) {
601 /* Only override the precision if this is the first assignment. For
602 * temporaries such as the ones generated for the ?: operator there
603 * can be multiple assignments with different precisions. This way we
604 * get the highest precision of all of the assignments.
606 if (var
->data
.precision
== GLSL_PRECISION_NONE
)
607 var
->data
.precision
= GLSL_PRECISION_MEDIUM
;
608 } else if (!ir
->rhs
->as_constant()) {
609 var
->data
.precision
= GLSL_PRECISION_HIGH
;
613 return visit_continue
;
617 find_lowerable_rvalues(const struct gl_shader_compiler_options
*options
,
618 exec_list
*instructions
,
621 find_lowerable_rvalues_visitor
v(result
, options
);
623 visit_list_elements(&v
, instructions
);
625 assert(v
.stack
.empty());
628 static const glsl_type
*
629 convert_type(bool up
, const glsl_type
*type
)
631 if (type
->is_array()) {
632 return glsl_type::get_array_instance(convert_type(up
, type
->fields
.array
),
634 type
->explicit_stride
);
637 glsl_base_type new_base_type
;
640 switch (type
->base_type
) {
641 case GLSL_TYPE_FLOAT16
:
642 new_base_type
= GLSL_TYPE_FLOAT
;
644 case GLSL_TYPE_INT16
:
645 new_base_type
= GLSL_TYPE_INT
;
647 case GLSL_TYPE_UINT16
:
648 new_base_type
= GLSL_TYPE_UINT
;
651 unreachable("invalid type");
655 switch (type
->base_type
) {
656 case GLSL_TYPE_FLOAT
:
657 new_base_type
= GLSL_TYPE_FLOAT16
;
660 new_base_type
= GLSL_TYPE_INT16
;
663 new_base_type
= GLSL_TYPE_UINT16
;
666 unreachable("invalid type");
671 return glsl_type::get_instance(new_base_type
,
672 type
->vector_elements
,
673 type
->matrix_columns
,
674 type
->explicit_stride
,
675 type
->interface_row_major
);
678 static const glsl_type
*
679 lower_glsl_type(const glsl_type
*type
)
681 return convert_type(false, type
);
685 convert_precision(bool up
, ir_rvalue
*ir
)
690 switch (ir
->type
->without_array()->base_type
) {
691 case GLSL_TYPE_FLOAT16
:
694 case GLSL_TYPE_INT16
:
697 case GLSL_TYPE_UINT16
:
701 unreachable("invalid type");
705 switch (ir
->type
->without_array()->base_type
) {
706 case GLSL_TYPE_FLOAT
:
716 unreachable("invalid type");
721 const glsl_type
*desired_type
= convert_type(up
, ir
->type
);
722 void *mem_ctx
= ralloc_parent(ir
);
723 return new(mem_ctx
) ir_expression(op
, desired_type
, ir
, NULL
);
727 lower_precision_visitor::handle_rvalue(ir_rvalue
**rvalue
)
729 ir_rvalue
*ir
= *rvalue
;
734 if (ir
->as_dereference()) {
735 if (!ir
->type
->is_boolean())
736 *rvalue
= convert_precision(false, ir
);
737 } else if (ir
->type
->is_32bit()) {
738 ir
->type
= lower_glsl_type(ir
->type
);
740 ir_constant
*const_ir
= ir
->as_constant();
743 ir_constant_data value
;
745 if (ir
->type
->base_type
== GLSL_TYPE_FLOAT16
) {
746 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.f16
); i
++)
747 value
.f16
[i
] = _mesa_float_to_half(const_ir
->value
.f
[i
]);
748 } else if (ir
->type
->base_type
== GLSL_TYPE_INT16
) {
749 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.i16
); i
++)
750 value
.i16
[i
] = const_ir
->value
.i
[i
];
751 } else if (ir
->type
->base_type
== GLSL_TYPE_UINT16
) {
752 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.u16
); i
++)
753 value
.u16
[i
] = const_ir
->value
.u
[i
];
755 unreachable("invalid type");
758 const_ir
->value
= value
;
764 lower_precision_visitor::visit_enter(ir_dereference_record
*ir
)
766 /* We don’t want to lower the variable */
767 return visit_continue_with_parent
;
771 lower_precision_visitor::visit_enter(ir_dereference_array
*ir
)
773 /* We don’t want to convert the array index or the variable. If the array
774 * index itself is lowerable that will be handled separately.
776 return visit_continue_with_parent
;
780 lower_precision_visitor::visit_enter(ir_call
*ir
)
782 /* We don’t want to convert the arguments. These will be handled separately.
784 return visit_continue_with_parent
;
788 lower_precision_visitor::visit_enter(ir_texture
*ir
)
790 /* We don’t want to convert the arguments. These will be handled separately.
792 return visit_continue_with_parent
;
796 lower_precision_visitor::visit_leave(ir_expression
*ir
)
798 ir_rvalue_visitor::visit_leave(ir
);
800 /* If the expression is a conversion operation to or from bool then fix the
803 switch (ir
->operation
) {
805 ir
->operation
= ir_unop_b2f16
;
808 ir
->operation
= ir_unop_f162b
;
812 /* Nothing to do - they both support int16. */
818 return visit_continue
;
822 find_precision_visitor::handle_rvalue(ir_rvalue
**rvalue
)
824 /* Checking the precision of rvalue can be lowered first throughout
825 * find_lowerable_rvalues_visitor.
826 * Once it found the precision of rvalue can be lowered, then we can
827 * add conversion f2fmp, etc. through lower_precision_visitor.
832 struct set_entry
*entry
= _mesa_set_search(lowerable_rvalues
, *rvalue
);
837 _mesa_set_remove(lowerable_rvalues
, entry
);
839 /* If the entire expression is just a variable dereference then trying to
840 * lower it will just directly add pointless to and from conversions without
841 * any actual operation in-between. Although these will eventually get
842 * optimised out, avoiding generating them here also avoids breaking inout
843 * parameters to functions.
845 if ((*rvalue
)->as_dereference())
848 lower_precision_visitor v
;
850 (*rvalue
)->accept(&v
);
851 v
.handle_rvalue(rvalue
);
853 /* We don’t need to add the final conversion if the final type has been
856 if ((*rvalue
)->type
->base_type
!= GLSL_TYPE_BOOL
) {
857 *rvalue
= convert_precision(true, *rvalue
);
862 find_precision_visitor::visit_enter(ir_call
*ir
)
864 ir_rvalue_enter_visitor::visit_enter(ir
);
866 ir_variable
*return_var
=
867 ir
->return_deref
? ir
->return_deref
->variable_referenced() : NULL
;
869 /* Don't do anything for image_load here. We have only changed the return
870 * value to mediump/lowp, so that following instructions can use reduced
873 * The return value type of the intrinsic itself isn't changed here, but
874 * can be changed in NIR if all users use the *2*mp opcode.
876 if (ir
->callee
->intrinsic_id
== ir_intrinsic_image_load
)
877 return visit_continue
;
879 /* If this is a call to a builtin and the find_lowerable_rvalues_visitor
880 * overrode the precision of the temporary return variable, then we can
881 * replace the builtin implementation with a lowered version.
884 if (!ir
->callee
->is_builtin() ||
885 return_var
== NULL
||
886 (return_var
->data
.precision
!= GLSL_PRECISION_MEDIUM
&&
887 return_var
->data
.precision
!= GLSL_PRECISION_LOW
))
888 return visit_continue
;
890 ir
->callee
= map_builtin(ir
->callee
);
891 ir
->generate_inline(ir
);
894 return visit_continue_with_parent
;
897 ir_function_signature
*
898 find_precision_visitor::map_builtin(ir_function_signature
*sig
)
900 if (lowered_builtins
== NULL
) {
901 lowered_builtins
= _mesa_pointer_hash_table_create(NULL
);
902 clone_ht
=_mesa_pointer_hash_table_create(NULL
);
903 lowered_builtin_mem_ctx
= ralloc_context(NULL
);
905 struct hash_entry
*entry
= _mesa_hash_table_search(lowered_builtins
, sig
);
907 return (ir_function_signature
*) entry
->data
;
910 ir_function_signature
*lowered_sig
=
911 sig
->clone(lowered_builtin_mem_ctx
, clone_ht
);
913 /* Functions that always return mediump or lowp should keep their
914 * parameters intact, because they can be highp. NIR can lower
915 * the up-conversion for parameters if needed.
917 if (!function_always_returns_mediump_or_lowp(sig
->function_name())) {
918 foreach_in_list(ir_variable
, param
, &lowered_sig
->parameters
) {
919 param
->data
.precision
= GLSL_PRECISION_MEDIUM
;
923 lower_precision(options
, &lowered_sig
->body
);
925 _mesa_hash_table_clear(clone_ht
, NULL
);
927 _mesa_hash_table_insert(lowered_builtins
, sig
, lowered_sig
);
932 find_precision_visitor::find_precision_visitor(const struct gl_shader_compiler_options
*options
)
933 : lowerable_rvalues(_mesa_pointer_set_create(NULL
)),
934 lowered_builtins(NULL
),
936 lowered_builtin_mem_ctx(NULL
),
941 find_precision_visitor::~find_precision_visitor()
943 _mesa_set_destroy(lowerable_rvalues
, NULL
);
945 if (lowered_builtins
) {
946 _mesa_hash_table_destroy(lowered_builtins
, NULL
);
947 _mesa_hash_table_destroy(clone_ht
, NULL
);
948 ralloc_free(lowered_builtin_mem_ctx
);
952 /* Lowering opcodes to 16 bits is not enough for programs with control flow
953 * (and the ?: operator, which is represented by if-then-else in the IR),
954 * because temporary variables, which are used for passing values between
955 * code blocks, are not lowered, resulting in 32-bit phis in NIR.
957 * First change the variable types to 16 bits, then change all ir_dereference
960 class lower_variables_visitor
: public ir_rvalue_enter_visitor
{
962 lower_variables_visitor(const struct gl_shader_compiler_options
*options
)
964 lower_vars
= _mesa_pointer_set_create(NULL
);
967 virtual ~lower_variables_visitor()
969 _mesa_set_destroy(lower_vars
, NULL
);
972 virtual ir_visitor_status
visit(ir_variable
*var
);
973 virtual ir_visitor_status
visit_enter(ir_assignment
*ir
);
974 virtual ir_visitor_status
visit_enter(ir_return
*ir
);
975 virtual ir_visitor_status
visit_enter(ir_call
*ir
);
976 virtual void handle_rvalue(ir_rvalue
**rvalue
);
978 void fix_types_in_deref_chain(ir_dereference
*ir
);
979 void convert_split_assignment(ir_dereference
*lhs
, ir_rvalue
*rhs
,
982 const struct gl_shader_compiler_options
*options
;
987 lower_constant(ir_constant
*ir
)
989 if (ir
->type
->is_array()) {
990 for (int i
= 0; i
< ir
->type
->array_size(); i
++)
991 lower_constant(ir
->get_array_element(i
));
993 ir
->type
= lower_glsl_type(ir
->type
);
997 ir
->type
= lower_glsl_type(ir
->type
);
998 ir_constant_data value
;
1000 if (ir
->type
->base_type
== GLSL_TYPE_FLOAT16
) {
1001 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.f16
); i
++)
1002 value
.f16
[i
] = _mesa_float_to_half(ir
->value
.f
[i
]);
1003 } else if (ir
->type
->base_type
== GLSL_TYPE_INT16
) {
1004 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.i16
); i
++)
1005 value
.i16
[i
] = ir
->value
.i
[i
];
1006 } else if (ir
->type
->base_type
== GLSL_TYPE_UINT16
) {
1007 for (unsigned i
= 0; i
< ARRAY_SIZE(value
.u16
); i
++)
1008 value
.u16
[i
] = ir
->value
.u
[i
];
1010 unreachable("invalid type");
1017 lower_variables_visitor::visit(ir_variable
*var
)
1019 if ((var
->data
.mode
!= ir_var_temporary
&&
1020 var
->data
.mode
!= ir_var_auto
) ||
1021 !var
->type
->without_array()->is_32bit() ||
1022 (var
->data
.precision
!= GLSL_PRECISION_MEDIUM
&&
1023 var
->data
.precision
!= GLSL_PRECISION_LOW
) ||
1024 !can_lower_type(options
, var
->type
))
1025 return visit_continue
;
1027 /* Lower constant initializers. */
1028 if (var
->constant_value
&&
1029 var
->type
== var
->constant_value
->type
) {
1030 var
->constant_value
=
1031 var
->constant_value
->clone(ralloc_parent(var
), NULL
);
1032 lower_constant(var
->constant_value
);
1035 if (var
->constant_initializer
&&
1036 var
->type
== var
->constant_initializer
->type
) {
1037 var
->constant_initializer
=
1038 var
->constant_initializer
->clone(ralloc_parent(var
), NULL
);
1039 lower_constant(var
->constant_initializer
);
1042 var
->type
= lower_glsl_type(var
->type
);
1043 _mesa_set_add(lower_vars
, var
);
1045 return visit_continue
;
1049 lower_variables_visitor::fix_types_in_deref_chain(ir_dereference
*ir
)
1051 assert(ir
->type
->without_array()->is_32bit());
1052 assert(_mesa_set_search(lower_vars
, ir
->variable_referenced()));
1054 /* Fix the type in the dereference node. */
1055 ir
->type
= lower_glsl_type(ir
->type
);
1057 /* If it's an array, fix the types in the whole dereference chain. */
1058 for (ir_dereference_array
*deref_array
= ir
->as_dereference_array();
1060 deref_array
= deref_array
->array
->as_dereference_array()) {
1061 assert(deref_array
->array
->type
->without_array()->is_32bit());
1062 deref_array
->array
->type
= lower_glsl_type(deref_array
->array
->type
);
1067 lower_variables_visitor::convert_split_assignment(ir_dereference
*lhs
,
1071 void *mem_ctx
= ralloc_parent(lhs
);
1073 if (lhs
->type
->is_array()) {
1074 for (unsigned i
= 0; i
< lhs
->type
->length
; i
++) {
1075 ir_dereference
*l
, *r
;
1077 l
= new(mem_ctx
) ir_dereference_array(lhs
->clone(mem_ctx
, NULL
),
1078 new(mem_ctx
) ir_constant(i
));
1079 r
= new(mem_ctx
) ir_dereference_array(rhs
->clone(mem_ctx
, NULL
),
1080 new(mem_ctx
) ir_constant(i
));
1081 convert_split_assignment(l
, r
, insert_before
);
1086 assert(lhs
->type
->is_16bit() || lhs
->type
->is_32bit());
1087 assert(rhs
->type
->is_16bit() || rhs
->type
->is_32bit());
1088 assert(lhs
->type
->is_16bit() != rhs
->type
->is_16bit());
1090 ir_assignment
*assign
=
1091 new(mem_ctx
) ir_assignment(lhs
, convert_precision(lhs
->type
->is_32bit(), rhs
));
1094 base_ir
->insert_before(assign
);
1096 base_ir
->insert_after(assign
);
1100 lower_variables_visitor::visit_enter(ir_assignment
*ir
)
1102 ir_dereference
*lhs
= ir
->lhs
;
1103 ir_variable
*var
= lhs
->variable_referenced();
1104 ir_dereference
*rhs_deref
= ir
->rhs
->as_dereference();
1105 ir_variable
*rhs_var
= rhs_deref
? rhs_deref
->variable_referenced() : NULL
;
1106 ir_constant
*rhs_const
= ir
->rhs
->as_constant();
1108 /* Legalize array assignments between lowered and non-lowered variables. */
1109 if (lhs
->type
->is_array() &&
1110 (rhs_var
|| rhs_const
) &&
1112 var
->type
->without_array()->is_16bit() !=
1113 rhs_var
->type
->without_array()->is_16bit()) &&
1115 (var
->type
->without_array()->is_16bit() &&
1116 rhs_const
->type
->without_array()->is_32bit()))) {
1117 assert(ir
->rhs
->type
->is_array());
1119 /* Fix array assignments from lowered to non-lowered. */
1120 if (rhs_var
&& _mesa_set_search(lower_vars
, rhs_var
)) {
1121 fix_types_in_deref_chain(rhs_deref
);
1122 /* Convert to 32 bits for LHS. */
1123 convert_split_assignment(lhs
, rhs_deref
, true);
1125 return visit_continue
;
1128 /* Fix array assignments from non-lowered to lowered. */
1129 if (_mesa_set_search(lower_vars
, var
) &&
1130 ir
->rhs
->type
->without_array()->is_32bit()) {
1131 fix_types_in_deref_chain(lhs
);
1132 /* Convert to 16 bits for LHS. */
1133 convert_split_assignment(lhs
, ir
->rhs
, true);
1135 return visit_continue
;
1139 /* Fix assignment types. */
1140 if (_mesa_set_search(lower_vars
, var
)) {
1141 /* Fix the LHS type. */
1142 if (lhs
->type
->without_array()->is_32bit())
1143 fix_types_in_deref_chain(lhs
);
1145 /* Fix the RHS type if it's a lowered variable. */
1147 _mesa_set_search(lower_vars
, rhs_var
) &&
1148 rhs_deref
->type
->without_array()->is_32bit())
1149 fix_types_in_deref_chain(rhs_deref
);
1151 /* Fix the RHS type if it's a non-array expression. */
1152 if (ir
->rhs
->type
->is_32bit()) {
1153 ir_expression
*expr
= ir
->rhs
->as_expression();
1155 /* Convert the RHS to the LHS type. */
1157 (expr
->operation
== ir_unop_f162f
||
1158 expr
->operation
== ir_unop_i2i
||
1159 expr
->operation
== ir_unop_u2u
) &&
1160 expr
->operands
[0]->type
->is_16bit()) {
1161 /* If there is an "up" conversion, just remove it.
1162 * This is optional. We could as well execute the else statement and
1163 * let NIR eliminate the up+down conversions.
1165 ir
->rhs
= expr
->operands
[0];
1167 /* Add a "down" conversion operation to fix the type of RHS. */
1168 ir
->rhs
= convert_precision(false, ir
->rhs
);
1173 return ir_rvalue_enter_visitor::visit_enter(ir
);
1177 lower_variables_visitor::visit_enter(ir_return
*ir
)
1179 void *mem_ctx
= ralloc_parent(ir
);
1181 ir_dereference
*deref
= ir
->value
? ir
->value
->as_dereference() : NULL
;
1183 ir_variable
*var
= deref
->variable_referenced();
1185 /* Fix the type of the return value. */
1186 if (_mesa_set_search(lower_vars
, var
) &&
1187 deref
->type
->without_array()->is_32bit()) {
1188 /* Create a 32-bit temporary variable. */
1189 ir_variable
*new_var
=
1190 new(mem_ctx
) ir_variable(deref
->type
, "lowerp", ir_var_temporary
);
1191 base_ir
->insert_before(new_var
);
1193 /* Fix types in dereferences. */
1194 fix_types_in_deref_chain(deref
);
1196 /* Convert to 32 bits for the return value. */
1197 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(new_var
),
1199 ir
->value
= new(mem_ctx
) ir_dereference_variable(new_var
);
1203 return ir_rvalue_enter_visitor::visit_enter(ir
);
1206 void lower_variables_visitor::handle_rvalue(ir_rvalue
**rvalue
)
1208 ir_rvalue
*ir
= *rvalue
;
1210 if (in_assignee
|| ir
== NULL
)
1213 ir_expression
*expr
= ir
->as_expression();
1214 ir_dereference
*expr_op0_deref
= expr
? expr
->operands
[0]->as_dereference() : NULL
;
1216 /* Remove f2fmp(float16). Same for int16 and uint16. */
1219 (expr
->operation
== ir_unop_f2fmp
||
1220 expr
->operation
== ir_unop_i2imp
||
1221 expr
->operation
== ir_unop_u2ump
||
1222 expr
->operation
== ir_unop_f2f16
||
1223 expr
->operation
== ir_unop_i2i
||
1224 expr
->operation
== ir_unop_u2u
) &&
1225 expr
->type
->without_array()->is_16bit() &&
1226 expr_op0_deref
->type
->without_array()->is_32bit() &&
1227 _mesa_set_search(lower_vars
, expr_op0_deref
->variable_referenced())) {
1228 fix_types_in_deref_chain(expr_op0_deref
);
1230 /* Remove f2fmp/i2imp/u2ump. */
1231 *rvalue
= expr_op0_deref
;
1235 ir_dereference
*deref
= ir
->as_dereference();
1238 ir_variable
*var
= deref
->variable_referenced();
1241 if (_mesa_set_search(lower_vars
, var
) &&
1242 deref
->type
->without_array()->is_32bit()) {
1243 fix_types_in_deref_chain(deref
);
1245 /* Then convert the type up. Optimizations should eliminate this. */
1246 *rvalue
= convert_precision(true, deref
);
1252 lower_variables_visitor::visit_enter(ir_call
*ir
)
1254 void *mem_ctx
= ralloc_parent(ir
);
1256 /* We can't pass 16-bit variables as 32-bit inout/out parameters. */
1257 foreach_two_lists(formal_node
, &ir
->callee
->parameters
,
1258 actual_node
, &ir
->actual_parameters
) {
1259 ir_dereference
*param_deref
=
1260 ((ir_rvalue
*)actual_node
)->as_dereference();
1261 ir_variable
*param
= (ir_variable
*)formal_node
;
1266 ir_variable
*var
= param_deref
->variable_referenced();
1268 if (_mesa_set_search(lower_vars
, var
) &&
1269 param
->type
->without_array()->is_32bit()) {
1270 fix_types_in_deref_chain(param_deref
);
1272 /* Create a 32-bit temporary variable for the parameter. */
1273 ir_variable
*new_var
=
1274 new(mem_ctx
) ir_variable(param
->type
, "lowerp", ir_var_temporary
);
1275 base_ir
->insert_before(new_var
);
1277 /* Replace the parameter. */
1278 actual_node
->replace_with(new(mem_ctx
) ir_dereference_variable(new_var
));
1280 if (param
->data
.mode
== ir_var_function_in
||
1281 param
->data
.mode
== ir_var_function_inout
) {
1282 /* Convert to 32 bits for passing in. */
1283 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(new_var
),
1284 param_deref
->clone(mem_ctx
, NULL
), true);
1286 if (param
->data
.mode
== ir_var_function_out
||
1287 param
->data
.mode
== ir_var_function_inout
) {
1288 /* Convert to 16 bits after returning. */
1289 convert_split_assignment(param_deref
,
1290 new(mem_ctx
) ir_dereference_variable(new_var
),
1296 /* Fix the type of return value dereferencies. */
1297 ir_dereference_variable
*ret_deref
= ir
->return_deref
;
1298 ir_variable
*ret_var
= ret_deref
? ret_deref
->variable_referenced() : NULL
;
1301 _mesa_set_search(lower_vars
, ret_var
) &&
1302 ret_deref
->type
->without_array()->is_32bit()) {
1303 /* Create a 32-bit temporary variable. */
1304 ir_variable
*new_var
=
1305 new(mem_ctx
) ir_variable(ir
->callee
->return_type
, "lowerp",
1307 base_ir
->insert_before(new_var
);
1309 /* Replace the return variable. */
1310 ret_deref
->var
= new_var
;
1312 /* Convert to 16 bits after returning. */
1313 convert_split_assignment(new(mem_ctx
) ir_dereference_variable(ret_var
),
1314 new(mem_ctx
) ir_dereference_variable(new_var
),
1318 return ir_rvalue_enter_visitor::visit_enter(ir
);
1324 lower_precision(const struct gl_shader_compiler_options
*options
,
1325 exec_list
*instructions
)
1327 find_precision_visitor
v(options
);
1328 find_lowerable_rvalues(options
, instructions
, v
.lowerable_rvalues
);
1329 visit_list_elements(&v
, instructions
);
1331 if (options
->LowerPrecisionTemporaries
) {
1332 lower_variables_visitor
vars(options
);
1333 visit_list_elements(&vars
, instructions
);