/* * Copyright © 2014 Intel Corporation * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the "Software"), * to deal in the Software without restriction, including without limitation * the rights to use, copy, modify, merge, publish, distribute, sublicense, * and/or sell copies of the Software, and to permit persons to whom the * Software is furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice (including the next * paragraph) shall be included in all copies or substantial portions of the * Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS * IN THE SOFTWARE. * * Authors: * Connor Abbott (cwabbott0@gmail.com) * */ #include "nir.h" #include "compiler/shader_enums.h" #include "util/half_float.h" #include "util/memstream.h" #include "util/mesa-sha1.h" #include "vulkan/vulkan_core.h" #include #include #include /* for PRIx64 macro */ static void print_tabs(unsigned num_tabs, FILE *fp) { for (unsigned i = 0; i < num_tabs; i++) fprintf(fp, "\t"); } typedef struct { FILE *fp; nir_shader *shader; /** map from nir_variable -> printable name */ struct hash_table *ht; /** set of names used so far for nir_variables */ struct set *syms; /* an index used to make new non-conflicting names */ unsigned index; /** * Optional table of annotations mapping nir object * (such as instr or var) to message to print. */ struct hash_table *annotations; } print_state; static void print_annotation(print_state *state, void *obj) { FILE *fp = state->fp; if (!state->annotations) return; struct hash_entry *entry = _mesa_hash_table_search(state->annotations, obj); if (!entry) return; const char *note = entry->data; _mesa_hash_table_remove(state->annotations, entry); fprintf(fp, "%s\n\n", note); } static void print_register(nir_register *reg, print_state *state) { FILE *fp = state->fp; fprintf(fp, "r%u", reg->index); } static const char *sizes[] = { "error", "vec1", "vec2", "vec3", "vec4", "vec5", "error", "error", "vec8", "error", "error", "error", "error", "error", "error", "error", "vec16"}; static void print_register_decl(nir_register *reg, print_state *state) { FILE *fp = state->fp; fprintf(fp, "decl_reg %s %u ", sizes[reg->num_components], reg->bit_size); print_register(reg, state); if (reg->num_array_elems != 0) fprintf(fp, "[%u]", reg->num_array_elems); fprintf(fp, "\n"); } static void print_ssa_def(nir_ssa_def *def, print_state *state) { FILE *fp = state->fp; const char *divergence = ""; if (state->shader->info.divergence_analysis_run) divergence = def->divergent ? "div " : "con "; fprintf(fp, "%s %2u %sssa_%u", sizes[def->num_components], def->bit_size, divergence, def->index); } static void print_const_from_load(nir_load_const_instr *instr, print_state *state) { FILE *fp = state->fp; /* * we don't really know the type of the constant (if it will be used as a * float or an int), so just print the raw constant in hex for fidelity * and then print in float again for readability. */ fprintf(fp, "("); for (unsigned i = 0; i < instr->def.num_components; i++) { if (i != 0) fprintf(fp, ", "); switch (instr->def.bit_size) { case 64: fprintf(fp, "0x%016" PRIx64, instr->value[i].u64); break; case 32: fprintf(fp, "0x%08x", instr->value[i].u32); break; case 16: fprintf(fp, "0x%04x", instr->value[i].u16); break; case 8: fprintf(fp, "0x%02x", instr->value[i].u8); break; case 1: fprintf(fp, "%s", instr->value[i].b ? "true" : "false"); break; } } if (instr->def.bit_size > 8) { if (instr->def.num_components > 1) fprintf(fp, ") = ("); else fprintf(fp, " = "); for (unsigned i = 0; i < instr->def.num_components; i++) { if (i != 0) fprintf(fp, ", "); switch (instr->def.bit_size) { case 64: fprintf(fp, "%f", instr->value[i].f64); break; case 32: fprintf(fp, "%f", instr->value[i].f32); break; case 16: fprintf(fp, "%f", _mesa_half_to_float(instr->value[i].u16)); break; default: unreachable("unhandled bit size"); } } } fprintf(fp, ")"); } static void print_load_const_instr(nir_load_const_instr *instr, print_state *state) { FILE *fp = state->fp; print_ssa_def(&instr->def, state); fprintf(fp, " = load_const "); print_const_from_load(instr, state); } static void print_ssa_use(nir_ssa_def *def, print_state *state) { FILE *fp = state->fp; fprintf(fp, "ssa_%u", def->index); nir_instr *instr = def->parent_instr; if (instr->type == nir_instr_type_load_const && NIR_DEBUG(PRINT_CONSTS)) { fprintf(fp, " /*"); print_const_from_load(nir_instr_as_load_const(instr), state); fprintf(fp, "*/"); } } static void print_src(const nir_src *src, print_state *state); static void print_reg_src(const nir_reg_src *src, print_state *state) { FILE *fp = state->fp; print_register(src->reg, state); if (src->reg->num_array_elems != 0) { fprintf(fp, "[%u", src->base_offset); if (src->indirect != NULL) { fprintf(fp, " + "); print_src(src->indirect, state); } fprintf(fp, "]"); } } static void print_reg_dest(nir_reg_dest *dest, print_state *state) { FILE *fp = state->fp; print_register(dest->reg, state); if (dest->reg->num_array_elems != 0) { fprintf(fp, "[%u", dest->base_offset); if (dest->indirect != NULL) { fprintf(fp, " + "); print_src(dest->indirect, state); } fprintf(fp, "]"); } } static void print_src(const nir_src *src, print_state *state) { if (src->is_ssa) print_ssa_use(src->ssa, state); else print_reg_src(&src->reg, state); } static void print_dest(nir_dest *dest, print_state *state) { if (dest->is_ssa) print_ssa_def(&dest->ssa, state); else print_reg_dest(&dest->reg, state); } static const char * comp_mask_string(unsigned num_components) { return (num_components > 4) ? "abcdefghijklmnop" : "xyzw"; } static void print_alu_src(nir_alu_instr *instr, unsigned src, print_state *state) { FILE *fp = state->fp; if (instr->src[src].negate) fprintf(fp, "-"); if (instr->src[src].abs) fprintf(fp, "abs("); print_src(&instr->src[src].src, state); bool print_swizzle = false; nir_component_mask_t used_channels = 0; for (unsigned i = 0; i < NIR_MAX_VEC_COMPONENTS; i++) { if (!nir_alu_instr_channel_used(instr, src, i)) continue; used_channels++; if (instr->src[src].swizzle[i] != i) { print_swizzle = true; break; } } unsigned live_channels = nir_src_num_components(instr->src[src].src); if (print_swizzle || used_channels != live_channels) { fprintf(fp, "."); for (unsigned i = 0; i < NIR_MAX_VEC_COMPONENTS; i++) { if (!nir_alu_instr_channel_used(instr, src, i)) continue; fprintf(fp, "%c", comp_mask_string(live_channels)[instr->src[src].swizzle[i]]); } } if (instr->src[src].abs) fprintf(fp, ")"); } static void print_alu_dest(nir_alu_dest *dest, print_state *state) { FILE *fp = state->fp; /* we're going to print the saturate modifier later, after the opcode */ print_dest(&dest->dest, state); if (!dest->dest.is_ssa && dest->write_mask != (1 << dest->dest.reg.reg->num_components) - 1) { unsigned live_channels = dest->dest.reg.reg->num_components; fprintf(fp, "."); for (unsigned i = 0; i < NIR_MAX_VEC_COMPONENTS; i++) if ((dest->write_mask >> i) & 1) fprintf(fp, "%c", comp_mask_string(live_channels)[i]); } } static void print_alu_instr(nir_alu_instr *instr, print_state *state) { FILE *fp = state->fp; print_alu_dest(&instr->dest, state); fprintf(fp, " = %s", nir_op_infos[instr->op].name); if (instr->exact) fprintf(fp, "!"); if (instr->dest.saturate) fprintf(fp, ".sat"); if (instr->no_signed_wrap) fprintf(fp, ".nsw"); if (instr->no_unsigned_wrap) fprintf(fp, ".nuw"); fprintf(fp, " "); for (unsigned i = 0; i < nir_op_infos[instr->op].num_inputs; i++) { if (i != 0) fprintf(fp, ", "); print_alu_src(instr, i, state); } } static const char * get_var_name(nir_variable *var, print_state *state) { if (state->ht == NULL) return var->name ? var->name : "unnamed"; assert(state->syms); struct hash_entry *entry = _mesa_hash_table_search(state->ht, var); if (entry) return entry->data; char *name; if (var->name == NULL) { name = ralloc_asprintf(state->syms, "@%u", state->index++); } else { struct set_entry *set_entry = _mesa_set_search(state->syms, var->name); if (set_entry != NULL) { /* we have a collision with another name, append an @ + a unique * index */ name = ralloc_asprintf(state->syms, "%s@%u", var->name, state->index++); } else { /* Mark this one as seen */ _mesa_set_add(state->syms, var->name); name = var->name; } } _mesa_hash_table_insert(state->ht, var, name); return name; } static const char * get_constant_sampler_addressing_mode(enum cl_sampler_addressing_mode mode) { switch (mode) { case SAMPLER_ADDRESSING_MODE_NONE: return "none"; case SAMPLER_ADDRESSING_MODE_CLAMP_TO_EDGE: return "clamp_to_edge"; case SAMPLER_ADDRESSING_MODE_CLAMP: return "clamp"; case SAMPLER_ADDRESSING_MODE_REPEAT: return "repeat"; case SAMPLER_ADDRESSING_MODE_REPEAT_MIRRORED: return "repeat_mirrored"; default: unreachable("Invalid addressing mode"); } } static const char * get_constant_sampler_filter_mode(enum cl_sampler_filter_mode mode) { switch (mode) { case SAMPLER_FILTER_MODE_NEAREST: return "nearest"; case SAMPLER_FILTER_MODE_LINEAR: return "linear"; default: unreachable("Invalid filter mode"); } } static void print_constant(nir_constant *c, const struct glsl_type *type, print_state *state) { FILE *fp = state->fp; const unsigned rows = glsl_get_vector_elements(type); const unsigned cols = glsl_get_matrix_columns(type); unsigned i; switch (glsl_get_base_type(type)) { case GLSL_TYPE_BOOL: /* Only float base types can be matrices. */ assert(cols == 1); for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "%s", c->values[i].b ? "true" : "false"); } break; case GLSL_TYPE_UINT8: case GLSL_TYPE_INT8: /* Only float base types can be matrices. */ assert(cols == 1); for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "0x%02x", c->values[i].u8); } break; case GLSL_TYPE_UINT16: case GLSL_TYPE_INT16: /* Only float base types can be matrices. */ assert(cols == 1); for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "0x%04x", c->values[i].u16); } break; case GLSL_TYPE_UINT: case GLSL_TYPE_INT: /* Only float base types can be matrices. */ assert(cols == 1); for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "0x%08x", c->values[i].u32); } break; case GLSL_TYPE_FLOAT16: case GLSL_TYPE_FLOAT: case GLSL_TYPE_DOUBLE: if (cols > 1) { for (i = 0; i < cols; i++) { if (i > 0) fprintf(fp, ", "); print_constant(c->elements[i], glsl_get_column_type(type), state); } } else { switch (glsl_get_base_type(type)) { case GLSL_TYPE_FLOAT16: for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "%f", _mesa_half_to_float(c->values[i].u16)); } break; case GLSL_TYPE_FLOAT: for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "%f", c->values[i].f32); } break; case GLSL_TYPE_DOUBLE: for (i = 0; i < rows; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "%f", c->values[i].f64); } break; default: unreachable("Cannot get here from the first level switch"); } } break; case GLSL_TYPE_UINT64: case GLSL_TYPE_INT64: /* Only float base types can be matrices. */ assert(cols == 1); for (i = 0; i < cols; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "0x%08" PRIx64, c->values[i].u64); } break; case GLSL_TYPE_STRUCT: case GLSL_TYPE_INTERFACE: for (i = 0; i < c->num_elements; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "{ "); print_constant(c->elements[i], glsl_get_struct_field(type, i), state); fprintf(fp, " }"); } break; case GLSL_TYPE_ARRAY: for (i = 0; i < c->num_elements; i++) { if (i > 0) fprintf(fp, ", "); fprintf(fp, "{ "); print_constant(c->elements[i], glsl_get_array_element(type), state); fprintf(fp, " }"); } break; default: unreachable("not reached"); } } static const char * get_variable_mode_str(nir_variable_mode mode, bool want_local_global_mode) { switch (mode) { case nir_var_shader_in: return "shader_in"; case nir_var_shader_out: return "shader_out"; case nir_var_uniform: return "uniform"; case nir_var_mem_ubo: return "ubo"; case nir_var_system_value: return "system"; case nir_var_mem_ssbo: return "ssbo"; case nir_var_mem_shared: return "shared"; case nir_var_mem_global: return "global"; case nir_var_mem_push_const: return "push_const"; case nir_var_mem_constant: return "constant"; case nir_var_image: return "image"; case nir_var_shader_temp: return want_local_global_mode ? "shader_temp" : ""; case nir_var_function_temp: return want_local_global_mode ? "function_temp" : ""; case nir_var_shader_call_data: return "shader_call_data"; case nir_var_ray_hit_attrib: return "ray_hit_attrib"; case nir_var_mem_task_payload: return "task_payload"; default: if (mode && (mode & nir_var_mem_generic) == mode) return "generic"; return ""; } } static void print_var_decl(nir_variable *var, print_state *state) { FILE *fp = state->fp; fprintf(fp, "decl_var "); const char *const bindless = (var->data.bindless) ? "bindless " : ""; const char *const cent = (var->data.centroid) ? "centroid " : ""; const char *const samp = (var->data.sample) ? "sample " : ""; const char *const patch = (var->data.patch) ? "patch " : ""; const char *const inv = (var->data.invariant) ? "invariant " : ""; const char *const per_view = (var->data.per_view) ? "per_view " : ""; const char *const per_primitive = (var->data.per_primitive) ? "per_primitive " : ""; const char *const ray_query = (var->data.ray_query) ? "ray_query " : ""; fprintf(fp, "%s%s%s%s%s%s%s%s%s %s ", bindless, cent, samp, patch, inv, per_view, per_primitive, ray_query, get_variable_mode_str(var->data.mode, false), glsl_interp_mode_name(var->data.interpolation)); enum gl_access_qualifier access = var->data.access; const char *const coher = (access & ACCESS_COHERENT) ? "coherent " : ""; const char *const volat = (access & ACCESS_VOLATILE) ? "volatile " : ""; const char *const restr = (access & ACCESS_RESTRICT) ? "restrict " : ""; const char *const ronly = (access & ACCESS_NON_WRITEABLE) ? "readonly " : ""; const char *const wonly = (access & ACCESS_NON_READABLE) ? "writeonly " : ""; const char *const reorder = (access & ACCESS_CAN_REORDER) ? "reorderable " : ""; const char *const stream_cache_policy = (access & ACCESS_STREAM_CACHE_POLICY) ? "stream-cache-policy " : ""; const char *const include_helpers = (access & ACCESS_INCLUDE_HELPERS) ? "include-helpers " : ""; fprintf(fp, "%s%s%s%s%s%s%s%s", coher, volat, restr, ronly, wonly, reorder, stream_cache_policy, include_helpers); if (glsl_get_base_type(glsl_without_array(var->type)) == GLSL_TYPE_IMAGE) { fprintf(fp, "%s ", util_format_short_name(var->data.image.format)); } if (var->data.precision) { const char *precisions[] = { "", "highp", "mediump", "lowp", }; fprintf(fp, "%s ", precisions[var->data.precision]); } fprintf(fp, "%s %s", glsl_get_type_name(var->type), get_var_name(var, state)); if (var->data.mode & (nir_var_shader_in | nir_var_shader_out | nir_var_uniform | nir_var_mem_ubo | nir_var_mem_ssbo | nir_var_image)) { const char *loc = NULL; char buf[4]; switch (state->shader->info.stage) { case MESA_SHADER_VERTEX: if (var->data.mode == nir_var_shader_in) loc = gl_vert_attrib_name(var->data.location); else if (var->data.mode == nir_var_shader_out) loc = gl_varying_slot_name_for_stage(var->data.location, state->shader->info.stage); break; case MESA_SHADER_TASK: case MESA_SHADER_MESH: case MESA_SHADER_GEOMETRY: if ((var->data.mode == nir_var_shader_in) || (var->data.mode == nir_var_shader_out)) { loc = gl_varying_slot_name_for_stage(var->data.location, state->shader->info.stage); } break; case MESA_SHADER_FRAGMENT: if (var->data.mode == nir_var_shader_in) { loc = gl_varying_slot_name_for_stage(var->data.location, state->shader->info.stage); } else if (var->data.mode == nir_var_shader_out) { loc = gl_frag_result_name(var->data.location); } break; case MESA_SHADER_TESS_CTRL: case MESA_SHADER_TESS_EVAL: case MESA_SHADER_COMPUTE: case MESA_SHADER_KERNEL: default: /* TODO */ break; } if (!loc) { if (var->data.location == ~0) { loc = "~0"; } else { snprintf(buf, sizeof(buf), "%u", var->data.location); loc = buf; } } /* For shader I/O vars that have been split to components or packed, * print the fractional location within the input/output. */ unsigned int num_components = glsl_get_components(glsl_without_array(var->type)); const char *components = NULL; char components_local[18] = {'.' /* the rest is 0-filled */}; switch (var->data.mode) { case nir_var_shader_in: case nir_var_shader_out: if (num_components < 16 && num_components != 0) { const char *xyzw = comp_mask_string(num_components); for (int i = 0; i < num_components; i++) components_local[i + 1] = xyzw[i + var->data.location_frac]; components = components_local; } break; default: break; } fprintf(fp, " (%s%s, %u, %u)%s", loc, components ? components : "", var->data.driver_location, var->data.binding, var->data.compact ? " compact" : ""); } if (var->constant_initializer) { fprintf(fp, " = { "); print_constant(var->constant_initializer, var->type, state); fprintf(fp, " }"); } if (glsl_type_is_sampler(var->type) && var->data.sampler.is_inline_sampler) { fprintf(fp, " = { %s, %s, %s }", get_constant_sampler_addressing_mode(var->data.sampler.addressing_mode), var->data.sampler.normalized_coordinates ? "true" : "false", get_constant_sampler_filter_mode(var->data.sampler.filter_mode)); } if (var->pointer_initializer) fprintf(fp, " = &%s", get_var_name(var->pointer_initializer, state)); fprintf(fp, "\n"); print_annotation(state, var); } static void print_deref_link(const nir_deref_instr *instr, bool whole_chain, print_state *state) { FILE *fp = state->fp; if (instr->deref_type == nir_deref_type_var) { fprintf(fp, "%s", get_var_name(instr->var, state)); return; } else if (instr->deref_type == nir_deref_type_cast) { fprintf(fp, "(%s *)", glsl_get_type_name(instr->type)); print_src(&instr->parent, state); return; } assert(instr->parent.is_ssa); nir_deref_instr *parent = nir_instr_as_deref(instr->parent.ssa->parent_instr); /* Is the parent we're going to print a bare cast? */ const bool is_parent_cast = whole_chain && parent->deref_type == nir_deref_type_cast; /* If we're not printing the whole chain, the parent we print will be a SSA * value that represents a pointer. The only deref type that naturally * gives a pointer is a cast. */ const bool is_parent_pointer = !whole_chain || parent->deref_type == nir_deref_type_cast; /* Struct derefs have a nice syntax that works on pointers, arrays derefs * do not. */ const bool need_deref = is_parent_pointer && instr->deref_type != nir_deref_type_struct; /* Cast need extra parens and so * dereferences */ if (is_parent_cast || need_deref) fprintf(fp, "("); if (need_deref) fprintf(fp, "*"); if (whole_chain) { print_deref_link(parent, whole_chain, state); } else { print_src(&instr->parent, state); } if (is_parent_cast || need_deref) fprintf(fp, ")"); switch (instr->deref_type) { case nir_deref_type_struct: fprintf(fp, "%s%s", is_parent_pointer ? "->" : ".", glsl_get_struct_elem_name(parent->type, instr->strct.index)); break; case nir_deref_type_array: case nir_deref_type_ptr_as_array: { if (nir_src_is_const(instr->arr.index)) { fprintf(fp, "[%"PRId64"]", nir_src_as_int(instr->arr.index)); } else { fprintf(fp, "["); print_src(&instr->arr.index, state); fprintf(fp, "]"); } break; } case nir_deref_type_array_wildcard: fprintf(fp, "[*]"); break; default: unreachable("Invalid deref instruction type"); } } static void print_deref_instr(nir_deref_instr *instr, print_state *state) { FILE *fp = state->fp; print_dest(&instr->dest, state); switch (instr->deref_type) { case nir_deref_type_var: fprintf(fp, " = deref_var "); break; case nir_deref_type_array: case nir_deref_type_array_wildcard: fprintf(fp, " = deref_array "); break; case nir_deref_type_struct: fprintf(fp, " = deref_struct "); break; case nir_deref_type_cast: fprintf(fp, " = deref_cast "); break; case nir_deref_type_ptr_as_array: fprintf(fp, " = deref_ptr_as_array "); break; default: unreachable("Invalid deref instruction type"); } /* Only casts naturally return a pointer type */ if (instr->deref_type != nir_deref_type_cast) fprintf(fp, "&"); print_deref_link(instr, false, state); fprintf(fp, " ("); unsigned modes = instr->modes; while (modes) { int m = u_bit_scan(&modes); fprintf(fp, "%s%s", get_variable_mode_str(1 << m, true), modes ? "|" : ""); } fprintf(fp, " %s) ", glsl_get_type_name(instr->type)); if (instr->deref_type != nir_deref_type_var && instr->deref_type != nir_deref_type_cast) { /* Print the entire chain as a comment */ fprintf(fp, "/* &"); print_deref_link(instr, true, state); fprintf(fp, " */"); } if (instr->deref_type == nir_deref_type_cast) { fprintf(fp, " /* ptr_stride=%u, align_mul=%u, align_offset=%u */", instr->cast.ptr_stride, instr->cast.align_mul, instr->cast.align_offset); } } static const char * vulkan_descriptor_type_name(VkDescriptorType type) { switch (type) { case VK_DESCRIPTOR_TYPE_SAMPLER: return "sampler"; case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER: return "texture+sampler"; case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE: return "texture"; case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE: return "image"; case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER: return "texture-buffer"; case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER: return "image-buffer"; case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER: return "UBO"; case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER: return "SSBO"; case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC: return "UBO"; case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC: return "SSBO"; case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT: return "input-att"; case VK_DESCRIPTOR_TYPE_INLINE_UNIFORM_BLOCK: return "inline-UBO"; case VK_DESCRIPTOR_TYPE_ACCELERATION_STRUCTURE_KHR: return "accel-struct"; default: return "unknown"; } } static void print_alu_type(nir_alu_type type, print_state *state) { FILE *fp = state->fp; unsigned size = nir_alu_type_get_type_size(type); const char *name; switch (nir_alu_type_get_base_type(type)) { case nir_type_int: name = "int"; break; case nir_type_uint: name = "uint"; break; case nir_type_bool: name = "bool"; break; case nir_type_float: name = "float"; break; default: name = "invalid"; } if (size) fprintf(fp, "%s%u", name, size); else fprintf(fp, "%s", name); } static void print_intrinsic_instr(nir_intrinsic_instr *instr, print_state *state) { const nir_intrinsic_info *info = &nir_intrinsic_infos[instr->intrinsic]; unsigned num_srcs = info->num_srcs; FILE *fp = state->fp; if (info->has_dest) { print_dest(&instr->dest, state); fprintf(fp, " = "); } fprintf(fp, "intrinsic %s (", info->name); for (unsigned i = 0; i < num_srcs; i++) { if (i != 0) fprintf(fp, ", "); print_src(&instr->src[i], state); } fprintf(fp, ") ("); for (unsigned i = 0; i < info->num_indices; i++) { unsigned idx = info->indices[i]; bool print_raw = true; if (i != 0) fprintf(fp, ", "); switch (idx) { case NIR_INTRINSIC_WRITE_MASK: { /* special case wrmask to show it as a writemask.. */ unsigned wrmask = nir_intrinsic_write_mask(instr); fprintf(fp, "wrmask="); for (unsigned i = 0; i < instr->num_components; i++) if ((wrmask >> i) & 1) fprintf(fp, "%c", comp_mask_string(instr->num_components)[i]); break; } case NIR_INTRINSIC_REDUCTION_OP: { nir_op reduction_op = nir_intrinsic_reduction_op(instr); fprintf(fp, "reduction_op=%s", nir_op_infos[reduction_op].name); break; } case NIR_INTRINSIC_IMAGE_DIM: { static const char *dim_name[] = { [GLSL_SAMPLER_DIM_1D] = "1D", [GLSL_SAMPLER_DIM_2D] = "2D", [GLSL_SAMPLER_DIM_3D] = "3D", [GLSL_SAMPLER_DIM_CUBE] = "Cube", [GLSL_SAMPLER_DIM_RECT] = "Rect", [GLSL_SAMPLER_DIM_BUF] = "Buf", [GLSL_SAMPLER_DIM_MS] = "2D-MSAA", [GLSL_SAMPLER_DIM_SUBPASS] = "Subpass", [GLSL_SAMPLER_DIM_SUBPASS_MS] = "Subpass-MSAA", }; enum glsl_sampler_dim dim = nir_intrinsic_image_dim(instr); assert(dim < ARRAY_SIZE(dim_name) && dim_name[dim]); fprintf(fp, "image_dim=%s", dim_name[dim]); break; } case NIR_INTRINSIC_IMAGE_ARRAY: { bool array = nir_intrinsic_image_array(instr); fprintf(fp, "image_array=%s", array ? "true" : "false"); break; } case NIR_INTRINSIC_FORMAT: { enum pipe_format format = nir_intrinsic_format(instr); fprintf(fp, "format=%s", util_format_short_name(format)); break; } case NIR_INTRINSIC_DESC_TYPE: { VkDescriptorType desc_type = nir_intrinsic_desc_type(instr); fprintf(fp, "desc_type=%s", vulkan_descriptor_type_name(desc_type)); break; } case NIR_INTRINSIC_SRC_TYPE: { fprintf(fp, "src_type="); print_alu_type(nir_intrinsic_src_type(instr), state); break; } case NIR_INTRINSIC_DEST_TYPE: { fprintf(fp, "dest_type="); print_alu_type(nir_intrinsic_dest_type(instr), state); break; } case NIR_INTRINSIC_SWIZZLE_MASK: { fprintf(fp, "swizzle_mask="); unsigned mask = nir_intrinsic_swizzle_mask(instr); if (instr->intrinsic == nir_intrinsic_quad_swizzle_amd) { for (unsigned i = 0; i < 4; i++) fprintf(fp, "%d", (mask >> (i * 2) & 3)); } else if (instr->intrinsic == nir_intrinsic_masked_swizzle_amd) { fprintf(fp, "((id & %d) | %d) ^ %d", mask & 0x1F, (mask >> 5) & 0x1F, (mask >> 10) & 0x1F); } else { fprintf(fp, "%d", mask); } break; } case NIR_INTRINSIC_MEMORY_SEMANTICS: { nir_memory_semantics semantics = nir_intrinsic_memory_semantics(instr); fprintf(fp, "mem_semantics="); switch (semantics & (NIR_MEMORY_ACQUIRE | NIR_MEMORY_RELEASE)) { case 0: fprintf(fp, "NONE"); break; case NIR_MEMORY_ACQUIRE: fprintf(fp, "ACQ"); break; case NIR_MEMORY_RELEASE: fprintf(fp, "REL"); break; default: fprintf(fp, "ACQ|REL"); break; } if (semantics & (NIR_MEMORY_MAKE_AVAILABLE)) fprintf(fp, "|AVAILABLE"); if (semantics & (NIR_MEMORY_MAKE_VISIBLE)) fprintf(fp, "|VISIBLE"); break; } case NIR_INTRINSIC_MEMORY_MODES: { fprintf(fp, "mem_modes="); unsigned int modes = nir_intrinsic_memory_modes(instr); while (modes) { nir_variable_mode m = u_bit_scan(&modes); fprintf(fp, "%s%s", get_variable_mode_str(1 << m, true), modes ? "|" : ""); } break; } case NIR_INTRINSIC_EXECUTION_SCOPE: case NIR_INTRINSIC_MEMORY_SCOPE: { fprintf(fp, "%s=", nir_intrinsic_index_names[idx]); nir_scope scope = idx == NIR_INTRINSIC_MEMORY_SCOPE ? nir_intrinsic_memory_scope(instr) : nir_intrinsic_execution_scope(instr); switch (scope) { case NIR_SCOPE_NONE: fprintf(fp, "NONE"); break; case NIR_SCOPE_DEVICE: fprintf(fp, "DEVICE"); break; case NIR_SCOPE_QUEUE_FAMILY: fprintf(fp, "QUEUE_FAMILY"); break; case NIR_SCOPE_WORKGROUP: fprintf(fp, "WORKGROUP"); break; case NIR_SCOPE_SHADER_CALL: fprintf(fp, "SHADER_CALL"); break; case NIR_SCOPE_SUBGROUP: fprintf(fp, "SUBGROUP"); break; case NIR_SCOPE_INVOCATION: fprintf(fp, "INVOCATION"); break; } break; } case NIR_INTRINSIC_IO_SEMANTICS: { struct nir_io_semantics io = nir_intrinsic_io_semantics(instr); fprintf(fp, "io location=%u slots=%u", io.location, io.num_slots); if (io.dual_source_blend_index) fprintf(fp, " dualsrc"); if (io.fb_fetch_output) fprintf(fp, " fbfetch"); if (io.per_view) fprintf(fp, " perview"); if (io.medium_precision) fprintf(fp, " mediump"); if (io.high_16bits) fprintf(fp, " high_16bits"); if (io.no_varying) fprintf(fp, " no_varying"); if (io.no_sysval_output) fprintf(fp, " no_sysval_output"); if (state->shader && state->shader->info.stage == MESA_SHADER_GEOMETRY && (instr->intrinsic == nir_intrinsic_store_output || instr->intrinsic == nir_intrinsic_store_per_primitive_output || instr->intrinsic == nir_intrinsic_store_per_vertex_output)) { unsigned gs_streams = io.gs_streams; fprintf(fp, " gs_streams("); for (unsigned i = 0; i < 4; i++) { fprintf(fp, "%s%c=%u", i ? " " : "", "xyzw"[i], (gs_streams >> (i * 2)) & 0x3); } fprintf(fp, ")"); } break; } case NIR_INTRINSIC_IO_XFB: case NIR_INTRINSIC_IO_XFB2: { /* This prints both IO_XFB and IO_XFB2. */ fprintf(fp, "xfb%s(", idx == NIR_INTRINSIC_IO_XFB ? "" : "2"); bool first = true; for (unsigned i = 0; i < 2; i++) { unsigned start_comp = (idx == NIR_INTRINSIC_IO_XFB ? 0 : 2) + i; nir_io_xfb xfb = start_comp < 2 ? nir_intrinsic_io_xfb(instr) : nir_intrinsic_io_xfb2(instr); if (!xfb.out[i].num_components) continue; if (!first) fprintf(fp, ", "); first = false; if (xfb.out[i].num_components > 1) { fprintf(fp, "components=%u..%u", start_comp, start_comp + xfb.out[i].num_components - 1); } else { fprintf(fp, "component=%u", start_comp); } fprintf(fp, " buffer=%u offset=%u", xfb.out[i].buffer, (uint32_t)xfb.out[i].offset * 4); } fprintf(fp, ")"); break; } case NIR_INTRINSIC_ROUNDING_MODE: { fprintf(fp, "rounding_mode="); switch (nir_intrinsic_rounding_mode(instr)) { case nir_rounding_mode_undef: fprintf(fp, "undef"); break; case nir_rounding_mode_rtne: fprintf(fp, "rtne"); break; case nir_rounding_mode_ru: fprintf(fp, "ru"); break; case nir_rounding_mode_rd: fprintf(fp, "rd"); break; case nir_rounding_mode_rtz: fprintf(fp, "rtz"); break; default: fprintf(fp, "unkown"); break; } break; } default: { unsigned off = info->index_map[idx] - 1; fprintf(fp, "%s=%d", nir_intrinsic_index_names[idx], instr->const_index[off]); print_raw = false; break; } } if (print_raw) fprintf(fp, " /*%d*/", instr->const_index[i]); } fprintf(fp, ")"); if (!state->shader) return; nir_variable_mode var_mode; switch (instr->intrinsic) { case nir_intrinsic_load_uniform: var_mode = nir_var_uniform; break; case nir_intrinsic_load_input: case nir_intrinsic_load_interpolated_input: case nir_intrinsic_load_per_vertex_input: var_mode = nir_var_shader_in; break; case nir_intrinsic_load_output: case nir_intrinsic_store_output: case nir_intrinsic_store_per_vertex_output: var_mode = nir_var_shader_out; break; default: return; } nir_foreach_variable_with_modes(var, state->shader, var_mode) { if ((var->data.driver_location == nir_intrinsic_base(instr)) && (instr->intrinsic == nir_intrinsic_load_uniform || (nir_intrinsic_component(instr) >= var->data.location_frac && nir_intrinsic_component(instr) < (var->data.location_frac + glsl_get_components(var->type)))) && var->name) { fprintf(fp, "\t/* %s */", var->name); break; } } } static void print_tex_instr(nir_tex_instr *instr, print_state *state) { FILE *fp = state->fp; print_dest(&instr->dest, state); fprintf(fp, " = ("); print_alu_type(instr->dest_type, state); fprintf(fp, ")"); switch (instr->op) { case nir_texop_tex: fprintf(fp, "tex "); break; case nir_texop_txb: fprintf(fp, "txb "); break; case nir_texop_txl: fprintf(fp, "txl "); break; case nir_texop_txd: fprintf(fp, "txd "); break; case nir_texop_txf: fprintf(fp, "txf "); break; case nir_texop_txf_ms: fprintf(fp, "txf_ms "); break; case nir_texop_txf_ms_fb: fprintf(fp, "txf_ms_fb "); break; case nir_texop_txf_ms_mcs_intel: fprintf(fp, "txf_ms_mcs_intel "); break; case nir_texop_txs: fprintf(fp, "txs "); break; case nir_texop_lod: fprintf(fp, "lod "); break; case nir_texop_tg4: fprintf(fp, "tg4 "); break; case nir_texop_query_levels: fprintf(fp, "query_levels "); break; case nir_texop_texture_samples: fprintf(fp, "texture_samples "); break; case nir_texop_samples_identical: fprintf(fp, "samples_identical "); break; case nir_texop_tex_prefetch: fprintf(fp, "tex (pre-dispatchable) "); break; case nir_texop_fragment_fetch_amd: fprintf(fp, "fragment_fetch_amd "); break; case nir_texop_fragment_mask_fetch_amd: fprintf(fp, "fragment_mask_fetch_amd "); break; default: unreachable("Invalid texture operation"); break; } bool has_texture_deref = false, has_sampler_deref = false; for (unsigned i = 0; i < instr->num_srcs; i++) { if (i > 0) { fprintf(fp, ", "); } print_src(&instr->src[i].src, state); fprintf(fp, " "); switch(instr->src[i].src_type) { case nir_tex_src_backend1: fprintf(fp, "(backend1)"); break; case nir_tex_src_backend2: fprintf(fp, "(backend2)"); break; case nir_tex_src_coord: fprintf(fp, "(coord)"); break; case nir_tex_src_projector: fprintf(fp, "(projector)"); break; case nir_tex_src_comparator: fprintf(fp, "(comparator)"); break; case nir_tex_src_offset: fprintf(fp, "(offset)"); break; case nir_tex_src_bias: fprintf(fp, "(bias)"); break; case nir_tex_src_lod: fprintf(fp, "(lod)"); break; case nir_tex_src_min_lod: fprintf(fp, "(min_lod)"); break; case nir_tex_src_ms_index: fprintf(fp, "(ms_index)"); break; case nir_tex_src_ms_mcs_intel: fprintf(fp, "(ms_mcs_intel)"); break; case nir_tex_src_ddx: fprintf(fp, "(ddx)"); break; case nir_tex_src_ddy: fprintf(fp, "(ddy)"); break; case nir_tex_src_texture_deref: has_texture_deref = true; fprintf(fp, "(texture_deref)"); break; case nir_tex_src_sampler_deref: has_sampler_deref = true; fprintf(fp, "(sampler_deref)"); break; case nir_tex_src_texture_offset: fprintf(fp, "(texture_offset)"); break; case nir_tex_src_sampler_offset: fprintf(fp, "(sampler_offset)"); break; case nir_tex_src_texture_handle: fprintf(fp, "(texture_handle)"); break; case nir_tex_src_sampler_handle: fprintf(fp, "(sampler_handle)"); break; case nir_tex_src_plane: fprintf(fp, "(plane)"); break; default: unreachable("Invalid texture source type"); break; } } if (instr->op == nir_texop_tg4) { fprintf(fp, ", %u (gather_component)", instr->component); } if (nir_tex_instr_has_explicit_tg4_offsets(instr)) { fprintf(fp, ", { (%i, %i)", instr->tg4_offsets[0][0], instr->tg4_offsets[0][1]); for (unsigned i = 1; i < 4; ++i) fprintf(fp, ", (%i, %i)", instr->tg4_offsets[i][0], instr->tg4_offsets[i][1]); fprintf(fp, " } (offsets)"); } if (instr->op != nir_texop_txf_ms_fb) { if (!has_texture_deref) { fprintf(fp, ", %u (texture)", instr->texture_index); } if (!has_sampler_deref) { fprintf(fp, ", %u (sampler)", instr->sampler_index); } } if (instr->texture_non_uniform) { fprintf(fp, ", texture non-uniform"); } if (instr->sampler_non_uniform) { fprintf(fp, ", sampler non-uniform"); } if (instr->is_sparse) { fprintf(fp, ", sparse"); } } static void print_call_instr(nir_call_instr *instr, print_state *state) { FILE *fp = state->fp; fprintf(fp, "call %s ", instr->callee->name); for (unsigned i = 0; i < instr->num_params; i++) { if (i != 0) fprintf(fp, ", "); print_src(&instr->params[i], state); } } static void print_jump_instr(nir_jump_instr *instr, print_state *state) { FILE *fp = state->fp; switch (instr->type) { case nir_jump_break: fprintf(fp, "break"); break; case nir_jump_continue: fprintf(fp, "continue"); break; case nir_jump_return: fprintf(fp, "return"); break; case nir_jump_halt: fprintf(fp, "halt"); break; case nir_jump_goto: fprintf(fp, "goto block_%u", instr->target ? instr->target->index : -1); break; case nir_jump_goto_if: fprintf(fp, "goto block_%u if ", instr->target ? instr->target->index : -1); print_src(&instr->condition, state); fprintf(fp, " else block_%u", instr->else_target ? instr->else_target->index : -1); break; } } static void print_ssa_undef_instr(nir_ssa_undef_instr* instr, print_state *state) { FILE *fp = state->fp; print_ssa_def(&instr->def, state); fprintf(fp, " = undefined"); } static void print_phi_instr(nir_phi_instr *instr, print_state *state) { FILE *fp = state->fp; print_dest(&instr->dest, state); fprintf(fp, " = phi "); nir_foreach_phi_src(src, instr) { if (&src->node != exec_list_get_head(&instr->srcs)) fprintf(fp, ", "); fprintf(fp, "block_%u: ", src->pred->index); print_src(&src->src, state); } } static void print_parallel_copy_instr(nir_parallel_copy_instr *instr, print_state *state) { FILE *fp = state->fp; nir_foreach_parallel_copy_entry(entry, instr) { if (&entry->node != exec_list_get_head(&instr->entries)) fprintf(fp, "; "); print_dest(&entry->dest, state); fprintf(fp, " = "); print_src(&entry->src, state); } } static void print_instr(const nir_instr *instr, print_state *state, unsigned tabs) { FILE *fp = state->fp; print_tabs(tabs, fp); switch (instr->type) { case nir_instr_type_alu: print_alu_instr(nir_instr_as_alu(instr), state); break; case nir_instr_type_deref: print_deref_instr(nir_instr_as_deref(instr), state); break; case nir_instr_type_call: print_call_instr(nir_instr_as_call(instr), state); break; case nir_instr_type_intrinsic: print_intrinsic_instr(nir_instr_as_intrinsic(instr), state); break; case nir_instr_type_tex: print_tex_instr(nir_instr_as_tex(instr), state); break; case nir_instr_type_load_const: print_load_const_instr(nir_instr_as_load_const(instr), state); break; case nir_instr_type_jump: print_jump_instr(nir_instr_as_jump(instr), state); break; case nir_instr_type_ssa_undef: print_ssa_undef_instr(nir_instr_as_ssa_undef(instr), state); break; case nir_instr_type_phi: print_phi_instr(nir_instr_as_phi(instr), state); break; case nir_instr_type_parallel_copy: print_parallel_copy_instr(nir_instr_as_parallel_copy(instr), state); break; default: unreachable("Invalid instruction type"); break; } } static void print_cf_node(nir_cf_node *node, print_state *state, unsigned tabs); static void print_block(nir_block *block, print_state *state, unsigned tabs) { FILE *fp = state->fp; print_tabs(tabs, fp); fprintf(fp, "block block_%u:\n", block->index); nir_block **preds = nir_block_get_predecessors_sorted(block, NULL); print_tabs(tabs, fp); fprintf(fp, "/* preds: "); for (unsigned i = 0; i < block->predecessors->entries; i++) { fprintf(fp, "block_%u ", preds[i]->index); } fprintf(fp, "*/\n"); ralloc_free(preds); nir_foreach_instr(instr, block) { print_instr(instr, state, tabs); fprintf(fp, "\n"); print_annotation(state, instr); } print_tabs(tabs, fp); fprintf(fp, "/* succs: "); for (unsigned i = 0; i < 2; i++) if (block->successors[i]) { fprintf(fp, "block_%u ", block->successors[i]->index); } fprintf(fp, "*/\n"); } static void print_if(nir_if *if_stmt, print_state *state, unsigned tabs) { FILE *fp = state->fp; print_tabs(tabs, fp); fprintf(fp, "if "); print_src(&if_stmt->condition, state); fprintf(fp, " {\n"); foreach_list_typed(nir_cf_node, node, node, &if_stmt->then_list) { print_cf_node(node, state, tabs + 1); } print_tabs(tabs, fp); fprintf(fp, "} else {\n"); foreach_list_typed(nir_cf_node, node, node, &if_stmt->else_list) { print_cf_node(node, state, tabs + 1); } print_tabs(tabs, fp); fprintf(fp, "}\n"); } static void print_loop(nir_loop *loop, print_state *state, unsigned tabs) { FILE *fp = state->fp; print_tabs(tabs, fp); fprintf(fp, "loop {\n"); foreach_list_typed(nir_cf_node, node, node, &loop->body) { print_cf_node(node, state, tabs + 1); } print_tabs(tabs, fp); fprintf(fp, "}\n"); } static void print_cf_node(nir_cf_node *node, print_state *state, unsigned int tabs) { switch (node->type) { case nir_cf_node_block: print_block(nir_cf_node_as_block(node), state, tabs); break; case nir_cf_node_if: print_if(nir_cf_node_as_if(node), state, tabs); break; case nir_cf_node_loop: print_loop(nir_cf_node_as_loop(node), state, tabs); break; default: unreachable("Invalid CFG node type"); } } static void print_function_impl(nir_function_impl *impl, print_state *state) { FILE *fp = state->fp; fprintf(fp, "\nimpl %s ", impl->function->name); fprintf(fp, "{\n"); if (impl->preamble) { fprintf(fp, "\tpreamble %s\n", impl->preamble->name); } nir_foreach_function_temp_variable(var, impl) { fprintf(fp, "\t"); print_var_decl(var, state); } foreach_list_typed(nir_register, reg, node, &impl->registers) { fprintf(fp, "\t"); print_register_decl(reg, state); } nir_index_blocks(impl); foreach_list_typed(nir_cf_node, node, node, &impl->body) { print_cf_node(node, state, 1); } fprintf(fp, "\tblock block_%u:\n}\n\n", impl->end_block->index); } static void print_function(nir_function *function, print_state *state) { FILE *fp = state->fp; fprintf(fp, "decl_function %s (%d params)", function->name, function->num_params); fprintf(fp, "\n"); if (function->impl != NULL) { print_function_impl(function->impl, state); return; } } static void init_print_state(print_state *state, nir_shader *shader, FILE *fp) { state->fp = fp; state->shader = shader; state->ht = _mesa_pointer_hash_table_create(NULL); state->syms = _mesa_set_create(NULL, _mesa_hash_string, _mesa_key_string_equal); state->index = 0; } static void destroy_print_state(print_state *state) { _mesa_hash_table_destroy(state->ht, NULL); _mesa_set_destroy(state->syms, NULL); } static const char * primitive_name(unsigned primitive) { #define PRIM(X) case SHADER_PRIM_ ## X : return #X switch (primitive) { PRIM(POINTS); PRIM(LINES); PRIM(LINE_LOOP); PRIM(LINE_STRIP); PRIM(TRIANGLES); PRIM(TRIANGLE_STRIP); PRIM(TRIANGLE_FAN); PRIM(QUADS); PRIM(QUAD_STRIP); PRIM(POLYGON); default: return "UNKNOWN"; } } void nir_print_shader_annotated(nir_shader *shader, FILE *fp, struct hash_table *annotations) { print_state state; init_print_state(&state, shader, fp); state.annotations = annotations; fprintf(fp, "shader: %s\n", gl_shader_stage_name(shader->info.stage)); fprintf(fp, "source_sha1: {"); _mesa_sha1_print(fp, shader->info.source_sha1); fprintf(fp, "}\n"); if (shader->info.name) fprintf(fp, "name: %s\n", shader->info.name); if (shader->info.label) fprintf(fp, "label: %s\n", shader->info.label); if (gl_shader_stage_uses_workgroup(shader->info.stage)) { fprintf(fp, "workgroup-size: %u, %u, %u%s\n", shader->info.workgroup_size[0], shader->info.workgroup_size[1], shader->info.workgroup_size[2], shader->info.workgroup_size_variable ? " (variable)" : ""); fprintf(fp, "shared-size: %u\n", shader->info.shared_size); } if (shader->info.stage == MESA_SHADER_MESH || shader->info.stage == MESA_SHADER_TASK) { fprintf(fp, "task_payload-size: %u\n", shader->info.task_payload_size); } fprintf(fp, "inputs: %u\n", shader->num_inputs); fprintf(fp, "outputs: %u\n", shader->num_outputs); fprintf(fp, "uniforms: %u\n", shader->num_uniforms); if (shader->info.num_ubos) fprintf(fp, "ubos: %u\n", shader->info.num_ubos); fprintf(fp, "shared: %u\n", shader->info.shared_size); fprintf(fp, "ray queries: %u\n", shader->info.ray_queries); if (shader->scratch_size) fprintf(fp, "scratch: %u\n", shader->scratch_size); if (shader->constant_data_size) fprintf(fp, "constants: %u\n", shader->constant_data_size); if (shader->info.stage == MESA_SHADER_GEOMETRY) { fprintf(fp, "invocations: %u\n", shader->info.gs.invocations); fprintf(fp, "vertices in: %u\n", shader->info.gs.vertices_in); fprintf(fp, "vertices out: %u\n", shader->info.gs.vertices_out); fprintf(fp, "input primitive: %s\n", primitive_name(shader->info.gs.input_primitive)); fprintf(fp, "output primitive: %s\n", primitive_name(shader->info.gs.output_primitive)); fprintf(fp, "active_stream_mask: 0x%x\n", shader->info.gs.active_stream_mask); fprintf(fp, "uses_end_primitive: %u\n", shader->info.gs.uses_end_primitive); } else if (shader->info.stage == MESA_SHADER_MESH) { fprintf(fp, "output primitive: %s\n", primitive_name(shader->info.mesh.primitive_type)); fprintf(fp, "max primitives out: %u\n", shader->info.mesh.max_primitives_out); fprintf(fp, "max vertices out: %u\n", shader->info.mesh.max_vertices_out); } nir_foreach_variable_in_shader(var, shader) print_var_decl(var, &state); foreach_list_typed(nir_function, func, node, &shader->functions) { print_function(func, &state); } destroy_print_state(&state); } void nir_print_shader(nir_shader *shader, FILE *fp) { nir_print_shader_annotated(shader, fp, NULL); fflush(fp); } char * nir_shader_as_str_annotated(nir_shader *nir, struct hash_table *annotations, void *mem_ctx) { char *stream_data = NULL; size_t stream_size = 0; struct u_memstream mem; if (u_memstream_open(&mem, &stream_data, &stream_size)) { FILE *const stream = u_memstream_get(&mem); nir_print_shader_annotated(nir, stream, annotations); u_memstream_close(&mem); } char *str = ralloc_size(mem_ctx, stream_size + 1); memcpy(str, stream_data, stream_size); str[stream_size] = '\0'; free(stream_data); return str; } char * nir_shader_as_str(nir_shader *nir, void *mem_ctx) { return nir_shader_as_str_annotated(nir, NULL, mem_ctx); } void nir_print_instr(const nir_instr *instr, FILE *fp) { print_state state = { .fp = fp, }; if (instr->block) { nir_function_impl *impl = nir_cf_node_get_function(&instr->block->cf_node); state.shader = impl->function->shader; } print_instr(instr, &state, 0); } void nir_print_deref(const nir_deref_instr *deref, FILE *fp) { print_state state = { .fp = fp, }; print_deref_link(deref, true, &state); } void nir_log_shader_annotated_tagged(enum mesa_log_level level, const char *tag, nir_shader *shader, struct hash_table *annotations) { char *str = nir_shader_as_str_annotated(shader, annotations, NULL); _mesa_log_multiline(level, tag, str); ralloc_free(str); }