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
)
65 if (!state
->annotations
)
68 struct hash_entry
*entry
= _mesa_hash_table_search(state
->annotations
, obj
);
72 const char *note
= entry
->data
;
73 _mesa_hash_table_remove(state
->annotations
, entry
);
75 fprintf(stderr
, "%s\n\n", note
);
79 print_register(nir_register
*reg
, print_state
*state
)
82 if (reg
->name
!= NULL
)
83 fprintf(fp
, "/* %s */ ", reg
->name
);
85 fprintf(fp
, "gr%u", reg
->index
);
87 fprintf(fp
, "r%u", reg
->index
);
90 static const char *sizes
[] = { "error", "vec1", "vec2", "vec3", "vec4",
91 "error", "error", "error", "vec8",
92 "error", "error", "error", "error",
93 "error", "error", "error", "vec16"};
96 print_register_decl(nir_register
*reg
, print_state
*state
)
99 fprintf(fp
, "decl_reg %s %u ", sizes
[reg
->num_components
], reg
->bit_size
);
101 fprintf(fp
, "(packed) ");
102 print_register(reg
, state
);
103 if (reg
->num_array_elems
!= 0)
104 fprintf(fp
, "[%u]", reg
->num_array_elems
);
109 print_ssa_def(nir_ssa_def
*def
, print_state
*state
)
111 FILE *fp
= state
->fp
;
112 if (def
->name
!= NULL
)
113 fprintf(fp
, "/* %s */ ", def
->name
);
114 fprintf(fp
, "%s %u ssa_%u", sizes
[def
->num_components
], def
->bit_size
,
119 print_ssa_use(nir_ssa_def
*def
, print_state
*state
)
121 FILE *fp
= state
->fp
;
122 if (def
->name
!= NULL
)
123 fprintf(fp
, "/* %s */ ", def
->name
);
124 fprintf(fp
, "ssa_%u", def
->index
);
127 static void print_src(const nir_src
*src
, print_state
*state
);
130 print_reg_src(const nir_reg_src
*src
, print_state
*state
)
132 FILE *fp
= state
->fp
;
133 print_register(src
->reg
, state
);
134 if (src
->reg
->num_array_elems
!= 0) {
135 fprintf(fp
, "[%u", src
->base_offset
);
136 if (src
->indirect
!= NULL
) {
138 print_src(src
->indirect
, state
);
145 print_reg_dest(nir_reg_dest
*dest
, print_state
*state
)
147 FILE *fp
= state
->fp
;
148 print_register(dest
->reg
, state
);
149 if (dest
->reg
->num_array_elems
!= 0) {
150 fprintf(fp
, "[%u", dest
->base_offset
);
151 if (dest
->indirect
!= NULL
) {
153 print_src(dest
->indirect
, state
);
160 print_src(const nir_src
*src
, print_state
*state
)
163 print_ssa_use(src
->ssa
, state
);
165 print_reg_src(&src
->reg
, state
);
169 print_dest(nir_dest
*dest
, print_state
*state
)
172 print_ssa_def(&dest
->ssa
, state
);
174 print_reg_dest(&dest
->reg
, state
);
178 print_alu_src(nir_alu_instr
*instr
, unsigned src
, print_state
*state
)
180 FILE *fp
= state
->fp
;
182 if (instr
->src
[src
].negate
)
184 if (instr
->src
[src
].abs
)
187 print_src(&instr
->src
[src
].src
, state
);
189 bool print_swizzle
= false;
190 nir_component_mask_t used_channels
= 0;
192 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++) {
193 if (!nir_alu_instr_channel_used(instr
, src
, i
))
198 if (instr
->src
[src
].swizzle
[i
] != i
) {
199 print_swizzle
= true;
204 unsigned live_channels
= nir_src_num_components(instr
->src
[src
].src
);
206 if (print_swizzle
|| used_channels
!= live_channels
) {
208 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++) {
209 if (!nir_alu_instr_channel_used(instr
, src
, i
))
212 fprintf(fp
, "%c", "xyzw"[instr
->src
[src
].swizzle
[i
]]);
216 if (instr
->src
[src
].abs
)
221 print_alu_dest(nir_alu_dest
*dest
, print_state
*state
)
223 FILE *fp
= state
->fp
;
224 /* we're going to print the saturate modifier later, after the opcode */
226 print_dest(&dest
->dest
, state
);
228 if (!dest
->dest
.is_ssa
&&
229 dest
->write_mask
!= (1 << dest
->dest
.reg
.reg
->num_components
) - 1) {
231 for (unsigned i
= 0; i
< NIR_MAX_VEC_COMPONENTS
; i
++)
232 if ((dest
->write_mask
>> i
) & 1)
233 fprintf(fp
, "%c", "xyzw"[i
]);
238 print_alu_instr(nir_alu_instr
*instr
, print_state
*state
)
240 FILE *fp
= state
->fp
;
242 print_alu_dest(&instr
->dest
, state
);
244 fprintf(fp
, " = %s", nir_op_infos
[instr
->op
].name
);
247 if (instr
->dest
.saturate
)
251 for (unsigned i
= 0; i
< nir_op_infos
[instr
->op
].num_inputs
; i
++) {
255 print_alu_src(instr
, i
, state
);
260 get_var_name(nir_variable
*var
, print_state
*state
)
262 if (state
->ht
== NULL
)
263 return var
->name
? var
->name
: "unnamed";
267 struct hash_entry
*entry
= _mesa_hash_table_search(state
->ht
, var
);
272 if (var
->name
== NULL
) {
273 name
= ralloc_asprintf(state
->syms
, "@%u", state
->index
++);
275 struct set_entry
*set_entry
= _mesa_set_search(state
->syms
, var
->name
);
276 if (set_entry
!= NULL
) {
277 /* we have a collision with another name, append an @ + a unique
279 name
= ralloc_asprintf(state
->syms
, "%s@%u", var
->name
,
282 /* Mark this one as seen */
283 _mesa_set_add(state
->syms
, var
->name
);
288 _mesa_hash_table_insert(state
->ht
, var
, name
);
294 print_constant(nir_constant
*c
, const struct glsl_type
*type
, print_state
*state
)
296 FILE *fp
= state
->fp
;
297 const unsigned rows
= glsl_get_vector_elements(type
);
298 const unsigned cols
= glsl_get_matrix_columns(type
);
301 switch (glsl_get_base_type(type
)) {
303 /* Only float base types can be matrices. */
306 for (i
= 0; i
< rows
; i
++) {
307 if (i
> 0) fprintf(fp
, ", ");
308 fprintf(fp
, "%s", c
->values
[0].b
[i
] ? "true" : "false");
312 case GLSL_TYPE_UINT8
:
314 /* Only float base types can be matrices. */
317 for (i
= 0; i
< rows
; i
++) {
318 if (i
> 0) fprintf(fp
, ", ");
319 fprintf(fp
, "0x%02x", c
->values
[0].u8
[i
]);
323 case GLSL_TYPE_UINT16
:
324 case GLSL_TYPE_INT16
:
325 /* Only float base types can be matrices. */
328 for (i
= 0; i
< rows
; i
++) {
329 if (i
> 0) fprintf(fp
, ", ");
330 fprintf(fp
, "0x%04x", c
->values
[0].u16
[i
]);
336 /* Only float base types can be matrices. */
339 for (i
= 0; i
< rows
; i
++) {
340 if (i
> 0) fprintf(fp
, ", ");
341 fprintf(fp
, "0x%08x", c
->values
[0].u32
[i
]);
345 case GLSL_TYPE_FLOAT16
:
346 for (i
= 0; i
< cols
; i
++) {
347 for (j
= 0; j
< rows
; j
++) {
348 if (i
+ j
> 0) fprintf(fp
, ", ");
349 fprintf(fp
, "%f", _mesa_half_to_float(c
->values
[i
].u16
[j
]));
354 case GLSL_TYPE_FLOAT
:
355 for (i
= 0; i
< cols
; i
++) {
356 for (j
= 0; j
< rows
; j
++) {
357 if (i
+ j
> 0) fprintf(fp
, ", ");
358 fprintf(fp
, "%f", c
->values
[i
].f32
[j
]);
363 case GLSL_TYPE_DOUBLE
:
364 for (i
= 0; i
< cols
; i
++) {
365 for (j
= 0; j
< rows
; j
++) {
366 if (i
+ j
> 0) fprintf(fp
, ", ");
367 fprintf(fp
, "%f", c
->values
[i
].f64
[j
]);
372 case GLSL_TYPE_UINT64
:
373 case GLSL_TYPE_INT64
:
374 /* Only float base types can be matrices. */
377 for (i
= 0; i
< cols
; i
++) {
378 if (i
> 0) fprintf(fp
, ", ");
379 fprintf(fp
, "0x%08" PRIx64
, c
->values
[0].u64
[i
]);
383 case GLSL_TYPE_STRUCT
:
384 for (i
= 0; i
< c
->num_elements
; i
++) {
385 if (i
> 0) fprintf(fp
, ", ");
387 print_constant(c
->elements
[i
], glsl_get_struct_field(type
, i
), state
);
392 case GLSL_TYPE_ARRAY
:
393 for (i
= 0; i
< c
->num_elements
; i
++) {
394 if (i
> 0) fprintf(fp
, ", ");
396 print_constant(c
->elements
[i
], glsl_get_array_element(type
), state
);
402 unreachable("not reached");
407 get_variable_mode_str(nir_variable_mode mode
, bool want_local_global_mode
)
410 case nir_var_shader_in
:
412 case nir_var_shader_out
:
414 case nir_var_uniform
:
416 case nir_var_mem_ubo
:
418 case nir_var_system_value
:
420 case nir_var_mem_ssbo
:
422 case nir_var_mem_shared
:
424 case nir_var_mem_global
:
426 case nir_var_shader_temp
:
427 return want_local_global_mode
? "shader_temp" : "";
428 case nir_var_function_temp
:
429 return want_local_global_mode
? "function_temp" : "";
436 print_var_decl(nir_variable
*var
, print_state
*state
)
438 FILE *fp
= state
->fp
;
440 fprintf(fp
, "decl_var ");
442 const char *const cent
= (var
->data
.centroid
) ? "centroid " : "";
443 const char *const samp
= (var
->data
.sample
) ? "sample " : "";
444 const char *const patch
= (var
->data
.patch
) ? "patch " : "";
445 const char *const inv
= (var
->data
.invariant
) ? "invariant " : "";
446 fprintf(fp
, "%s%s%s%s%s %s ",
447 cent
, samp
, patch
, inv
, get_variable_mode_str(var
->data
.mode
, false),
448 glsl_interp_mode_name(var
->data
.interpolation
));
450 enum gl_access_qualifier access
= var
->data
.image
.access
;
451 const char *const coher
= (access
& ACCESS_COHERENT
) ? "coherent " : "";
452 const char *const volat
= (access
& ACCESS_VOLATILE
) ? "volatile " : "";
453 const char *const restr
= (access
& ACCESS_RESTRICT
) ? "restrict " : "";
454 const char *const ronly
= (access
& ACCESS_NON_WRITEABLE
) ? "readonly " : "";
455 const char *const wonly
= (access
& ACCESS_NON_READABLE
) ? "writeonly " : "";
456 fprintf(fp
, "%s%s%s%s%s", coher
, volat
, restr
, ronly
, wonly
);
458 #define FORMAT_CASE(x) case x: fprintf(stderr, #x " "); break
459 switch (var
->data
.image
.format
) {
460 FORMAT_CASE(GL_RGBA32F
);
461 FORMAT_CASE(GL_RGBA32UI
);
462 FORMAT_CASE(GL_RGBA32I
);
463 FORMAT_CASE(GL_R32F
);
464 FORMAT_CASE(GL_R32UI
);
465 FORMAT_CASE(GL_R32I
);
466 FORMAT_CASE(GL_RG32F
);
467 FORMAT_CASE(GL_RG32UI
);
468 FORMAT_CASE(GL_RG32I
);
471 FORMAT_CASE(GL_RGBA8
);
472 FORMAT_CASE(GL_R8_SNORM
);
473 FORMAT_CASE(GL_RG8_SNORM
);
474 FORMAT_CASE(GL_RGBA8_SNORM
);
476 FORMAT_CASE(GL_RG16
);
477 FORMAT_CASE(GL_RGBA16
);
478 FORMAT_CASE(GL_R16_SNORM
);
479 FORMAT_CASE(GL_RG16_SNORM
);
480 FORMAT_CASE(GL_RGBA16_SNORM
);
481 FORMAT_CASE(GL_R16F
);
482 FORMAT_CASE(GL_RG16F
);
483 FORMAT_CASE(GL_RGBA16F
);
484 FORMAT_CASE(GL_R8UI
);
486 FORMAT_CASE(GL_RG8UI
);
487 FORMAT_CASE(GL_RG8I
);
488 FORMAT_CASE(GL_RGBA8UI
);
489 FORMAT_CASE(GL_RGBA8I
);
490 FORMAT_CASE(GL_R16UI
);
491 FORMAT_CASE(GL_R16I
);
492 FORMAT_CASE(GL_RG16UI
);
493 FORMAT_CASE(GL_RG16I
);
494 FORMAT_CASE(GL_RGBA16UI
);
495 FORMAT_CASE(GL_RGBA16I
);
496 FORMAT_CASE(GL_R11F_G11F_B10F
);
497 FORMAT_CASE(GL_RGB9_E5
);
498 FORMAT_CASE(GL_RGB10_A2
);
499 FORMAT_CASE(GL_RGB10_A2UI
);
500 default: /* Including the normal GL_NONE */
505 fprintf(fp
, "%s %s", glsl_get_type_name(var
->type
),
506 get_var_name(var
, state
));
508 if (var
->data
.mode
== nir_var_shader_in
||
509 var
->data
.mode
== nir_var_shader_out
||
510 var
->data
.mode
== nir_var_uniform
||
511 var
->data
.mode
== nir_var_mem_ubo
||
512 var
->data
.mode
== nir_var_mem_ssbo
) {
513 const char *loc
= NULL
;
516 switch (state
->shader
->info
.stage
) {
517 case MESA_SHADER_VERTEX
:
518 if (var
->data
.mode
== nir_var_shader_in
)
519 loc
= gl_vert_attrib_name(var
->data
.location
);
520 else if (var
->data
.mode
== nir_var_shader_out
)
521 loc
= gl_varying_slot_name(var
->data
.location
);
523 case MESA_SHADER_GEOMETRY
:
524 if ((var
->data
.mode
== nir_var_shader_in
) ||
525 (var
->data
.mode
== nir_var_shader_out
))
526 loc
= gl_varying_slot_name(var
->data
.location
);
528 case MESA_SHADER_FRAGMENT
:
529 if (var
->data
.mode
== nir_var_shader_in
)
530 loc
= gl_varying_slot_name(var
->data
.location
);
531 else if (var
->data
.mode
== nir_var_shader_out
)
532 loc
= gl_frag_result_name(var
->data
.location
);
534 case MESA_SHADER_TESS_CTRL
:
535 case MESA_SHADER_TESS_EVAL
:
536 case MESA_SHADER_COMPUTE
:
537 case MESA_SHADER_KERNEL
:
544 snprintf(buf
, sizeof(buf
), "%u", var
->data
.location
);
548 /* For shader I/O vars that have been split to components or packed,
549 * print the fractional location within the input/output.
551 unsigned int num_components
=
552 glsl_get_components(glsl_without_array(var
->type
));
553 const char *components
= NULL
;
554 char components_local
[6] = {'.' /* the rest is 0-filled */};
555 switch (var
->data
.mode
) {
556 case nir_var_shader_in
:
557 case nir_var_shader_out
:
558 if (num_components
< 4 && num_components
!= 0) {
559 const char *xyzw
= "xyzw";
560 for (int i
= 0; i
< num_components
; i
++)
561 components_local
[i
+ 1] = xyzw
[i
+ var
->data
.location_frac
];
563 components
= components_local
;
570 fprintf(fp
, " (%s%s, %u, %u)%s", loc
,
571 components
? components
: "",
572 var
->data
.driver_location
, var
->data
.binding
,
573 var
->data
.compact
? " compact" : "");
576 if (var
->constant_initializer
) {
577 fprintf(fp
, " = { ");
578 print_constant(var
->constant_initializer
, var
->type
, state
);
583 print_annotation(state
, var
);
587 print_deref_link(const nir_deref_instr
*instr
, bool whole_chain
, print_state
*state
)
589 FILE *fp
= state
->fp
;
591 if (instr
->deref_type
== nir_deref_type_var
) {
592 fprintf(fp
, "%s", get_var_name(instr
->var
, state
));
594 } else if (instr
->deref_type
== nir_deref_type_cast
) {
595 fprintf(fp
, "(%s *)", glsl_get_type_name(instr
->type
));
596 print_src(&instr
->parent
, state
);
600 assert(instr
->parent
.is_ssa
);
601 nir_deref_instr
*parent
=
602 nir_instr_as_deref(instr
->parent
.ssa
->parent_instr
);
604 /* Is the parent we're going to print a bare cast? */
605 const bool is_parent_cast
=
606 whole_chain
&& parent
->deref_type
== nir_deref_type_cast
;
608 /* If we're not printing the whole chain, the parent we print will be a SSA
609 * value that represents a pointer. The only deref type that naturally
610 * gives a pointer is a cast.
612 const bool is_parent_pointer
=
613 !whole_chain
|| parent
->deref_type
== nir_deref_type_cast
;
615 /* Struct derefs have a nice syntax that works on pointers, arrays derefs
618 const bool need_deref
=
619 is_parent_pointer
&& instr
->deref_type
!= nir_deref_type_struct
;
621 /* Cast need extra parens and so * dereferences */
622 if (is_parent_cast
|| need_deref
)
629 print_deref_link(parent
, whole_chain
, state
);
631 print_src(&instr
->parent
, state
);
634 if (is_parent_cast
|| need_deref
)
637 switch (instr
->deref_type
) {
638 case nir_deref_type_struct
:
639 fprintf(fp
, "%s%s", is_parent_pointer
? "->" : ".",
640 glsl_get_struct_elem_name(parent
->type
, instr
->strct
.index
));
643 case nir_deref_type_array
:
644 case nir_deref_type_ptr_as_array
: {
645 nir_const_value
*const_index
= nir_src_as_const_value(instr
->arr
.index
);
647 fprintf(fp
, "[%u]", const_index
->u32
[0]);
650 print_src(&instr
->arr
.index
, state
);
656 case nir_deref_type_array_wildcard
:
661 unreachable("Invalid deref instruction type");
666 print_deref_instr(nir_deref_instr
*instr
, print_state
*state
)
668 FILE *fp
= state
->fp
;
670 print_dest(&instr
->dest
, state
);
672 switch (instr
->deref_type
) {
673 case nir_deref_type_var
:
674 fprintf(fp
, " = deref_var ");
676 case nir_deref_type_array
:
677 case nir_deref_type_array_wildcard
:
678 fprintf(fp
, " = deref_array ");
680 case nir_deref_type_struct
:
681 fprintf(fp
, " = deref_struct ");
683 case nir_deref_type_cast
:
684 fprintf(fp
, " = deref_cast ");
686 case nir_deref_type_ptr_as_array
:
687 fprintf(fp
, " = deref_ptr_as_array ");
690 unreachable("Invalid deref instruction type");
693 /* Only casts naturally return a pointer type */
694 if (instr
->deref_type
!= nir_deref_type_cast
)
697 print_deref_link(instr
, false, state
);
699 fprintf(fp
, " (%s %s) ",
700 get_variable_mode_str(instr
->mode
, true),
701 glsl_get_type_name(instr
->type
));
703 if (instr
->deref_type
!= nir_deref_type_var
&&
704 instr
->deref_type
!= nir_deref_type_cast
) {
705 /* Print the entire chain as a comment */
707 print_deref_link(instr
, true, state
);
713 vulkan_descriptor_type_name(VkDescriptorType type
)
716 case VK_DESCRIPTOR_TYPE_SAMPLER
: return "sampler";
717 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER
: return "texture+sampler";
718 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE
: return "texture";
719 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE
: return "image";
720 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER
: return "texture-buffer";
721 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER
: return "image-buffer";
722 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER
: return "UBO";
723 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER
: return "SSBO";
724 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC
: return "UBO";
725 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC
: return "SSBO";
726 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT
: return "input-att";
727 case VK_DESCRIPTOR_TYPE_INLINE_UNIFORM_BLOCK_EXT
: return "inline-UBO";
728 default: return "unknown";
733 print_intrinsic_instr(nir_intrinsic_instr
*instr
, print_state
*state
)
735 const nir_intrinsic_info
*info
= &nir_intrinsic_infos
[instr
->intrinsic
];
736 unsigned num_srcs
= info
->num_srcs
;
737 FILE *fp
= state
->fp
;
739 if (info
->has_dest
) {
740 print_dest(&instr
->dest
, state
);
744 fprintf(fp
, "intrinsic %s (", info
->name
);
746 for (unsigned i
= 0; i
< num_srcs
; i
++) {
750 print_src(&instr
->src
[i
], state
);
755 for (unsigned i
= 0; i
< info
->num_indices
; i
++) {
759 fprintf(fp
, "%d", instr
->const_index
[i
]);
764 static const char *index_name
[NIR_INTRINSIC_NUM_INDEX_FLAGS
] = {
765 [NIR_INTRINSIC_BASE
] = "base",
766 [NIR_INTRINSIC_WRMASK
] = "wrmask",
767 [NIR_INTRINSIC_STREAM_ID
] = "stream-id",
768 [NIR_INTRINSIC_UCP_ID
] = "ucp-id",
769 [NIR_INTRINSIC_RANGE
] = "range",
770 [NIR_INTRINSIC_DESC_SET
] = "desc-set",
771 [NIR_INTRINSIC_BINDING
] = "binding",
772 [NIR_INTRINSIC_COMPONENT
] = "component",
773 [NIR_INTRINSIC_INTERP_MODE
] = "interp_mode",
774 [NIR_INTRINSIC_REDUCTION_OP
] = "reduction_op",
775 [NIR_INTRINSIC_CLUSTER_SIZE
] = "cluster_size",
776 [NIR_INTRINSIC_PARAM_IDX
] = "param_idx",
777 [NIR_INTRINSIC_IMAGE_DIM
] = "image_dim",
778 [NIR_INTRINSIC_IMAGE_ARRAY
] = "image_array",
779 [NIR_INTRINSIC_ACCESS
] = "access",
780 [NIR_INTRINSIC_FORMAT
] = "format",
781 [NIR_INTRINSIC_ALIGN_MUL
] = "align_mul",
782 [NIR_INTRINSIC_ALIGN_OFFSET
] = "align_offset",
783 [NIR_INTRINSIC_DESC_TYPE
] = "desc_type",
785 for (unsigned idx
= 1; idx
< NIR_INTRINSIC_NUM_INDEX_FLAGS
; idx
++) {
786 if (!info
->index_map
[idx
])
789 if (idx
== NIR_INTRINSIC_WRMASK
) {
790 /* special case wrmask to show it as a writemask.. */
791 unsigned wrmask
= nir_intrinsic_write_mask(instr
);
792 fprintf(fp
, " wrmask=");
793 for (unsigned i
= 0; i
< 4; i
++)
794 if ((wrmask
>> i
) & 1)
795 fprintf(fp
, "%c", "xyzw"[i
]);
796 } else if (idx
== NIR_INTRINSIC_REDUCTION_OP
) {
797 nir_op reduction_op
= nir_intrinsic_reduction_op(instr
);
798 fprintf(fp
, " reduction_op=%s", nir_op_infos
[reduction_op
].name
);
799 } else if (idx
== NIR_INTRINSIC_IMAGE_DIM
) {
800 static const char *dim_name
[] = {
801 [GLSL_SAMPLER_DIM_1D
] = "1D",
802 [GLSL_SAMPLER_DIM_2D
] = "2D",
803 [GLSL_SAMPLER_DIM_3D
] = "3D",
804 [GLSL_SAMPLER_DIM_CUBE
] = "Cube",
805 [GLSL_SAMPLER_DIM_RECT
] = "Rect",
806 [GLSL_SAMPLER_DIM_BUF
] = "Buf",
807 [GLSL_SAMPLER_DIM_MS
] = "2D-MSAA",
808 [GLSL_SAMPLER_DIM_SUBPASS
] = "Subpass",
809 [GLSL_SAMPLER_DIM_SUBPASS_MS
] = "Subpass-MSAA",
811 enum glsl_sampler_dim dim
= nir_intrinsic_image_dim(instr
);
812 assert(dim
< ARRAY_SIZE(dim_name
) && dim_name
[dim
]);
813 fprintf(fp
, " image_dim=%s", dim_name
[dim
]);
814 } else if (idx
== NIR_INTRINSIC_IMAGE_ARRAY
) {
815 bool array
= nir_intrinsic_image_array(instr
);
816 fprintf(fp
, " image_array=%s", array
? "true" : "false");
817 } else if (idx
== NIR_INTRINSIC_DESC_TYPE
) {
818 VkDescriptorType desc_type
= nir_intrinsic_desc_type(instr
);
819 fprintf(fp
, " desc_type=%s", vulkan_descriptor_type_name(desc_type
));
821 unsigned off
= info
->index_map
[idx
] - 1;
822 assert(index_name
[idx
]); /* forgot to update index_name table? */
823 fprintf(fp
, " %s=%d", index_name
[idx
], instr
->const_index
[off
]);
831 struct exec_list
*var_list
= NULL
;
833 switch (instr
->intrinsic
) {
834 case nir_intrinsic_load_uniform
:
835 var_list
= &state
->shader
->uniforms
;
837 case nir_intrinsic_load_input
:
838 case nir_intrinsic_load_interpolated_input
:
839 case nir_intrinsic_load_per_vertex_input
:
840 var_list
= &state
->shader
->inputs
;
842 case nir_intrinsic_load_output
:
843 case nir_intrinsic_store_output
:
844 case nir_intrinsic_store_per_vertex_output
:
845 var_list
= &state
->shader
->outputs
;
851 nir_foreach_variable(var
, var_list
) {
852 if ((var
->data
.driver_location
== nir_intrinsic_base(instr
)) &&
853 (instr
->intrinsic
== nir_intrinsic_load_uniform
||
854 (nir_intrinsic_component(instr
) >= var
->data
.location_frac
&&
855 nir_intrinsic_component(instr
) <
856 (var
->data
.location_frac
+ glsl_get_components(var
->type
)))) &&
858 fprintf(fp
, "\t/* %s */", var
->name
);
865 print_tex_instr(nir_tex_instr
*instr
, print_state
*state
)
867 FILE *fp
= state
->fp
;
869 print_dest(&instr
->dest
, state
);
889 case nir_texop_txf_ms
:
890 fprintf(fp
, "txf_ms ");
892 case nir_texop_txf_ms_mcs
:
893 fprintf(fp
, "txf_ms_mcs ");
904 case nir_texop_query_levels
:
905 fprintf(fp
, "query_levels ");
907 case nir_texop_texture_samples
:
908 fprintf(fp
, "texture_samples ");
910 case nir_texop_samples_identical
:
911 fprintf(fp
, "samples_identical ");
914 unreachable("Invalid texture operation");
918 bool has_texture_deref
= false, has_sampler_deref
= false;
919 for (unsigned i
= 0; i
< instr
->num_srcs
; i
++) {
920 print_src(&instr
->src
[i
].src
, state
);
924 switch(instr
->src
[i
].src_type
) {
925 case nir_tex_src_coord
:
926 fprintf(fp
, "(coord)");
928 case nir_tex_src_projector
:
929 fprintf(fp
, "(projector)");
931 case nir_tex_src_comparator
:
932 fprintf(fp
, "(comparator)");
934 case nir_tex_src_offset
:
935 fprintf(fp
, "(offset)");
937 case nir_tex_src_bias
:
938 fprintf(fp
, "(bias)");
940 case nir_tex_src_lod
:
941 fprintf(fp
, "(lod)");
943 case nir_tex_src_min_lod
:
944 fprintf(fp
, "(min_lod)");
946 case nir_tex_src_ms_index
:
947 fprintf(fp
, "(ms_index)");
949 case nir_tex_src_ms_mcs
:
950 fprintf(fp
, "(ms_mcs)");
952 case nir_tex_src_ddx
:
953 fprintf(fp
, "(ddx)");
955 case nir_tex_src_ddy
:
956 fprintf(fp
, "(ddy)");
958 case nir_tex_src_texture_deref
:
959 has_texture_deref
= true;
960 fprintf(fp
, "(texture_deref)");
962 case nir_tex_src_sampler_deref
:
963 has_sampler_deref
= true;
964 fprintf(fp
, "(sampler_deref)");
966 case nir_tex_src_texture_offset
:
967 fprintf(fp
, "(texture_offset)");
969 case nir_tex_src_sampler_offset
:
970 fprintf(fp
, "(sampler_offset)");
972 case nir_tex_src_texture_handle
:
973 fprintf(fp
, "(texture_handle)");
975 case nir_tex_src_sampler_handle
:
976 fprintf(fp
, "(sampler_handle)");
978 case nir_tex_src_plane
:
979 fprintf(fp
, "(plane)");
983 unreachable("Invalid texture source type");
990 if (instr
->op
== nir_texop_tg4
) {
991 fprintf(fp
, "%u (gather_component), ", instr
->component
);
994 if (nir_tex_instr_has_explicit_tg4_offsets(instr
)) {
995 fprintf(fp
, "{ (%i, %i)", instr
->tg4_offsets
[0][0], instr
->tg4_offsets
[0][1]);
996 for (unsigned i
= 1; i
< 4; ++i
)
997 fprintf(fp
, ", (%i, %i)", instr
->tg4_offsets
[i
][0],
998 instr
->tg4_offsets
[i
][1]);
999 fprintf(fp
, " } (offsets), ");
1002 if (!has_texture_deref
) {
1003 fprintf(fp
, "%u (texture), ", instr
->texture_index
);
1006 if (!has_sampler_deref
) {
1007 fprintf(fp
, "%u (sampler), ", instr
->sampler_index
);
1012 print_call_instr(nir_call_instr
*instr
, print_state
*state
)
1014 FILE *fp
= state
->fp
;
1016 fprintf(fp
, "call %s ", instr
->callee
->name
);
1018 for (unsigned i
= 0; i
< instr
->num_params
; i
++) {
1022 print_src(&instr
->params
[i
], state
);
1027 print_load_const_instr(nir_load_const_instr
*instr
, print_state
*state
)
1029 FILE *fp
= state
->fp
;
1031 print_ssa_def(&instr
->def
, state
);
1033 fprintf(fp
, " = load_const (");
1035 for (unsigned i
= 0; i
< instr
->def
.num_components
; i
++) {
1040 * we don't really know the type of the constant (if it will be used as a
1041 * float or an int), so just print the raw constant in hex for fidelity
1042 * and then print the float in a comment for readability.
1045 switch (instr
->def
.bit_size
) {
1047 fprintf(fp
, "0x%16" PRIx64
" /* %f */", instr
->value
.u64
[i
],
1048 instr
->value
.f64
[i
]);
1051 fprintf(fp
, "0x%08x /* %f */", instr
->value
.u32
[i
], instr
->value
.f32
[i
]);
1054 fprintf(fp
, "0x%04x /* %f */", instr
->value
.u16
[i
],
1055 _mesa_half_to_float(instr
->value
.u16
[i
]));
1058 fprintf(fp
, "0x%02x", instr
->value
.u8
[i
]);
1061 fprintf(fp
, "%s", instr
->value
.b
[i
] ? "true" : "false");
1070 print_jump_instr(nir_jump_instr
*instr
, print_state
*state
)
1072 FILE *fp
= state
->fp
;
1074 switch (instr
->type
) {
1075 case nir_jump_break
:
1076 fprintf(fp
, "break");
1079 case nir_jump_continue
:
1080 fprintf(fp
, "continue");
1083 case nir_jump_return
:
1084 fprintf(fp
, "return");
1090 print_ssa_undef_instr(nir_ssa_undef_instr
* instr
, print_state
*state
)
1092 FILE *fp
= state
->fp
;
1093 print_ssa_def(&instr
->def
, state
);
1094 fprintf(fp
, " = undefined");
1098 print_phi_instr(nir_phi_instr
*instr
, print_state
*state
)
1100 FILE *fp
= state
->fp
;
1101 print_dest(&instr
->dest
, state
);
1102 fprintf(fp
, " = phi ");
1103 nir_foreach_phi_src(src
, instr
) {
1104 if (&src
->node
!= exec_list_get_head(&instr
->srcs
))
1107 fprintf(fp
, "block_%u: ", src
->pred
->index
);
1108 print_src(&src
->src
, state
);
1113 print_parallel_copy_instr(nir_parallel_copy_instr
*instr
, print_state
*state
)
1115 FILE *fp
= state
->fp
;
1116 nir_foreach_parallel_copy_entry(entry
, instr
) {
1117 if (&entry
->node
!= exec_list_get_head(&instr
->entries
))
1120 print_dest(&entry
->dest
, state
);
1122 print_src(&entry
->src
, state
);
1127 print_instr(const nir_instr
*instr
, print_state
*state
, unsigned tabs
)
1129 FILE *fp
= state
->fp
;
1130 print_tabs(tabs
, fp
);
1132 switch (instr
->type
) {
1133 case nir_instr_type_alu
:
1134 print_alu_instr(nir_instr_as_alu(instr
), state
);
1137 case nir_instr_type_deref
:
1138 print_deref_instr(nir_instr_as_deref(instr
), state
);
1141 case nir_instr_type_call
:
1142 print_call_instr(nir_instr_as_call(instr
), state
);
1145 case nir_instr_type_intrinsic
:
1146 print_intrinsic_instr(nir_instr_as_intrinsic(instr
), state
);
1149 case nir_instr_type_tex
:
1150 print_tex_instr(nir_instr_as_tex(instr
), state
);
1153 case nir_instr_type_load_const
:
1154 print_load_const_instr(nir_instr_as_load_const(instr
), state
);
1157 case nir_instr_type_jump
:
1158 print_jump_instr(nir_instr_as_jump(instr
), state
);
1161 case nir_instr_type_ssa_undef
:
1162 print_ssa_undef_instr(nir_instr_as_ssa_undef(instr
), state
);
1165 case nir_instr_type_phi
:
1166 print_phi_instr(nir_instr_as_phi(instr
), state
);
1169 case nir_instr_type_parallel_copy
:
1170 print_parallel_copy_instr(nir_instr_as_parallel_copy(instr
), state
);
1174 unreachable("Invalid instruction type");
1180 compare_block_index(const void *p1
, const void *p2
)
1182 const nir_block
*block1
= *((const nir_block
**) p1
);
1183 const nir_block
*block2
= *((const nir_block
**) p2
);
1185 return (int) block1
->index
- (int) block2
->index
;
1188 static void print_cf_node(nir_cf_node
*node
, print_state
*state
,
1192 print_block(nir_block
*block
, print_state
*state
, unsigned tabs
)
1194 FILE *fp
= state
->fp
;
1196 print_tabs(tabs
, fp
);
1197 fprintf(fp
, "block block_%u:\n", block
->index
);
1199 /* sort the predecessors by index so we consistently print the same thing */
1202 malloc(block
->predecessors
->entries
* sizeof(nir_block
*));
1205 set_foreach(block
->predecessors
, entry
) {
1206 preds
[i
++] = (nir_block
*) entry
->key
;
1209 qsort(preds
, block
->predecessors
->entries
, sizeof(nir_block
*),
1210 compare_block_index
);
1212 print_tabs(tabs
, fp
);
1213 fprintf(fp
, "/* preds: ");
1214 for (unsigned i
= 0; i
< block
->predecessors
->entries
; i
++) {
1215 fprintf(fp
, "block_%u ", preds
[i
]->index
);
1217 fprintf(fp
, "*/\n");
1221 nir_foreach_instr(instr
, block
) {
1222 print_instr(instr
, state
, tabs
);
1224 print_annotation(state
, instr
);
1227 print_tabs(tabs
, fp
);
1228 fprintf(fp
, "/* succs: ");
1229 for (unsigned i
= 0; i
< 2; i
++)
1230 if (block
->successors
[i
]) {
1231 fprintf(fp
, "block_%u ", block
->successors
[i
]->index
);
1233 fprintf(fp
, "*/\n");
1237 print_if(nir_if
*if_stmt
, print_state
*state
, unsigned tabs
)
1239 FILE *fp
= state
->fp
;
1241 print_tabs(tabs
, fp
);
1243 print_src(&if_stmt
->condition
, state
);
1244 fprintf(fp
, " {\n");
1245 foreach_list_typed(nir_cf_node
, node
, node
, &if_stmt
->then_list
) {
1246 print_cf_node(node
, state
, tabs
+ 1);
1248 print_tabs(tabs
, fp
);
1249 fprintf(fp
, "} else {\n");
1250 foreach_list_typed(nir_cf_node
, node
, node
, &if_stmt
->else_list
) {
1251 print_cf_node(node
, state
, tabs
+ 1);
1253 print_tabs(tabs
, fp
);
1258 print_loop(nir_loop
*loop
, print_state
*state
, unsigned tabs
)
1260 FILE *fp
= state
->fp
;
1262 print_tabs(tabs
, fp
);
1263 fprintf(fp
, "loop {\n");
1264 foreach_list_typed(nir_cf_node
, node
, node
, &loop
->body
) {
1265 print_cf_node(node
, state
, tabs
+ 1);
1267 print_tabs(tabs
, fp
);
1272 print_cf_node(nir_cf_node
*node
, print_state
*state
, unsigned int tabs
)
1274 switch (node
->type
) {
1275 case nir_cf_node_block
:
1276 print_block(nir_cf_node_as_block(node
), state
, tabs
);
1279 case nir_cf_node_if
:
1280 print_if(nir_cf_node_as_if(node
), state
, tabs
);
1283 case nir_cf_node_loop
:
1284 print_loop(nir_cf_node_as_loop(node
), state
, tabs
);
1288 unreachable("Invalid CFG node type");
1293 print_function_impl(nir_function_impl
*impl
, print_state
*state
)
1295 FILE *fp
= state
->fp
;
1297 fprintf(fp
, "\nimpl %s ", impl
->function
->name
);
1301 nir_foreach_variable(var
, &impl
->locals
) {
1303 print_var_decl(var
, state
);
1306 foreach_list_typed(nir_register
, reg
, node
, &impl
->registers
) {
1308 print_register_decl(reg
, state
);
1311 nir_index_blocks(impl
);
1313 foreach_list_typed(nir_cf_node
, node
, node
, &impl
->body
) {
1314 print_cf_node(node
, state
, 1);
1317 fprintf(fp
, "\tblock block_%u:\n}\n\n", impl
->end_block
->index
);
1321 print_function(nir_function
*function
, print_state
*state
)
1323 FILE *fp
= state
->fp
;
1325 fprintf(fp
, "decl_function %s (%d params)", function
->name
,
1326 function
->num_params
);
1330 if (function
->impl
!= NULL
) {
1331 print_function_impl(function
->impl
, state
);
1337 init_print_state(print_state
*state
, nir_shader
*shader
, FILE *fp
)
1340 state
->shader
= shader
;
1341 state
->ht
= _mesa_pointer_hash_table_create(NULL
);
1342 state
->syms
= _mesa_set_create(NULL
, _mesa_key_hash_string
,
1343 _mesa_key_string_equal
);
1348 destroy_print_state(print_state
*state
)
1350 _mesa_hash_table_destroy(state
->ht
, NULL
);
1351 _mesa_set_destroy(state
->syms
, NULL
);
1355 nir_print_shader_annotated(nir_shader
*shader
, FILE *fp
,
1356 struct hash_table
*annotations
)
1359 init_print_state(&state
, shader
, fp
);
1361 state
.annotations
= annotations
;
1363 fprintf(fp
, "shader: %s\n", gl_shader_stage_name(shader
->info
.stage
));
1365 if (shader
->info
.name
)
1366 fprintf(fp
, "name: %s\n", shader
->info
.name
);
1368 if (shader
->info
.label
)
1369 fprintf(fp
, "label: %s\n", shader
->info
.label
);
1371 if (gl_shader_stage_is_compute(shader
->info
.stage
)) {
1372 fprintf(fp
, "local-size: %u, %u, %u%s\n",
1373 shader
->info
.cs
.local_size
[0],
1374 shader
->info
.cs
.local_size
[1],
1375 shader
->info
.cs
.local_size
[2],
1376 shader
->info
.cs
.local_size_variable
? " (variable)" : "");
1377 fprintf(fp
, "shared-size: %u\n", shader
->info
.cs
.shared_size
);
1380 fprintf(fp
, "inputs: %u\n", shader
->num_inputs
);
1381 fprintf(fp
, "outputs: %u\n", shader
->num_outputs
);
1382 fprintf(fp
, "uniforms: %u\n", shader
->num_uniforms
);
1383 fprintf(fp
, "shared: %u\n", shader
->num_shared
);
1385 nir_foreach_variable(var
, &shader
->uniforms
) {
1386 print_var_decl(var
, &state
);
1389 nir_foreach_variable(var
, &shader
->inputs
) {
1390 print_var_decl(var
, &state
);
1393 nir_foreach_variable(var
, &shader
->outputs
) {
1394 print_var_decl(var
, &state
);
1397 nir_foreach_variable(var
, &shader
->shared
) {
1398 print_var_decl(var
, &state
);
1401 nir_foreach_variable(var
, &shader
->globals
) {
1402 print_var_decl(var
, &state
);
1405 nir_foreach_variable(var
, &shader
->system_values
) {
1406 print_var_decl(var
, &state
);
1409 foreach_list_typed(nir_register
, reg
, node
, &shader
->registers
) {
1410 print_register_decl(reg
, &state
);
1413 foreach_list_typed(nir_function
, func
, node
, &shader
->functions
) {
1414 print_function(func
, &state
);
1417 destroy_print_state(&state
);
1421 nir_print_shader(nir_shader
*shader
, FILE *fp
)
1423 nir_print_shader_annotated(shader
, fp
, NULL
);
1428 nir_print_instr(const nir_instr
*instr
, FILE *fp
)
1430 print_state state
= {
1433 print_instr(instr
, &state
, 0);
1438 nir_print_deref(const nir_deref_instr
*deref
, FILE *fp
)
1440 print_state state
= {
1443 print_deref_link(deref
, true, &state
);