2 * Copyright © 2014 Intel Corporation
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 DEALINGS
24 * Connor Abbott (cwabbott0@gmail.com)
29 #include "compiler/shader_enums.h"
30 #include "util/half_float.h"
31 #include "vulkan/vulkan_core.h"
34 #include <inttypes.h> /* for PRIx64 macro */
37 print_tabs(unsigned num_tabs
, FILE *fp
)
39 for (unsigned i
= 0; i
< num_tabs
; i
++)
46 /** map from nir_variable -> printable name */
47 struct hash_table
*ht
;
49 /** set of names used so far for nir_variables */
52 /* an index used to make new non-conflicting names */
56 * Optional table of annotations mapping nir object
57 * (such as instr or var) to message to print.
59 struct hash_table
*annotations
;
63 print_annotation(print_state
*state
, void *obj
)
67 if (!state
->annotations
)
70 struct hash_entry
*entry
= _mesa_hash_table_search(state
->annotations
, obj
);
74 const char *note
= entry
->data
;
75 _mesa_hash_table_remove(state
->annotations
, entry
);
77 fprintf(fp
, "%s\n\n", note
);
81 print_register(nir_register
*reg
, print_state
*state
)
84 if (reg
->name
!= NULL
)
85 fprintf(fp
, "/* %s */ ", reg
->name
);
86 fprintf(fp
, "r%u", reg
->index
);
89 static const char *sizes
[] = { "error", "vec1", "vec2", "vec3", "vec4",
90 "error", "error", "error", "vec8",
91 "error", "error", "error", "error",
92 "error", "error", "error", "vec16"};
95 print_register_decl(nir_register
*reg
, print_state
*state
)
98 fprintf(fp
, "decl_reg %s %u ", sizes
[reg
->num_components
], reg
->bit_size
);
99 print_register(reg
, state
);
100 if (reg
->num_array_elems
!= 0)
101 fprintf(fp
, "[%u]", reg
->num_array_elems
);
106 print_ssa_def(nir_ssa_def
*def
, print_state
*state
)
108 FILE *fp
= state
->fp
;
109 if (def
->name
!= NULL
)
110 fprintf(fp
, "/* %s */ ", def
->name
);
111 fprintf(fp
, "%s %u ssa_%u", sizes
[def
->num_components
], def
->bit_size
,
116 print_ssa_use(nir_ssa_def
*def
, print_state
*state
)
118 FILE *fp
= state
->fp
;
119 if (def
->name
!= NULL
)
120 fprintf(fp
, "/* %s */ ", def
->name
);
121 fprintf(fp
, "ssa_%u", def
->index
);
124 static void print_src(const nir_src
*src
, print_state
*state
);
127 print_reg_src(const nir_reg_src
*src
, print_state
*state
)
129 FILE *fp
= state
->fp
;
130 print_register(src
->reg
, state
);
131 if (src
->reg
->num_array_elems
!= 0) {
132 fprintf(fp
, "[%u", src
->base_offset
);
133 if (src
->indirect
!= NULL
) {
135 print_src(src
->indirect
, state
);
142 print_reg_dest(nir_reg_dest
*dest
, print_state
*state
)
144 FILE *fp
= state
->fp
;
145 print_register(dest
->reg
, state
);
146 if (dest
->reg
->num_array_elems
!= 0) {
147 fprintf(fp
, "[%u", dest
->base_offset
);
148 if (dest
->indirect
!= NULL
) {
150 print_src(dest
->indirect
, state
);
157 print_src(const nir_src
*src
, print_state
*state
)
160 print_ssa_use(src
->ssa
, state
);
162 print_reg_src(&src
->reg
, state
);
166 print_dest(nir_dest
*dest
, print_state
*state
)
169 print_ssa_def(&dest
->ssa
, state
);
171 print_reg_dest(&dest
->reg
, state
);
175 comp_mask_string(unsigned num_components
)
177 return (num_components
> 4) ? "abcdefghijklmnop" : "xyzw";
181 print_alu_src(nir_alu_instr
*instr
, unsigned src
, print_state
*state
)
183 FILE *fp
= state
->fp
;
185 if (instr
->src
[src
].negate
)
187 if (instr
->src
[src
].abs
)
190 print_src(&instr
->src
[src
].src
, state
);
192 bool print_swizzle
= false;
193 nir_component_mask_t used_channels
= 0;
195 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++) {
196 if (!nir_alu_instr_channel_used(instr
, src
, i
))
201 if (instr
->src
[src
].swizzle
[i
] != i
) {
202 print_swizzle
= true;
207 unsigned live_channels
= nir_src_num_components(instr
->src
[src
].src
);
209 if (print_swizzle
|| used_channels
!= live_channels
) {
211 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++) {
212 if (!nir_alu_instr_channel_used(instr
, src
, i
))
215 fprintf(fp
, "%c", comp_mask_string(live_channels
)[instr
->src
[src
].swizzle
[i
]]);
219 if (instr
->src
[src
].abs
)
224 print_alu_dest(nir_alu_dest
*dest
, print_state
*state
)
226 FILE *fp
= state
->fp
;
227 /* we're going to print the saturate modifier later, after the opcode */
229 print_dest(&dest
->dest
, state
);
231 if (!dest
->dest
.is_ssa
&&
232 dest
->write_mask
!= (1 << dest
->dest
.reg
.reg
->num_components
) - 1) {
233 unsigned live_channels
= dest
->dest
.reg
.reg
->num_components
;
235 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++)
236 if ((dest
->write_mask
>> i
) & 1)
237 fprintf(fp
, "%c", comp_mask_string(live_channels
)[i
]);
242 print_alu_instr(nir_alu_instr
*instr
, print_state
*state
)
244 FILE *fp
= state
->fp
;
246 print_alu_dest(&instr
->dest
, state
);
248 fprintf(fp
, " = %s", nir_op_infos
[instr
->op
].name
);
251 if (instr
->dest
.saturate
)
253 if (instr
->no_signed_wrap
)
255 if (instr
->no_unsigned_wrap
)
259 for (unsigned i
= 0; i
< nir_op_infos
[instr
->op
].num_inputs
; i
++) {
263 print_alu_src(instr
, i
, state
);
268 get_var_name(nir_variable
*var
, print_state
*state
)
270 if (state
->ht
== NULL
)
271 return var
->name
? var
->name
: "unnamed";
275 struct hash_entry
*entry
= _mesa_hash_table_search(state
->ht
, var
);
280 if (var
->name
== NULL
) {
281 name
= ralloc_asprintf(state
->syms
, "@%u", state
->index
++);
283 struct set_entry
*set_entry
= _mesa_set_search(state
->syms
, var
->name
);
284 if (set_entry
!= NULL
) {
285 /* we have a collision with another name, append an @ + a unique
287 name
= ralloc_asprintf(state
->syms
, "%s@%u", var
->name
,
290 /* Mark this one as seen */
291 _mesa_set_add(state
->syms
, var
->name
);
296 _mesa_hash_table_insert(state
->ht
, var
, name
);
302 print_constant(nir_constant
*c
, const struct glsl_type
*type
, print_state
*state
)
304 FILE *fp
= state
->fp
;
305 const unsigned rows
= glsl_get_vector_elements(type
);
306 const unsigned cols
= glsl_get_matrix_columns(type
);
309 switch (glsl_get_base_type(type
)) {
311 /* Only float base types can be matrices. */
314 for (i
= 0; i
< rows
; i
++) {
315 if (i
> 0) fprintf(fp
, ", ");
316 fprintf(fp
, "%s", c
->values
[i
].b
? "true" : "false");
320 case GLSL_TYPE_UINT8
:
322 /* Only float base types can be matrices. */
325 for (i
= 0; i
< rows
; i
++) {
326 if (i
> 0) fprintf(fp
, ", ");
327 fprintf(fp
, "0x%02x", c
->values
[i
].u8
);
331 case GLSL_TYPE_UINT16
:
332 case GLSL_TYPE_INT16
:
333 /* Only float base types can be matrices. */
336 for (i
= 0; i
< rows
; i
++) {
337 if (i
> 0) fprintf(fp
, ", ");
338 fprintf(fp
, "0x%04x", c
->values
[i
].u16
);
344 /* Only float base types can be matrices. */
347 for (i
= 0; i
< rows
; i
++) {
348 if (i
> 0) fprintf(fp
, ", ");
349 fprintf(fp
, "0x%08x", c
->values
[i
].u32
);
353 case GLSL_TYPE_FLOAT16
:
354 case GLSL_TYPE_FLOAT
:
355 case GLSL_TYPE_DOUBLE
:
357 for (i
= 0; i
< cols
; i
++) {
358 if (i
> 0) fprintf(fp
, ", ");
359 print_constant(c
->elements
[i
], glsl_get_column_type(type
), state
);
362 switch (glsl_get_base_type(type
)) {
363 case GLSL_TYPE_FLOAT16
:
364 for (i
= 0; i
< rows
; i
++) {
365 if (i
> 0) fprintf(fp
, ", ");
366 fprintf(fp
, "%f", _mesa_half_to_float(c
->values
[i
].u16
));
370 case GLSL_TYPE_FLOAT
:
371 for (i
= 0; i
< rows
; i
++) {
372 if (i
> 0) fprintf(fp
, ", ");
373 fprintf(fp
, "%f", c
->values
[i
].f32
);
377 case GLSL_TYPE_DOUBLE
:
378 for (i
= 0; i
< rows
; i
++) {
379 if (i
> 0) fprintf(fp
, ", ");
380 fprintf(fp
, "%f", c
->values
[i
].f64
);
385 unreachable("Cannot get here from the first level switch");
390 case GLSL_TYPE_UINT64
:
391 case GLSL_TYPE_INT64
:
392 /* Only float base types can be matrices. */
395 for (i
= 0; i
< cols
; i
++) {
396 if (i
> 0) fprintf(fp
, ", ");
397 fprintf(fp
, "0x%08" PRIx64
, c
->values
[i
].u64
);
401 case GLSL_TYPE_STRUCT
:
402 for (i
= 0; i
< c
->num_elements
; i
++) {
403 if (i
> 0) fprintf(fp
, ", ");
405 print_constant(c
->elements
[i
], glsl_get_struct_field(type
, i
), state
);
410 case GLSL_TYPE_ARRAY
:
411 for (i
= 0; i
< c
->num_elements
; i
++) {
412 if (i
> 0) fprintf(fp
, ", ");
414 print_constant(c
->elements
[i
], glsl_get_array_element(type
), state
);
420 unreachable("not reached");
425 get_variable_mode_str(nir_variable_mode mode
, bool want_local_global_mode
)
428 case nir_var_shader_in
:
430 case nir_var_shader_out
:
432 case nir_var_uniform
:
434 case nir_var_mem_ubo
:
436 case nir_var_system_value
:
438 case nir_var_mem_ssbo
:
440 case nir_var_mem_shared
:
442 case nir_var_mem_global
:
444 case nir_var_shader_temp
:
445 return want_local_global_mode
? "shader_temp" : "";
446 case nir_var_function_temp
:
447 return want_local_global_mode
? "function_temp" : "";
454 print_var_decl(nir_variable
*var
, print_state
*state
)
456 FILE *fp
= state
->fp
;
458 fprintf(fp
, "decl_var ");
460 const char *const cent
= (var
->data
.centroid
) ? "centroid " : "";
461 const char *const samp
= (var
->data
.sample
) ? "sample " : "";
462 const char *const patch
= (var
->data
.patch
) ? "patch " : "";
463 const char *const inv
= (var
->data
.invariant
) ? "invariant " : "";
464 fprintf(fp
, "%s%s%s%s%s %s ",
465 cent
, samp
, patch
, inv
, get_variable_mode_str(var
->data
.mode
, false),
466 glsl_interp_mode_name(var
->data
.interpolation
));
468 enum gl_access_qualifier access
= var
->data
.access
;
469 const char *const coher
= (access
& ACCESS_COHERENT
) ? "coherent " : "";
470 const char *const volat
= (access
& ACCESS_VOLATILE
) ? "volatile " : "";
471 const char *const restr
= (access
& ACCESS_RESTRICT
) ? "restrict " : "";
472 const char *const ronly
= (access
& ACCESS_NON_WRITEABLE
) ? "readonly " : "";
473 const char *const wonly
= (access
& ACCESS_NON_READABLE
) ? "writeonly " : "";
474 const char *const reorder
= (access
& ACCESS_CAN_REORDER
) ? "reorderable " : "";
475 fprintf(fp
, "%s%s%s%s%s%s", coher
, volat
, restr
, ronly
, wonly
, reorder
);
477 if (glsl_get_base_type(glsl_without_array(var
->type
)) == GLSL_TYPE_IMAGE
) {
478 fprintf(fp
, "%s ", util_format_short_name(var
->data
.image
.format
));
481 fprintf(fp
, "%s %s", glsl_get_type_name(var
->type
),
482 get_var_name(var
, state
));
484 if (var
->data
.mode
== nir_var_shader_in
||
485 var
->data
.mode
== nir_var_shader_out
||
486 var
->data
.mode
== nir_var_uniform
||
487 var
->data
.mode
== nir_var_mem_ubo
||
488 var
->data
.mode
== nir_var_mem_ssbo
) {
489 const char *loc
= NULL
;
492 switch (state
->shader
->info
.stage
) {
493 case MESA_SHADER_VERTEX
:
494 if (var
->data
.mode
== nir_var_shader_in
)
495 loc
= gl_vert_attrib_name(var
->data
.location
);
496 else if (var
->data
.mode
== nir_var_shader_out
)
497 loc
= gl_varying_slot_name(var
->data
.location
);
499 case MESA_SHADER_GEOMETRY
:
500 if ((var
->data
.mode
== nir_var_shader_in
) ||
501 (var
->data
.mode
== nir_var_shader_out
))
502 loc
= gl_varying_slot_name(var
->data
.location
);
504 case MESA_SHADER_FRAGMENT
:
505 if (var
->data
.mode
== nir_var_shader_in
)
506 loc
= gl_varying_slot_name(var
->data
.location
);
507 else if (var
->data
.mode
== nir_var_shader_out
)
508 loc
= gl_frag_result_name(var
->data
.location
);
510 case MESA_SHADER_TESS_CTRL
:
511 case MESA_SHADER_TESS_EVAL
:
512 case MESA_SHADER_COMPUTE
:
513 case MESA_SHADER_KERNEL
:
520 if (var
->data
.location
== ~0) {
523 snprintf(buf
, sizeof(buf
), "%u", var
->data
.location
);
528 /* For shader I/O vars that have been split to components or packed,
529 * print the fractional location within the input/output.
531 unsigned int num_components
=
532 glsl_get_components(glsl_without_array(var
->type
));
533 const char *components
= NULL
;
534 char components_local
[18] = {'.' /* the rest is 0-filled */};
535 switch (var
->data
.mode
) {
536 case nir_var_shader_in
:
537 case nir_var_shader_out
:
538 if (num_components
< 16 && num_components
!= 0) {
539 const char *xyzw
= comp_mask_string(num_components
);
540 for (int i
= 0; i
< num_components
; i
++)
541 components_local
[i
+ 1] = xyzw
[i
+ var
->data
.location_frac
];
543 components
= components_local
;
550 fprintf(fp
, " (%s%s, %u, %u)%s", loc
,
551 components
? components
: "",
552 var
->data
.driver_location
, var
->data
.binding
,
553 var
->data
.compact
? " compact" : "");
556 if (var
->constant_initializer
) {
557 fprintf(fp
, " = { ");
558 print_constant(var
->constant_initializer
, var
->type
, state
);
563 print_annotation(state
, var
);
567 print_deref_link(const nir_deref_instr
*instr
, bool whole_chain
, print_state
*state
)
569 FILE *fp
= state
->fp
;
571 if (instr
->deref_type
== nir_deref_type_var
) {
572 fprintf(fp
, "%s", get_var_name(instr
->var
, state
));
574 } else if (instr
->deref_type
== nir_deref_type_cast
) {
575 fprintf(fp
, "(%s *)", glsl_get_type_name(instr
->type
));
576 print_src(&instr
->parent
, state
);
580 assert(instr
->parent
.is_ssa
);
581 nir_deref_instr
*parent
=
582 nir_instr_as_deref(instr
->parent
.ssa
->parent_instr
);
584 /* Is the parent we're going to print a bare cast? */
585 const bool is_parent_cast
=
586 whole_chain
&& parent
->deref_type
== nir_deref_type_cast
;
588 /* If we're not printing the whole chain, the parent we print will be a SSA
589 * value that represents a pointer. The only deref type that naturally
590 * gives a pointer is a cast.
592 const bool is_parent_pointer
=
593 !whole_chain
|| parent
->deref_type
== nir_deref_type_cast
;
595 /* Struct derefs have a nice syntax that works on pointers, arrays derefs
598 const bool need_deref
=
599 is_parent_pointer
&& instr
->deref_type
!= nir_deref_type_struct
;
601 /* Cast need extra parens and so * dereferences */
602 if (is_parent_cast
|| need_deref
)
609 print_deref_link(parent
, whole_chain
, state
);
611 print_src(&instr
->parent
, state
);
614 if (is_parent_cast
|| need_deref
)
617 switch (instr
->deref_type
) {
618 case nir_deref_type_struct
:
619 fprintf(fp
, "%s%s", is_parent_pointer
? "->" : ".",
620 glsl_get_struct_elem_name(parent
->type
, instr
->strct
.index
));
623 case nir_deref_type_array
:
624 case nir_deref_type_ptr_as_array
: {
625 if (nir_src_is_const(instr
->arr
.index
)) {
626 fprintf(fp
, "[%"PRId64
"]", nir_src_as_int(instr
->arr
.index
));
629 print_src(&instr
->arr
.index
, state
);
635 case nir_deref_type_array_wildcard
:
640 unreachable("Invalid deref instruction type");
645 print_deref_instr(nir_deref_instr
*instr
, print_state
*state
)
647 FILE *fp
= state
->fp
;
649 print_dest(&instr
->dest
, state
);
651 switch (instr
->deref_type
) {
652 case nir_deref_type_var
:
653 fprintf(fp
, " = deref_var ");
655 case nir_deref_type_array
:
656 case nir_deref_type_array_wildcard
:
657 fprintf(fp
, " = deref_array ");
659 case nir_deref_type_struct
:
660 fprintf(fp
, " = deref_struct ");
662 case nir_deref_type_cast
:
663 fprintf(fp
, " = deref_cast ");
665 case nir_deref_type_ptr_as_array
:
666 fprintf(fp
, " = deref_ptr_as_array ");
669 unreachable("Invalid deref instruction type");
672 /* Only casts naturally return a pointer type */
673 if (instr
->deref_type
!= nir_deref_type_cast
)
676 print_deref_link(instr
, false, state
);
678 fprintf(fp
, " (%s %s) ",
679 get_variable_mode_str(instr
->mode
, true),
680 glsl_get_type_name(instr
->type
));
682 if (instr
->deref_type
!= nir_deref_type_var
&&
683 instr
->deref_type
!= nir_deref_type_cast
) {
684 /* Print the entire chain as a comment */
686 print_deref_link(instr
, true, state
);
690 if (instr
->deref_type
== nir_deref_type_cast
) {
691 fprintf(fp
, " /* ptr_stride=%u */", instr
->cast
.ptr_stride
);
696 vulkan_descriptor_type_name(VkDescriptorType type
)
699 case VK_DESCRIPTOR_TYPE_SAMPLER
: return "sampler";
700 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER
: return "texture+sampler";
701 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE
: return "texture";
702 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE
: return "image";
703 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER
: return "texture-buffer";
704 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER
: return "image-buffer";
705 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER
: return "UBO";
706 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER
: return "SSBO";
707 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC
: return "UBO";
708 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC
: return "SSBO";
709 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT
: return "input-att";
710 case VK_DESCRIPTOR_TYPE_INLINE_UNIFORM_BLOCK_EXT
: return "inline-UBO";
711 default: return "unknown";
716 print_intrinsic_instr(nir_intrinsic_instr
*instr
, print_state
*state
)
718 const nir_intrinsic_info
*info
= &nir_intrinsic_infos
[instr
->intrinsic
];
719 unsigned num_srcs
= info
->num_srcs
;
720 FILE *fp
= state
->fp
;
722 if (info
->has_dest
) {
723 print_dest(&instr
->dest
, state
);
727 fprintf(fp
, "intrinsic %s (", info
->name
);
729 for (unsigned i
= 0; i
< num_srcs
; i
++) {
733 print_src(&instr
->src
[i
], state
);
738 for (unsigned i
= 0; i
< info
->num_indices
; i
++) {
742 fprintf(fp
, "%d", instr
->const_index
[i
]);
747 static const char *index_name
[NIR_INTRINSIC_NUM_INDEX_FLAGS
] = {
748 [NIR_INTRINSIC_BASE
] = "base",
749 [NIR_INTRINSIC_WRMASK
] = "wrmask",
750 [NIR_INTRINSIC_STREAM_ID
] = "stream-id",
751 [NIR_INTRINSIC_UCP_ID
] = "ucp-id",
752 [NIR_INTRINSIC_RANGE
] = "range",
753 [NIR_INTRINSIC_DESC_SET
] = "desc-set",
754 [NIR_INTRINSIC_BINDING
] = "binding",
755 [NIR_INTRINSIC_COMPONENT
] = "component",
756 [NIR_INTRINSIC_INTERP_MODE
] = "interp_mode",
757 [NIR_INTRINSIC_REDUCTION_OP
] = "reduction_op",
758 [NIR_INTRINSIC_CLUSTER_SIZE
] = "cluster_size",
759 [NIR_INTRINSIC_PARAM_IDX
] = "param_idx",
760 [NIR_INTRINSIC_IMAGE_DIM
] = "image_dim",
761 [NIR_INTRINSIC_IMAGE_ARRAY
] = "image_array",
762 [NIR_INTRINSIC_ACCESS
] = "access",
763 [NIR_INTRINSIC_SRC_ACCESS
] = "src-access",
764 [NIR_INTRINSIC_DST_ACCESS
] = "dst-access",
765 [NIR_INTRINSIC_FORMAT
] = "format",
766 [NIR_INTRINSIC_ALIGN_MUL
] = "align_mul",
767 [NIR_INTRINSIC_ALIGN_OFFSET
] = "align_offset",
768 [NIR_INTRINSIC_DESC_TYPE
] = "desc_type",
769 [NIR_INTRINSIC_TYPE
] = "type",
770 [NIR_INTRINSIC_SWIZZLE_MASK
] = "swizzle_mask",
771 [NIR_INTRINSIC_DRIVER_LOCATION
] = "driver_location",
772 [NIR_INTRINSIC_MEMORY_SEMANTICS
] = "mem_semantics",
773 [NIR_INTRINSIC_MEMORY_MODES
] = "mem_modes",
774 [NIR_INTRINSIC_MEMORY_SCOPE
] = "mem_scope",
776 for (unsigned idx
= 1; idx
< NIR_INTRINSIC_NUM_INDEX_FLAGS
; idx
++) {
777 if (!info
->index_map
[idx
])
781 case NIR_INTRINSIC_WRMASK
: {
782 /* special case wrmask to show it as a writemask.. */
783 unsigned wrmask
= nir_intrinsic_write_mask(instr
);
784 fprintf(fp
, " wrmask=");
785 for (unsigned i
= 0; i
< instr
->num_components
; i
++)
786 if ((wrmask
>> i
) & 1)
787 fprintf(fp
, "%c", comp_mask_string(instr
->num_components
)[i
]);
791 case NIR_INTRINSIC_REDUCTION_OP
: {
792 nir_op reduction_op
= nir_intrinsic_reduction_op(instr
);
793 fprintf(fp
, " reduction_op=%s", nir_op_infos
[reduction_op
].name
);
797 case NIR_INTRINSIC_IMAGE_DIM
: {
798 static const char *dim_name
[] = {
799 [GLSL_SAMPLER_DIM_1D
] = "1D",
800 [GLSL_SAMPLER_DIM_2D
] = "2D",
801 [GLSL_SAMPLER_DIM_3D
] = "3D",
802 [GLSL_SAMPLER_DIM_CUBE
] = "Cube",
803 [GLSL_SAMPLER_DIM_RECT
] = "Rect",
804 [GLSL_SAMPLER_DIM_BUF
] = "Buf",
805 [GLSL_SAMPLER_DIM_MS
] = "2D-MSAA",
806 [GLSL_SAMPLER_DIM_SUBPASS
] = "Subpass",
807 [GLSL_SAMPLER_DIM_SUBPASS_MS
] = "Subpass-MSAA",
809 enum glsl_sampler_dim dim
= nir_intrinsic_image_dim(instr
);
810 assert(dim
< ARRAY_SIZE(dim_name
) && dim_name
[dim
]);
811 fprintf(fp
, " image_dim=%s", dim_name
[dim
]);
815 case NIR_INTRINSIC_IMAGE_ARRAY
: {
816 bool array
= nir_intrinsic_image_array(instr
);
817 fprintf(fp
, " image_array=%s", array
? "true" : "false");
821 case NIR_INTRINSIC_DESC_TYPE
: {
822 VkDescriptorType desc_type
= nir_intrinsic_desc_type(instr
);
823 fprintf(fp
, " desc_type=%s", vulkan_descriptor_type_name(desc_type
));
827 case NIR_INTRINSIC_TYPE
: {
828 nir_alu_type type
= nir_intrinsic_type(instr
);
829 unsigned size
= nir_alu_type_get_type_size(type
);
831 switch (nir_alu_type_get_base_type(type
)) {
832 case nir_type_int
: name
= "int"; break;
833 case nir_type_uint
: name
= "uint"; break;
834 case nir_type_bool
: name
= "bool"; break;
835 case nir_type_float
: name
= "float"; break;
836 default: name
= "invalid";
839 fprintf(fp
, " type=%s%u", name
, size
);
841 fprintf(fp
, " type=%s", name
);
845 case NIR_INTRINSIC_SWIZZLE_MASK
: {
846 fprintf(fp
, " swizzle_mask=");
847 unsigned mask
= nir_intrinsic_swizzle_mask(instr
);
848 if (instr
->intrinsic
== nir_intrinsic_quad_swizzle_amd
) {
849 for (unsigned i
= 0; i
< 4; i
++)
850 fprintf(fp
, "%d", (mask
>> (i
* 2) & 3));
851 } else if (instr
->intrinsic
== nir_intrinsic_masked_swizzle_amd
) {
852 fprintf(fp
, "((id & %d) | %d) ^ %d", mask
& 0x1F,
854 (mask
>> 10) & 0x1F);
856 fprintf(fp
, "%d", mask
);
861 case NIR_INTRINSIC_MEMORY_SEMANTICS
: {
862 nir_memory_semantics semantics
= nir_intrinsic_memory_semantics(instr
);
863 fprintf(fp
, " mem_semantics=");
864 switch (semantics
& (NIR_MEMORY_ACQUIRE
| NIR_MEMORY_RELEASE
)) {
865 case 0: fprintf(fp
, "NONE"); break;
866 case NIR_MEMORY_ACQUIRE
: fprintf(fp
, "ACQ"); break;
867 case NIR_MEMORY_RELEASE
: fprintf(fp
, "REL"); break;
868 default: fprintf(fp
, "ACQ|REL"); break;
870 if (semantics
& (NIR_MEMORY_MAKE_AVAILABLE
)) fprintf(fp
, "|AVAILABLE");
871 if (semantics
& (NIR_MEMORY_MAKE_VISIBLE
)) fprintf(fp
, "|VISIBLE");
875 case NIR_INTRINSIC_MEMORY_MODES
: {
876 fprintf(fp
, " mem_modes=");
877 unsigned int modes
= nir_intrinsic_memory_modes(instr
);
879 nir_variable_mode m
= u_bit_scan(&modes
);
880 fprintf(fp
, "%s%s", get_variable_mode_str(1 << m
, true), modes
? "|" : "");
885 case NIR_INTRINSIC_MEMORY_SCOPE
: {
886 fprintf(fp
, " mem_scope=");
887 switch (nir_intrinsic_memory_scope(instr
)) {
888 case NIR_SCOPE_DEVICE
: fprintf(fp
, "DEVICE"); break;
889 case NIR_SCOPE_QUEUE_FAMILY
: fprintf(fp
, "QUEUE_FAMILY"); break;
890 case NIR_SCOPE_WORKGROUP
: fprintf(fp
, "WORKGROUP"); break;
891 case NIR_SCOPE_SUBGROUP
: fprintf(fp
, "SUBGROUP"); break;
892 case NIR_SCOPE_INVOCATION
: fprintf(fp
, "INVOCATION"); break;
898 unsigned off
= info
->index_map
[idx
] - 1;
899 assert(index_name
[idx
]); /* forgot to update index_name table? */
900 fprintf(fp
, " %s=%d", index_name
[idx
], instr
->const_index
[off
]);
910 struct exec_list
*var_list
= NULL
;
912 switch (instr
->intrinsic
) {
913 case nir_intrinsic_load_uniform
:
914 var_list
= &state
->shader
->uniforms
;
916 case nir_intrinsic_load_input
:
917 case nir_intrinsic_load_interpolated_input
:
918 case nir_intrinsic_load_per_vertex_input
:
919 var_list
= &state
->shader
->inputs
;
921 case nir_intrinsic_load_output
:
922 case nir_intrinsic_store_output
:
923 case nir_intrinsic_store_per_vertex_output
:
924 var_list
= &state
->shader
->outputs
;
930 nir_foreach_variable(var
, var_list
) {
931 if ((var
->data
.driver_location
== nir_intrinsic_base(instr
)) &&
932 (instr
->intrinsic
== nir_intrinsic_load_uniform
||
933 (nir_intrinsic_component(instr
) >= var
->data
.location_frac
&&
934 nir_intrinsic_component(instr
) <
935 (var
->data
.location_frac
+ glsl_get_components(var
->type
)))) &&
937 fprintf(fp
, "\t/* %s */", var
->name
);
944 print_tex_instr(nir_tex_instr
*instr
, print_state
*state
)
946 FILE *fp
= state
->fp
;
948 print_dest(&instr
->dest
, state
);
968 case nir_texop_txf_ms
:
969 fprintf(fp
, "txf_ms ");
971 case nir_texop_txf_ms_fb
:
972 fprintf(fp
, "txf_ms_fb ");
974 case nir_texop_txf_ms_mcs
:
975 fprintf(fp
, "txf_ms_mcs ");
986 case nir_texop_query_levels
:
987 fprintf(fp
, "query_levels ");
989 case nir_texop_texture_samples
:
990 fprintf(fp
, "texture_samples ");
992 case nir_texop_samples_identical
:
993 fprintf(fp
, "samples_identical ");
995 case nir_texop_tex_prefetch
:
996 fprintf(fp
, "tex (pre-dispatchable) ");
998 case nir_texop_fragment_fetch
:
999 fprintf(fp
, "fragment_fetch ");
1001 case nir_texop_fragment_mask_fetch
:
1002 fprintf(fp
, "fragment_mask_fetch ");
1005 unreachable("Invalid texture operation");
1009 bool has_texture_deref
= false, has_sampler_deref
= false;
1010 for (unsigned i
= 0; i
< instr
->num_srcs
; i
++) {
1015 print_src(&instr
->src
[i
].src
, state
);
1018 switch(instr
->src
[i
].src_type
) {
1019 case nir_tex_src_coord
:
1020 fprintf(fp
, "(coord)");
1022 case nir_tex_src_projector
:
1023 fprintf(fp
, "(projector)");
1025 case nir_tex_src_comparator
:
1026 fprintf(fp
, "(comparator)");
1028 case nir_tex_src_offset
:
1029 fprintf(fp
, "(offset)");
1031 case nir_tex_src_bias
:
1032 fprintf(fp
, "(bias)");
1034 case nir_tex_src_lod
:
1035 fprintf(fp
, "(lod)");
1037 case nir_tex_src_min_lod
:
1038 fprintf(fp
, "(min_lod)");
1040 case nir_tex_src_ms_index
:
1041 fprintf(fp
, "(ms_index)");
1043 case nir_tex_src_ms_mcs
:
1044 fprintf(fp
, "(ms_mcs)");
1046 case nir_tex_src_ddx
:
1047 fprintf(fp
, "(ddx)");
1049 case nir_tex_src_ddy
:
1050 fprintf(fp
, "(ddy)");
1052 case nir_tex_src_texture_deref
:
1053 has_texture_deref
= true;
1054 fprintf(fp
, "(texture_deref)");
1056 case nir_tex_src_sampler_deref
:
1057 has_sampler_deref
= true;
1058 fprintf(fp
, "(sampler_deref)");
1060 case nir_tex_src_texture_offset
:
1061 fprintf(fp
, "(texture_offset)");
1063 case nir_tex_src_sampler_offset
:
1064 fprintf(fp
, "(sampler_offset)");
1066 case nir_tex_src_texture_handle
:
1067 fprintf(fp
, "(texture_handle)");
1069 case nir_tex_src_sampler_handle
:
1070 fprintf(fp
, "(sampler_handle)");
1072 case nir_tex_src_plane
:
1073 fprintf(fp
, "(plane)");
1077 unreachable("Invalid texture source type");
1082 if (instr
->op
== nir_texop_tg4
) {
1083 fprintf(fp
, ", %u (gather_component)", instr
->component
);
1086 if (nir_tex_instr_has_explicit_tg4_offsets(instr
)) {
1087 fprintf(fp
, ", { (%i, %i)", instr
->tg4_offsets
[0][0], instr
->tg4_offsets
[0][1]);
1088 for (unsigned i
= 1; i
< 4; ++i
)
1089 fprintf(fp
, ", (%i, %i)", instr
->tg4_offsets
[i
][0],
1090 instr
->tg4_offsets
[i
][1]);
1091 fprintf(fp
, " } (offsets)");
1094 if (instr
->op
!= nir_texop_txf_ms_fb
) {
1095 if (!has_texture_deref
) {
1096 fprintf(fp
, ", %u (texture)", instr
->texture_index
);
1099 if (!has_sampler_deref
) {
1100 fprintf(fp
, ", %u (sampler)", instr
->sampler_index
);
1104 if (instr
->texture_non_uniform
) {
1105 fprintf(fp
, ", texture non-uniform");
1108 if (instr
->sampler_non_uniform
) {
1109 fprintf(fp
, ", sampler non-uniform");
1114 print_call_instr(nir_call_instr
*instr
, print_state
*state
)
1116 FILE *fp
= state
->fp
;
1118 fprintf(fp
, "call %s ", instr
->callee
->name
);
1120 for (unsigned i
= 0; i
< instr
->num_params
; i
++) {
1124 print_src(&instr
->params
[i
], state
);
1129 print_load_const_instr(nir_load_const_instr
*instr
, print_state
*state
)
1131 FILE *fp
= state
->fp
;
1133 print_ssa_def(&instr
->def
, state
);
1135 fprintf(fp
, " = load_const (");
1137 for (unsigned i
= 0; i
< instr
->def
.num_components
; i
++) {
1142 * we don't really know the type of the constant (if it will be used as a
1143 * float or an int), so just print the raw constant in hex for fidelity
1144 * and then print the float in a comment for readability.
1147 switch (instr
->def
.bit_size
) {
1149 fprintf(fp
, "0x%16" PRIx64
" /* %f */", instr
->value
[i
].u64
,
1150 instr
->value
[i
].f64
);
1153 fprintf(fp
, "0x%08x /* %f */", instr
->value
[i
].u32
, instr
->value
[i
].f32
);
1156 fprintf(fp
, "0x%04x /* %f */", instr
->value
[i
].u16
,
1157 _mesa_half_to_float(instr
->value
[i
].u16
));
1160 fprintf(fp
, "0x%02x", instr
->value
[i
].u8
);
1163 fprintf(fp
, "%s", instr
->value
[i
].b
? "true" : "false");
1172 print_jump_instr(nir_jump_instr
*instr
, print_state
*state
)
1174 FILE *fp
= state
->fp
;
1176 switch (instr
->type
) {
1177 case nir_jump_break
:
1178 fprintf(fp
, "break");
1181 case nir_jump_continue
:
1182 fprintf(fp
, "continue");
1185 case nir_jump_return
:
1186 fprintf(fp
, "return");
1192 print_ssa_undef_instr(nir_ssa_undef_instr
* instr
, print_state
*state
)
1194 FILE *fp
= state
->fp
;
1195 print_ssa_def(&instr
->def
, state
);
1196 fprintf(fp
, " = undefined");
1200 print_phi_instr(nir_phi_instr
*instr
, print_state
*state
)
1202 FILE *fp
= state
->fp
;
1203 print_dest(&instr
->dest
, state
);
1204 fprintf(fp
, " = phi ");
1205 nir_foreach_phi_src(src
, instr
) {
1206 if (&src
->node
!= exec_list_get_head(&instr
->srcs
))
1209 fprintf(fp
, "block_%u: ", src
->pred
->index
);
1210 print_src(&src
->src
, state
);
1215 print_parallel_copy_instr(nir_parallel_copy_instr
*instr
, print_state
*state
)
1217 FILE *fp
= state
->fp
;
1218 nir_foreach_parallel_copy_entry(entry
, instr
) {
1219 if (&entry
->node
!= exec_list_get_head(&instr
->entries
))
1222 print_dest(&entry
->dest
, state
);
1224 print_src(&entry
->src
, state
);
1229 print_instr(const nir_instr
*instr
, print_state
*state
, unsigned tabs
)
1231 FILE *fp
= state
->fp
;
1232 print_tabs(tabs
, fp
);
1234 switch (instr
->type
) {
1235 case nir_instr_type_alu
:
1236 print_alu_instr(nir_instr_as_alu(instr
), state
);
1239 case nir_instr_type_deref
:
1240 print_deref_instr(nir_instr_as_deref(instr
), state
);
1243 case nir_instr_type_call
:
1244 print_call_instr(nir_instr_as_call(instr
), state
);
1247 case nir_instr_type_intrinsic
:
1248 print_intrinsic_instr(nir_instr_as_intrinsic(instr
), state
);
1251 case nir_instr_type_tex
:
1252 print_tex_instr(nir_instr_as_tex(instr
), state
);
1255 case nir_instr_type_load_const
:
1256 print_load_const_instr(nir_instr_as_load_const(instr
), state
);
1259 case nir_instr_type_jump
:
1260 print_jump_instr(nir_instr_as_jump(instr
), state
);
1263 case nir_instr_type_ssa_undef
:
1264 print_ssa_undef_instr(nir_instr_as_ssa_undef(instr
), state
);
1267 case nir_instr_type_phi
:
1268 print_phi_instr(nir_instr_as_phi(instr
), state
);
1271 case nir_instr_type_parallel_copy
:
1272 print_parallel_copy_instr(nir_instr_as_parallel_copy(instr
), state
);
1276 unreachable("Invalid instruction type");
1282 compare_block_index(const void *p1
, const void *p2
)
1284 const nir_block
*block1
= *((const nir_block
**) p1
);
1285 const nir_block
*block2
= *((const nir_block
**) p2
);
1287 return (int) block1
->index
- (int) block2
->index
;
1290 static void print_cf_node(nir_cf_node
*node
, print_state
*state
,
1294 print_block(nir_block
*block
, print_state
*state
, unsigned tabs
)
1296 FILE *fp
= state
->fp
;
1298 print_tabs(tabs
, fp
);
1299 fprintf(fp
, "block block_%u:\n", block
->index
);
1301 /* sort the predecessors by index so we consistently print the same thing */
1304 malloc(block
->predecessors
->entries
* sizeof(nir_block
*));
1307 set_foreach(block
->predecessors
, entry
) {
1308 preds
[i
++] = (nir_block
*) entry
->key
;
1311 qsort(preds
, block
->predecessors
->entries
, sizeof(nir_block
*),
1312 compare_block_index
);
1314 print_tabs(tabs
, fp
);
1315 fprintf(fp
, "/* preds: ");
1316 for (unsigned i
= 0; i
< block
->predecessors
->entries
; i
++) {
1317 fprintf(fp
, "block_%u ", preds
[i
]->index
);
1319 fprintf(fp
, "*/\n");
1323 nir_foreach_instr(instr
, block
) {
1324 print_instr(instr
, state
, tabs
);
1326 print_annotation(state
, instr
);
1329 print_tabs(tabs
, fp
);
1330 fprintf(fp
, "/* succs: ");
1331 for (unsigned i
= 0; i
< 2; i
++)
1332 if (block
->successors
[i
]) {
1333 fprintf(fp
, "block_%u ", block
->successors
[i
]->index
);
1335 fprintf(fp
, "*/\n");
1339 print_if(nir_if
*if_stmt
, print_state
*state
, unsigned tabs
)
1341 FILE *fp
= state
->fp
;
1343 print_tabs(tabs
, fp
);
1345 print_src(&if_stmt
->condition
, state
);
1346 fprintf(fp
, " {\n");
1347 foreach_list_typed(nir_cf_node
, node
, node
, &if_stmt
->then_list
) {
1348 print_cf_node(node
, state
, tabs
+ 1);
1350 print_tabs(tabs
, fp
);
1351 fprintf(fp
, "} else {\n");
1352 foreach_list_typed(nir_cf_node
, node
, node
, &if_stmt
->else_list
) {
1353 print_cf_node(node
, state
, tabs
+ 1);
1355 print_tabs(tabs
, fp
);
1360 print_loop(nir_loop
*loop
, print_state
*state
, unsigned tabs
)
1362 FILE *fp
= state
->fp
;
1364 print_tabs(tabs
, fp
);
1365 fprintf(fp
, "loop {\n");
1366 foreach_list_typed(nir_cf_node
, node
, node
, &loop
->body
) {
1367 print_cf_node(node
, state
, tabs
+ 1);
1369 print_tabs(tabs
, fp
);
1374 print_cf_node(nir_cf_node
*node
, print_state
*state
, unsigned int tabs
)
1376 switch (node
->type
) {
1377 case nir_cf_node_block
:
1378 print_block(nir_cf_node_as_block(node
), state
, tabs
);
1381 case nir_cf_node_if
:
1382 print_if(nir_cf_node_as_if(node
), state
, tabs
);
1385 case nir_cf_node_loop
:
1386 print_loop(nir_cf_node_as_loop(node
), state
, tabs
);
1390 unreachable("Invalid CFG node type");
1395 print_function_impl(nir_function_impl
*impl
, print_state
*state
)
1397 FILE *fp
= state
->fp
;
1399 fprintf(fp
, "\nimpl %s ", impl
->function
->name
);
1403 nir_foreach_variable(var
, &impl
->locals
) {
1405 print_var_decl(var
, state
);
1408 foreach_list_typed(nir_register
, reg
, node
, &impl
->registers
) {
1410 print_register_decl(reg
, state
);
1413 nir_index_blocks(impl
);
1415 foreach_list_typed(nir_cf_node
, node
, node
, &impl
->body
) {
1416 print_cf_node(node
, state
, 1);
1419 fprintf(fp
, "\tblock block_%u:\n}\n\n", impl
->end_block
->index
);
1423 print_function(nir_function
*function
, print_state
*state
)
1425 FILE *fp
= state
->fp
;
1427 fprintf(fp
, "decl_function %s (%d params)", function
->name
,
1428 function
->num_params
);
1432 if (function
->impl
!= NULL
) {
1433 print_function_impl(function
->impl
, state
);
1439 init_print_state(print_state
*state
, nir_shader
*shader
, FILE *fp
)
1442 state
->shader
= shader
;
1443 state
->ht
= _mesa_pointer_hash_table_create(NULL
);
1444 state
->syms
= _mesa_set_create(NULL
, _mesa_hash_string
,
1445 _mesa_key_string_equal
);
1450 destroy_print_state(print_state
*state
)
1452 _mesa_hash_table_destroy(state
->ht
, NULL
);
1453 _mesa_set_destroy(state
->syms
, NULL
);
1457 nir_print_shader_annotated(nir_shader
*shader
, FILE *fp
,
1458 struct hash_table
*annotations
)
1461 init_print_state(&state
, shader
, fp
);
1463 state
.annotations
= annotations
;
1465 fprintf(fp
, "shader: %s\n", gl_shader_stage_name(shader
->info
.stage
));
1467 if (shader
->info
.name
)
1468 fprintf(fp
, "name: %s\n", shader
->info
.name
);
1470 if (shader
->info
.label
)
1471 fprintf(fp
, "label: %s\n", shader
->info
.label
);
1473 if (gl_shader_stage_is_compute(shader
->info
.stage
)) {
1474 fprintf(fp
, "local-size: %u, %u, %u%s\n",
1475 shader
->info
.cs
.local_size
[0],
1476 shader
->info
.cs
.local_size
[1],
1477 shader
->info
.cs
.local_size
[2],
1478 shader
->info
.cs
.local_size_variable
? " (variable)" : "");
1479 fprintf(fp
, "shared-size: %u\n", shader
->info
.cs
.shared_size
);
1482 fprintf(fp
, "inputs: %u\n", shader
->num_inputs
);
1483 fprintf(fp
, "outputs: %u\n", shader
->num_outputs
);
1484 fprintf(fp
, "uniforms: %u\n", shader
->num_uniforms
);
1485 fprintf(fp
, "shared: %u\n", shader
->num_shared
);
1486 if (shader
->scratch_size
)
1487 fprintf(fp
, "scratch: %u\n", shader
->scratch_size
);
1489 nir_foreach_variable(var
, &shader
->uniforms
) {
1490 print_var_decl(var
, &state
);
1493 nir_foreach_variable(var
, &shader
->inputs
) {
1494 print_var_decl(var
, &state
);
1497 nir_foreach_variable(var
, &shader
->outputs
) {
1498 print_var_decl(var
, &state
);
1501 nir_foreach_variable(var
, &shader
->shared
) {
1502 print_var_decl(var
, &state
);
1505 nir_foreach_variable(var
, &shader
->globals
) {
1506 print_var_decl(var
, &state
);
1509 nir_foreach_variable(var
, &shader
->system_values
) {
1510 print_var_decl(var
, &state
);
1513 foreach_list_typed(nir_function
, func
, node
, &shader
->functions
) {
1514 print_function(func
, &state
);
1517 destroy_print_state(&state
);
1521 nir_print_shader(nir_shader
*shader
, FILE *fp
)
1523 nir_print_shader_annotated(shader
, fp
, NULL
);
1528 nir_print_instr(const nir_instr
*instr
, FILE *fp
)
1530 print_state state
= {
1533 print_instr(instr
, &state
, 0);
1538 nir_print_deref(const nir_deref_instr
*deref
, FILE *fp
)
1540 print_state state
= {
1543 print_deref_link(deref
, true, &state
);