/* * Copyright © 2016 Red Hat. * Copyright © 2016 Bas Nieuwenhuizen * * based in part on anv driver which is: * Copyright © 2015 Intel Corporation * * SPDX-License-Identifier: MIT */ #include "radv_pipeline.h" #include "meta/radv_meta.h" #include "nir/nir.h" #include "nir/radv_nir.h" #include "spirv/nir_spirv.h" #include "util/disk_cache.h" #include "util/os_time.h" #include "util/u_atomic.h" #include "radv_cs.h" #include "radv_debug.h" #include "radv_descriptors.h" #include "radv_pipeline_rt.h" #include "radv_rmv.h" #include "radv_shader.h" #include "radv_shader_args.h" #include "vk_pipeline.h" #include "vk_render_pass.h" #include "vk_util.h" #include "util/u_debug.h" #include "ac_binary.h" #include "ac_nir.h" #include "ac_shader_util.h" #include "aco_interface.h" #include "sid.h" #include "vk_format.h" #include "vk_nir_convert_ycbcr.h" #include "vk_ycbcr_conversion.h" bool radv_shader_need_indirect_descriptor_sets(const struct radv_shader *shader) { const struct radv_userdata_info *loc = radv_get_user_sgpr_info(shader, AC_UD_INDIRECT_DESCRIPTOR_SETS); return loc->sgpr_idx != -1; } bool radv_pipeline_capture_shaders(const struct radv_device *device, VkPipelineCreateFlags2 flags) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); return (flags & VK_PIPELINE_CREATE_2_CAPTURE_INTERNAL_REPRESENTATIONS_BIT_KHR) || (instance->debug_flags & RADV_DEBUG_DUMP_SHADERS) || device->keep_shader_info; } bool radv_pipeline_capture_shader_stats(const struct radv_device *device, VkPipelineCreateFlags2 flags) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); /* Capture shader statistics when RGP is enabled to correlate shader hashes with Fossilize. */ return (flags & VK_PIPELINE_CREATE_2_CAPTURE_STATISTICS_BIT_KHR) || (instance->debug_flags & (RADV_DEBUG_DUMP_SHADER_STATS | RADV_DEBUG_PSO_HISTORY)) || device->keep_shader_info || (instance->vk.trace_mode & RADV_TRACE_MODE_RGP); } bool radv_pipeline_skip_shaders_cache(const struct radv_device *device, const struct radv_pipeline *pipeline) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); /* Skip the shaders cache when any of the below are true: * - shaders are dumped for debugging (RADV_DEBUG=shaders) * - shaders IR are captured (NIR, backend IR and ASM) * - binaries are captured (driver shouldn't store data to an internal cache) */ return (instance->debug_flags & RADV_DEBUG_DUMP_SHADERS) || (pipeline->create_flags & (VK_PIPELINE_CREATE_2_CAPTURE_INTERNAL_REPRESENTATIONS_BIT_KHR | VK_PIPELINE_CREATE_2_CAPTURE_DATA_BIT_KHR)); } void radv_pipeline_init(struct radv_device *device, struct radv_pipeline *pipeline, enum radv_pipeline_type type) { vk_object_base_init(&device->vk, &pipeline->base, VK_OBJECT_TYPE_PIPELINE); pipeline->type = type; } void radv_pipeline_destroy(struct radv_device *device, struct radv_pipeline *pipeline, const VkAllocationCallbacks *allocator) { if (pipeline->cache_object) vk_pipeline_cache_object_unref(&device->vk, pipeline->cache_object); switch (pipeline->type) { case RADV_PIPELINE_GRAPHICS: radv_destroy_graphics_pipeline(device, radv_pipeline_to_graphics(pipeline)); break; case RADV_PIPELINE_GRAPHICS_LIB: radv_destroy_graphics_lib_pipeline(device, radv_pipeline_to_graphics_lib(pipeline)); break; case RADV_PIPELINE_COMPUTE: radv_destroy_compute_pipeline(device, radv_pipeline_to_compute(pipeline)); break; case RADV_PIPELINE_RAY_TRACING: radv_destroy_ray_tracing_pipeline(device, radv_pipeline_to_ray_tracing(pipeline)); break; default: UNREACHABLE("invalid pipeline type"); } radv_rmv_log_resource_destroy(device, (uint64_t)radv_pipeline_to_handle(pipeline)); vk_object_base_finish(&pipeline->base); vk_free2(&device->vk.alloc, allocator, pipeline); } VKAPI_ATTR void VKAPI_CALL radv_DestroyPipeline(VkDevice _device, VkPipeline _pipeline, const VkAllocationCallbacks *pAllocator) { VK_FROM_HANDLE(radv_device, device, _device); VK_FROM_HANDLE(radv_pipeline, pipeline, _pipeline); if (!_pipeline) return; radv_pipeline_destroy(device, pipeline, pAllocator); } struct radv_shader_stage_key radv_pipeline_get_shader_key(const struct radv_device *device, const VkPipelineShaderStageCreateInfo *stage, VkPipelineCreateFlags2 flags, const void *pNext) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); gl_shader_stage s = vk_to_mesa_shader_stage(stage->stage); struct vk_pipeline_robustness_state rs; struct radv_shader_stage_key key = {0}; key.keep_statistic_info = radv_pipeline_capture_shader_stats(device, flags); if (flags & VK_PIPELINE_CREATE_2_DISABLE_OPTIMIZATION_BIT) key.optimisations_disabled = 1; if (flags & VK_PIPELINE_CREATE_2_VIEW_INDEX_FROM_DEVICE_INDEX_BIT) key.view_index_from_device_index = 1; if (flags & VK_PIPELINE_CREATE_2_INDIRECT_BINDABLE_BIT_EXT) key.indirect_bindable = 1; if (stage->stage & RADV_GRAPHICS_STAGE_BITS) { key.version = instance->drirc.override_graphics_shader_version; } else if (stage->stage & RADV_RT_STAGE_BITS) { key.version = instance->drirc.override_ray_tracing_shader_version; } else { assert(stage->stage == VK_SHADER_STAGE_COMPUTE_BIT); key.version = instance->drirc.override_compute_shader_version; } vk_pipeline_robustness_state_fill(&device->vk, &rs, pNext, stage->pNext); radv_set_stage_key_robustness(&rs, s, &key); const VkPipelineShaderStageRequiredSubgroupSizeCreateInfo *const subgroup_size = vk_find_struct_const(stage->pNext, PIPELINE_SHADER_STAGE_REQUIRED_SUBGROUP_SIZE_CREATE_INFO); if (subgroup_size) { if (subgroup_size->requiredSubgroupSize == 32) key.subgroup_required_size = RADV_REQUIRED_WAVE32; else if (subgroup_size->requiredSubgroupSize == 64) key.subgroup_required_size = RADV_REQUIRED_WAVE64; else UNREACHABLE("Unsupported required subgroup size."); } if (stage->flags & VK_PIPELINE_SHADER_STAGE_CREATE_REQUIRE_FULL_SUBGROUPS_BIT) { key.subgroup_require_full = 1; } return key; } void radv_pipeline_stage_init(VkPipelineCreateFlags2 pipeline_flags, const VkPipelineShaderStageCreateInfo *sinfo, const struct radv_pipeline_layout *pipeline_layout, const struct radv_shader_stage_key *stage_key, struct radv_shader_stage *out_stage) { const VkShaderModuleCreateInfo *minfo = vk_find_struct_const(sinfo->pNext, SHADER_MODULE_CREATE_INFO); const VkPipelineShaderStageModuleIdentifierCreateInfoEXT *iinfo = vk_find_struct_const(sinfo->pNext, PIPELINE_SHADER_STAGE_MODULE_IDENTIFIER_CREATE_INFO_EXT); if (sinfo->module == VK_NULL_HANDLE && !minfo && !iinfo) return; memset(out_stage, 0, sizeof(*out_stage)); out_stage->stage = vk_to_mesa_shader_stage(sinfo->stage); out_stage->next_stage = MESA_SHADER_NONE; out_stage->entrypoint = sinfo->pName; out_stage->spec_info = sinfo->pSpecializationInfo; out_stage->feedback.flags = VK_PIPELINE_CREATION_FEEDBACK_VALID_BIT; out_stage->key = *stage_key; if (sinfo->module != VK_NULL_HANDLE) { struct vk_shader_module *module = vk_shader_module_from_handle(sinfo->module); out_stage->spirv.data = module->data; out_stage->spirv.size = module->size; out_stage->spirv.object = &module->base; if (module->nir) out_stage->internal_nir = module->nir; } else if (minfo) { out_stage->spirv.data = (const char *)minfo->pCode; out_stage->spirv.size = minfo->codeSize; } radv_shader_layout_init(pipeline_layout, out_stage->stage, &out_stage->layout); vk_pipeline_hash_shader_stage(pipeline_flags, sinfo, NULL, out_stage->shader_sha1); } void radv_shader_layout_init(const struct radv_pipeline_layout *pipeline_layout, gl_shader_stage stage, struct radv_shader_layout *layout) { layout->num_sets = pipeline_layout->num_sets; for (unsigned i = 0; i < pipeline_layout->num_sets; i++) { layout->set[i].layout = pipeline_layout->set[i].layout; layout->set[i].dynamic_offset_start = pipeline_layout->set[i].dynamic_offset_start; } layout->push_constant_size = pipeline_layout->push_constant_size; layout->use_dynamic_descriptors = pipeline_layout->dynamic_offset_count && (pipeline_layout->dynamic_shader_stages & mesa_to_vk_shader_stage(stage)); } static const struct vk_ycbcr_conversion_state * ycbcr_conversion_lookup(const void *data, uint32_t set, uint32_t binding, uint32_t array_index) { const struct radv_shader_layout *layout = data; const struct radv_descriptor_set_layout *set_layout = layout->set[set].layout; const struct vk_ycbcr_conversion_state *ycbcr_samplers = radv_immutable_ycbcr_samplers(set_layout, binding); if (!ycbcr_samplers) return NULL; return ycbcr_samplers + array_index; } static uint8_t max_alu_src_identity_swizzle(const nir_alu_instr *alu, const nir_alu_src *src) { uint8_t max_vector = 32 / alu->def.bit_size; if (nir_src_is_const(src->src)) return max_vector; /* Return the number of correctly swizzled components. */ for (unsigned i = 1; i < alu->def.num_components; i++) { if (src->swizzle[i] != src->swizzle[0] + i) /* Ensure that the result is a power of 2. */ return MAX2(i & 0x6, 1); } return max_vector; } static uint8_t opt_vectorize_callback(const nir_instr *instr, const void *_) { if (instr->type != nir_instr_type_alu) return 0; const struct radv_device *device = _; const struct radv_physical_device *pdev = radv_device_physical(device); enum amd_gfx_level chip = pdev->info.gfx_level; if (chip < GFX9) return 1; const nir_alu_instr *alu = nir_instr_as_alu(instr); switch (alu->op) { case nir_op_f2e4m3fn: case nir_op_f2e4m3fn_sat: case nir_op_f2e4m3fn_satfn: case nir_op_f2e5m2: case nir_op_f2e5m2_sat: case nir_op_e4m3fn2f: case nir_op_e5m22f: return 2; default: break; } const unsigned bit_size = alu->def.bit_size; if (bit_size == 16 && aco_nir_op_supports_packed_math_16bit(alu)) return 2; if (bit_size != 8 && bit_size != 16) return 1; /* Keep some opcodes vectorized if the operation can be performed as * 32-bit instruction with packed sources. The condition is that the * sources must have identity swizzles. */ uint8_t target_width = 32 / bit_size; switch (alu->op) { case nir_op_bcsel: /* Must have scalar condition. */ for (unsigned i = 1; i < alu->def.num_components; i++) { if (alu->src[0].swizzle[i] != alu->src[0].swizzle[0]) return 1; } for (unsigned idx = 1; idx < 3; idx++) target_width = MIN2(target_width, max_alu_src_identity_swizzle(alu, &alu->src[idx])); break; case nir_op_iand: case nir_op_ior: case nir_op_ixor: case nir_op_inot: case nir_op_bitfield_select: for (unsigned idx = 0; idx < nir_op_infos[alu->op].num_inputs; idx++) target_width = MIN2(target_width, max_alu_src_identity_swizzle(alu, &alu->src[idx])); break; default: return 1; } return target_width; } static nir_component_mask_t non_uniform_access_callback(const nir_src *src, void *_) { if (src->ssa->num_components == 1) return 0x1; return nir_chase_binding(*src).success ? 0x2 : 0x3; } void radv_postprocess_nir(struct radv_device *device, const struct radv_graphics_state_key *gfx_state, struct radv_shader_stage *stage) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); enum amd_gfx_level gfx_level = pdev->info.gfx_level; bool progress; /* Wave and workgroup size should already be filled. */ assert(stage->info.wave_size && stage->info.workgroup_size); if (stage->stage == MESA_SHADER_FRAGMENT) { if (!stage->key.optimisations_disabled) { NIR_PASS(_, stage->nir, nir_opt_cse); } NIR_PASS(_, stage->nir, radv_nir_lower_fs_intrinsics, stage, gfx_state); } /* LLVM could support more of these in theory. */ bool use_llvm = radv_use_llvm_for_stage(pdev, stage->stage); radv_nir_opt_tid_function_options tid_options = { .use_masked_swizzle_amd = true, .use_dpp16_shift_amd = !use_llvm && gfx_level >= GFX8, .use_clustered_rotate = !use_llvm, .hw_subgroup_size = stage->info.wave_size, .hw_ballot_bit_size = stage->info.wave_size, .hw_ballot_num_comp = 1, }; NIR_PASS(_, stage->nir, radv_nir_opt_tid_function, &tid_options); nir_divergence_analysis(stage->nir); NIR_PASS(_, stage->nir, ac_nir_flag_smem_for_loads, gfx_level, use_llvm, false); NIR_PASS(_, stage->nir, nir_lower_memory_model); nir_load_store_vectorize_options vectorize_opts = { .modes = nir_var_mem_ssbo | nir_var_mem_ubo | nir_var_mem_push_const | nir_var_mem_shared | nir_var_mem_global | nir_var_shader_temp, .callback = ac_nir_mem_vectorize_callback, .cb_data = &(struct ac_nir_config){gfx_level, !use_llvm}, .robust_modes = 0, /* On GFX6, read2/write2 is out-of-bounds if the offset register is negative, even if * the final offset is not. */ .has_shared2_amd = gfx_level >= GFX7, }; if (stage->key.uniform_robustness2) vectorize_opts.robust_modes |= nir_var_mem_ubo; if (stage->key.storage_robustness2) vectorize_opts.robust_modes |= nir_var_mem_ssbo; bool constant_fold_for_push_const = false; if (!stage->key.optimisations_disabled) { progress = false; NIR_PASS(progress, stage->nir, nir_opt_load_store_vectorize, &vectorize_opts); if (progress) { NIR_PASS(_, stage->nir, nir_copy_prop); NIR_PASS(_, stage->nir, nir_opt_shrink_stores, !instance->drirc.disable_shrink_image_store); constant_fold_for_push_const = true; } } enum nir_lower_non_uniform_access_type lower_non_uniform_access_types = nir_lower_non_uniform_ubo_access | nir_lower_non_uniform_ssbo_access | nir_lower_non_uniform_texture_access | nir_lower_non_uniform_image_access; /* In practice, most shaders do not have non-uniform-qualified * accesses (see * https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/17558#note_1475069) * thus a cheaper and likely to fail check is run first. */ if (nir_has_non_uniform_access(stage->nir, lower_non_uniform_access_types)) { if (!stage->key.optimisations_disabled) { NIR_PASS(_, stage->nir, nir_opt_non_uniform_access); } if (!radv_use_llvm_for_stage(pdev, stage->stage)) { nir_lower_non_uniform_access_options options = { .types = lower_non_uniform_access_types, .callback = &non_uniform_access_callback, .callback_data = NULL, }; NIR_PASS(_, stage->nir, nir_lower_non_uniform_access, &options); } } progress = false; NIR_PASS(progress, stage->nir, ac_nir_lower_mem_access_bit_sizes, gfx_level, use_llvm); if (progress) constant_fold_for_push_const = true; progress = false; NIR_PASS(progress, stage->nir, nir_vk_lower_ycbcr_tex, ycbcr_conversion_lookup, &stage->layout); /* Gather info in the case that nir_vk_lower_ycbcr_tex might have emitted resinfo instructions. */ if (progress) nir_shader_gather_info(stage->nir, nir_shader_get_entrypoint(stage->nir)); NIR_PASS( _, stage->nir, ac_nir_lower_tex, &(ac_nir_lower_tex_options){ .gfx_level = gfx_level, .lower_array_layer_round_even = !pdev->info.conformant_trunc_coord || instance->drirc.disable_trunc_coord, .fix_derivs_in_divergent_cf = stage->stage == MESA_SHADER_FRAGMENT && !radv_use_llvm_for_stage(pdev, stage->stage), .max_wqm_vgprs = 64, // TODO: improve spiller and RA support for linear VGPRs }); if (stage->nir->info.uses_resource_info_query) NIR_PASS(_, stage->nir, ac_nir_lower_resinfo, gfx_level); /* Ensure split load_push_constant still have constant offsets, for radv_nir_apply_pipeline_layout. */ if (constant_fold_for_push_const && stage->args.ac.inline_push_const_mask) NIR_PASS(_, stage->nir, nir_opt_constant_folding); NIR_PASS(_, stage->nir, radv_nir_apply_pipeline_layout, device, stage); NIR_PASS(_, stage->nir, nir_lower_alu_width, opt_vectorize_callback, device); nir_move_options sink_opts = nir_move_const_undef | nir_move_copies | nir_dont_move_byte_word_vecs; if (!stage->key.optimisations_disabled) { NIR_PASS(_, stage->nir, nir_opt_licm); if (stage->stage == MESA_SHADER_VERTEX) { /* Always load all VS inputs at the top to eliminate needless VMEM->s_wait->VMEM sequences. * Each s_wait can cost 1000 cycles, so make sure all VS input loads are grouped. */ NIR_PASS(_, stage->nir, nir_opt_move_to_top, nir_move_to_top_input_loads); NIR_PASS(_, stage->nir, nir_opt_sink, sink_opts); NIR_PASS(_, stage->nir, nir_opt_move, sink_opts); } else { if (stage->stage != MESA_SHADER_FRAGMENT || !pdev->cache_key.disable_sinking_load_input_fs) sink_opts |= nir_move_load_input | nir_move_load_frag_coord; NIR_PASS(_, stage->nir, nir_opt_sink, sink_opts); NIR_PASS(_, stage->nir, nir_opt_move, sink_opts | nir_move_load_input | nir_move_load_frag_coord); } } /* Lower VS inputs. We need to do this after nir_opt_sink, because * load_input can be reordered, but buffer loads can't. */ if (stage->stage == MESA_SHADER_VERTEX) { NIR_PASS(_, stage->nir, radv_nir_lower_vs_inputs, stage, gfx_state, &pdev->info); } /* Lower I/O intrinsics to memory instructions. */ bool is_last_vgt_stage = radv_is_last_vgt_stage(stage); bool io_to_mem = radv_nir_lower_io_to_mem(device, stage); bool lowered_ngg = stage->info.is_ngg && is_last_vgt_stage; if (lowered_ngg) { radv_lower_ngg(device, stage, gfx_state); } else if (is_last_vgt_stage) { if (stage->stage != MESA_SHADER_GEOMETRY) { NIR_PASS(_, stage->nir, ac_nir_lower_legacy_vs, gfx_level, stage->info.outinfo.clip_dist_mask | stage->info.outinfo.cull_dist_mask, false, stage->info.outinfo.vs_output_param_offset, stage->info.outinfo.param_exports, stage->info.outinfo.export_prim_id, false, stage->info.force_vrs_per_vertex); } else { ac_nir_lower_legacy_gs_options options = { .has_gen_prim_query = false, .has_pipeline_stats_query = false, .gfx_level = pdev->info.gfx_level, .export_clipdist_mask = stage->info.outinfo.clip_dist_mask | stage->info.outinfo.cull_dist_mask, .param_offsets = stage->info.outinfo.vs_output_param_offset, .has_param_exports = stage->info.outinfo.param_exports, .force_vrs = stage->info.force_vrs_per_vertex, }; ac_nir_legacy_gs_info info = {0}; NIR_PASS(_, stage->nir, ac_nir_lower_legacy_gs, &options, &stage->gs_copy_shader, &info); for (unsigned i = 0; i < 4; i++) stage->info.gs.num_components_per_stream[i] = info.num_components_per_stream[i]; } } else if (stage->stage == MESA_SHADER_FRAGMENT) { ac_nir_lower_ps_late_options late_options = { .gfx_level = gfx_level, .family = pdev->info.family, .use_aco = !radv_use_llvm_for_stage(pdev, stage->stage), .bc_optimize_for_persp = G_0286CC_PERSP_CENTER_ENA(stage->info.ps.spi_ps_input_ena) && G_0286CC_PERSP_CENTROID_ENA(stage->info.ps.spi_ps_input_ena), .bc_optimize_for_linear = G_0286CC_LINEAR_CENTER_ENA(stage->info.ps.spi_ps_input_ena) && G_0286CC_LINEAR_CENTROID_ENA(stage->info.ps.spi_ps_input_ena), .uses_discard = stage->info.ps.can_discard, .dcc_decompress_gfx11 = gfx_state->dcc_decompress_gfx11, .no_color_export = stage->info.ps.has_epilog, .no_depth_export = stage->info.ps.exports_mrtz_via_epilog, }; if (!late_options.no_color_export) { late_options.dual_src_blend_swizzle = gfx_state->ps.epilog.mrt0_is_dual_src && gfx_level >= GFX11; late_options.color_is_int8 = gfx_state->ps.epilog.color_is_int8; late_options.color_is_int10 = gfx_state->ps.epilog.color_is_int10; late_options.enable_mrt_output_nan_fixup = gfx_state->ps.epilog.enable_mrt_output_nan_fixup && !stage->nir->info.internal; /* Need to filter out unwritten color slots. */ late_options.spi_shader_col_format = gfx_state->ps.epilog.spi_shader_col_format & stage->info.ps.colors_written; late_options.alpha_to_one = gfx_state->ps.epilog.alpha_to_one; } if (!late_options.no_depth_export) { /* Compared to gfx_state.ps.alpha_to_coverage_via_mrtz, * radv_shader_info.ps.writes_mrt0_alpha need any depth/stencil/sample_mask exist. * ac_nir_lower_ps() require this field to reflect whether alpha via mrtz is really * present. */ late_options.alpha_to_coverage_via_mrtz = stage->info.ps.writes_mrt0_alpha; } NIR_PASS(_, stage->nir, ac_nir_lower_ps_late, &late_options); } if (radv_shader_should_clear_lds(device, stage->nir)) { const unsigned chunk_size = 16; /* max single store size */ const unsigned shared_size = ALIGN(stage->nir->info.shared_size, chunk_size); NIR_PASS(_, stage->nir, nir_clear_shared_memory, shared_size, chunk_size); } /* This must be after lowering resources to descriptor loads and before lowering intrinsics * to args and lowering int64. */ if (!radv_use_llvm_for_stage(pdev, stage->stage)) ac_nir_optimize_uniform_atomics(stage->nir); NIR_PASS(_, stage->nir, nir_lower_int64); NIR_PASS(_, stage->nir, nir_opt_idiv_const, 8); NIR_PASS(_, stage->nir, nir_lower_idiv, &(nir_lower_idiv_options){ .allow_fp16 = gfx_level >= GFX9, }); NIR_PASS(_, stage->nir, ac_nir_lower_global_access); NIR_PASS(_, stage->nir, ac_nir_lower_intrinsics_to_args, gfx_level, pdev->info.has_ls_vgpr_init_bug && gfx_state && !gfx_state->vs.has_prolog, radv_select_hw_stage(&stage->info, gfx_level), stage->info.wave_size, stage->info.workgroup_size, &stage->args.ac); NIR_PASS(_, stage->nir, radv_nir_lower_abi, gfx_level, stage, gfx_state, pdev->info.address32_hi); if (!stage->key.optimisations_disabled) { NIR_PASS(_, stage->nir, nir_opt_dce); NIR_PASS(_, stage->nir, nir_opt_shrink_vectors, true); NIR_PASS(_, stage->nir, nir_copy_prop); NIR_PASS(_, stage->nir, nir_opt_constant_folding); NIR_PASS(_, stage->nir, nir_opt_cse); nir_load_store_vectorize_options late_vectorize_opts = { .modes = nir_var_mem_global | nir_var_mem_shared | nir_var_shader_out | nir_var_mem_task_payload | nir_var_shader_in, .callback = ac_nir_mem_vectorize_callback, .cb_data = &(struct ac_nir_config){gfx_level, !use_llvm}, .robust_modes = 0, /* On GFX6, read2/write2 is out-of-bounds if the offset register is negative, even if * the final offset is not. */ .has_shared2_amd = gfx_level >= GFX7, }; progress = false; NIR_PASS(progress, stage->nir, nir_opt_load_store_vectorize, &late_vectorize_opts); if (progress) NIR_PASS(_, stage->nir, ac_nir_lower_mem_access_bit_sizes, gfx_level, use_llvm); } radv_optimize_nir_algebraic( stage->nir, io_to_mem || lowered_ngg || stage->stage == MESA_SHADER_COMPUTE || stage->stage == MESA_SHADER_TASK, gfx_level >= GFX8); if (stage->nir->info.cs.has_cooperative_matrix) NIR_PASS(_, stage->nir, radv_nir_opt_cooperative_matrix, gfx_level); NIR_PASS(_, stage->nir, nir_lower_fp16_casts, nir_lower_fp16_split_fp64); if (ac_nir_might_lower_bit_size(stage->nir)) { if (gfx_level >= GFX8) nir_divergence_analysis(stage->nir); if (nir_lower_bit_size(stage->nir, ac_nir_lower_bit_size_callback, &gfx_level)) { NIR_PASS(_, stage->nir, nir_opt_constant_folding); } } if (gfx_level >= GFX9) { bool separate_g16 = gfx_level >= GFX10; struct nir_opt_tex_srcs_options opt_srcs_options[] = { { .sampler_dims = ~(BITFIELD_BIT(GLSL_SAMPLER_DIM_CUBE) | BITFIELD_BIT(GLSL_SAMPLER_DIM_BUF)), .src_types = (1 << nir_tex_src_coord) | (1 << nir_tex_src_lod) | (1 << nir_tex_src_bias) | (1 << nir_tex_src_min_lod) | (1 << nir_tex_src_ms_index) | (separate_g16 ? 0 : (1 << nir_tex_src_ddx) | (1 << nir_tex_src_ddy)), }, { .sampler_dims = ~BITFIELD_BIT(GLSL_SAMPLER_DIM_CUBE), .src_types = (1 << nir_tex_src_ddx) | (1 << nir_tex_src_ddy), }, }; struct nir_opt_16bit_tex_image_options opt_16bit_options = { .rounding_mode = nir_rounding_mode_undef, .opt_tex_dest_types = nir_type_float | nir_type_int | nir_type_uint, .opt_image_dest_types = nir_type_float | nir_type_int | nir_type_uint, .integer_dest_saturates = true, .opt_image_store_data = true, .opt_image_srcs = true, .opt_srcs_options_count = separate_g16 ? 2 : 1, .opt_srcs_options = opt_srcs_options, }; bool run_copy_prop = false; NIR_PASS(run_copy_prop, stage->nir, nir_opt_16bit_tex_image, &opt_16bit_options); /* Optimizing 16bit texture/image dests leaves scalar moves that stops * nir_opt_vectorize from vectorzing the alu uses of them. */ if (run_copy_prop) { NIR_PASS(_, stage->nir, nir_copy_prop); NIR_PASS(_, stage->nir, nir_opt_dce); } if (!stage->key.optimisations_disabled) { NIR_PASS(_, stage->nir, nir_opt_vectorize, opt_vectorize_callback, device); } } /* cleanup passes */ NIR_PASS(_, stage->nir, nir_lower_alu_width, opt_vectorize_callback, device); /* This pass changes the global float control mode to RTZ, so can't be used * with LLVM, which only supports RTNE, or RT, where the mode needs to match * across separately compiled stages. */ if (!radv_use_llvm_for_stage(pdev, stage->stage) && !gl_shader_stage_is_rt(stage->stage)) NIR_PASS(_, stage->nir, ac_nir_opt_pack_half, gfx_level); NIR_PASS(_, stage->nir, nir_lower_load_const_to_scalar); NIR_PASS(_, stage->nir, nir_copy_prop); NIR_PASS(_, stage->nir, nir_opt_dce); if (!stage->key.optimisations_disabled) { sink_opts |= nir_move_comparisons | nir_move_load_ubo | nir_move_load_ssbo | nir_move_alu; NIR_PASS(_, stage->nir, nir_opt_sink, sink_opts); nir_move_options move_opts = nir_move_const_undef | nir_move_load_ubo | nir_move_load_input | nir_move_load_frag_coord | nir_move_comparisons | nir_move_copies | nir_dont_move_byte_word_vecs | nir_move_alu; NIR_PASS(_, stage->nir, nir_opt_move, move_opts); /* Run nir_opt_move again to make sure that comparision are as close as possible to the first use to prevent SCC * spilling. */ NIR_PASS(_, stage->nir, nir_opt_move, nir_move_comparisons); } stage->info.nir_shared_size = stage->nir->info.shared_size; } bool radv_shader_should_clear_lds(const struct radv_device *device, const nir_shader *shader) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); return (shader->info.stage == MESA_SHADER_COMPUTE || shader->info.stage == MESA_SHADER_MESH || shader->info.stage == MESA_SHADER_TASK) && shader->info.shared_size > 0 && instance->drirc.clear_lds; } static uint32_t radv_get_executable_count(struct radv_pipeline *pipeline) { uint32_t ret = 0; if (pipeline->type == RADV_PIPELINE_RAY_TRACING) { struct radv_ray_tracing_pipeline *rt_pipeline = radv_pipeline_to_ray_tracing(pipeline); for (uint32_t i = 0; i < rt_pipeline->stage_count; i++) ret += rt_pipeline->stages[i].shader ? 1 : 0; } for (int i = 0; i < MESA_VULKAN_SHADER_STAGES; ++i) { if (!pipeline->shaders[i]) continue; ret += 1u; if (i == MESA_SHADER_GEOMETRY && pipeline->gs_copy_shader) { ret += 1u; } } return ret; } static struct radv_shader * radv_get_shader_from_executable_index(struct radv_pipeline *pipeline, int index, gl_shader_stage *stage) { if (pipeline->type == RADV_PIPELINE_RAY_TRACING) { struct radv_ray_tracing_pipeline *rt_pipeline = radv_pipeline_to_ray_tracing(pipeline); for (uint32_t i = 0; i < rt_pipeline->stage_count; i++) { struct radv_ray_tracing_stage *rt_stage = &rt_pipeline->stages[i]; if (!rt_stage->shader) continue; if (!index) { *stage = rt_stage->stage; return rt_stage->shader; } index--; } } for (int i = 0; i < MESA_VULKAN_SHADER_STAGES; ++i) { if (!pipeline->shaders[i]) continue; if (!index) { *stage = i; return pipeline->shaders[i]; } --index; if (i == MESA_SHADER_GEOMETRY && pipeline->gs_copy_shader) { if (!index) { *stage = i; return pipeline->gs_copy_shader; } --index; } } *stage = -1; return NULL; } VKAPI_ATTR VkResult VKAPI_CALL radv_GetPipelineExecutablePropertiesKHR(VkDevice _device, const VkPipelineInfoKHR *pPipelineInfo, uint32_t *pExecutableCount, VkPipelineExecutablePropertiesKHR *pProperties) { VK_FROM_HANDLE(radv_pipeline, pipeline, pPipelineInfo->pipeline); VK_OUTARRAY_MAKE_TYPED(VkPipelineExecutablePropertiesKHR, out, pProperties, pExecutableCount); const uint32_t count = radv_get_executable_count(pipeline); for (uint32_t executable_idx = 0; executable_idx < count; executable_idx++) { VkPipelineExecutablePropertiesKHR *props = vk_outarray_next_typed(VkPipelineExecutablePropertiesKHR, &out); if (!props) continue; gl_shader_stage stage; struct radv_shader *shader = radv_get_shader_from_executable_index(pipeline, executable_idx, &stage); props->stages = mesa_to_vk_shader_stage(stage); const char *name = _mesa_shader_stage_to_string(stage); const char *description = NULL; switch (stage) { case MESA_SHADER_VERTEX: description = "Vulkan Vertex Shader"; break; case MESA_SHADER_TESS_CTRL: if (!pipeline->shaders[MESA_SHADER_VERTEX]) { props->stages |= VK_SHADER_STAGE_VERTEX_BIT; name = "vertex + tessellation control"; description = "Combined Vulkan Vertex and Tessellation Control Shaders"; } else { description = "Vulkan Tessellation Control Shader"; } break; case MESA_SHADER_TESS_EVAL: description = "Vulkan Tessellation Evaluation Shader"; break; case MESA_SHADER_GEOMETRY: if (shader->info.type == RADV_SHADER_TYPE_GS_COPY) { name = "geometry copy"; description = "Extra shader stage that loads the GS output ringbuffer into the rasterizer"; break; } if (pipeline->shaders[MESA_SHADER_TESS_CTRL] && !pipeline->shaders[MESA_SHADER_TESS_EVAL]) { props->stages |= VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT; name = "tessellation evaluation + geometry"; description = "Combined Vulkan Tessellation Evaluation and Geometry Shaders"; } else if (!pipeline->shaders[MESA_SHADER_TESS_CTRL] && !pipeline->shaders[MESA_SHADER_VERTEX]) { props->stages |= VK_SHADER_STAGE_VERTEX_BIT; name = "vertex + geometry"; description = "Combined Vulkan Vertex and Geometry Shaders"; } else { description = "Vulkan Geometry Shader"; } break; case MESA_SHADER_FRAGMENT: description = "Vulkan Fragment Shader"; break; case MESA_SHADER_COMPUTE: description = "Vulkan Compute Shader"; break; case MESA_SHADER_MESH: description = "Vulkan Mesh Shader"; break; case MESA_SHADER_TASK: description = "Vulkan Task Shader"; break; case MESA_SHADER_RAYGEN: description = "Vulkan Ray Generation Shader"; break; case MESA_SHADER_ANY_HIT: description = "Vulkan Any-Hit Shader"; break; case MESA_SHADER_CLOSEST_HIT: description = "Vulkan Closest-Hit Shader"; break; case MESA_SHADER_MISS: description = "Vulkan Miss Shader"; break; case MESA_SHADER_INTERSECTION: description = "Shader responsible for traversing the acceleration structure"; break; case MESA_SHADER_CALLABLE: description = "Vulkan Callable Shader"; break; default: UNREACHABLE("Unsupported shader stage"); } props->subgroupSize = shader->info.wave_size; VK_COPY_STR(props->name, name); VK_COPY_STR(props->description, description); } return vk_outarray_status(&out); } VKAPI_ATTR VkResult VKAPI_CALL radv_GetPipelineExecutableStatisticsKHR(VkDevice _device, const VkPipelineExecutableInfoKHR *pExecutableInfo, uint32_t *pStatisticCount, VkPipelineExecutableStatisticKHR *pStatistics) { VK_FROM_HANDLE(radv_device, device, _device); VK_FROM_HANDLE(radv_pipeline, pipeline, pExecutableInfo->pipeline); gl_shader_stage stage; struct radv_shader *shader = radv_get_shader_from_executable_index(pipeline, pExecutableInfo->executableIndex, &stage); const struct radv_physical_device *pdev = radv_device_physical(device); const enum amd_gfx_level gfx_level = pdev->info.gfx_level; unsigned lds_increment = gfx_level >= GFX11 && stage == MESA_SHADER_FRAGMENT ? 1024 : pdev->info.lds_encode_granularity; VK_OUTARRAY_MAKE_TYPED(VkPipelineExecutableStatisticKHR, out, pStatistics, pStatisticCount); struct amd_stats stats = {0}; if (shader->statistics) stats = *shader->statistics; stats.driverhash = pipeline->pipeline_hash; stats.sgprs = shader->config.num_sgprs; stats.vgprs = shader->config.num_vgprs; stats.spillsgprs = shader->config.spilled_sgprs; stats.spillvgprs = shader->config.spilled_vgprs; stats.codesize = shader->exec_size; stats.lds = shader->config.lds_size * lds_increment; stats.scratch = shader->config.scratch_bytes_per_wave; stats.maxwaves = shader->max_waves; switch (stage) { case MESA_SHADER_VERTEX: if (gfx_level <= GFX8 || (!shader->info.vs.as_es && !shader->info.vs.as_ls)) { /* VS inputs when VS is a separate stage */ stats.inputs += util_bitcount(shader->info.vs.input_slot_usage_mask); } break; case MESA_SHADER_TESS_CTRL: if (gfx_level >= GFX9) { /* VS inputs when pipeline has tess */ stats.inputs += util_bitcount(shader->info.vs.input_slot_usage_mask); } /* VS -> TCS inputs */ stats.inputs += shader->info.tcs.num_linked_inputs; break; case MESA_SHADER_TESS_EVAL: if (gfx_level <= GFX8 || !shader->info.tes.as_es) { /* TCS -> TES inputs when TES is a separate stage */ stats.inputs += shader->info.tes.num_linked_inputs + shader->info.tes.num_linked_patch_inputs; } break; case MESA_SHADER_GEOMETRY: /* The IO stats of the GS copy shader are already reflected by GS and FS, so leave it empty. */ if (shader->info.type == RADV_SHADER_TYPE_GS_COPY) break; if (gfx_level >= GFX9) { if (shader->info.gs.es_type == MESA_SHADER_VERTEX) { /* VS inputs when pipeline has GS but no tess */ stats.inputs += util_bitcount(shader->info.vs.input_slot_usage_mask); } else if (shader->info.gs.es_type == MESA_SHADER_TESS_EVAL) { /* TCS -> TES inputs when pipeline has GS */ stats.inputs += shader->info.tes.num_linked_inputs + shader->info.tes.num_linked_patch_inputs; } } /* VS -> GS or TES -> GS inputs */ stats.inputs += shader->info.gs.num_linked_inputs; break; case MESA_SHADER_FRAGMENT: stats.inputs += shader->info.ps.num_inputs; break; default: /* Other stages don't have IO or we are not interested in them. */ break; } switch (stage) { case MESA_SHADER_VERTEX: if (!shader->info.vs.as_ls && !shader->info.vs.as_es) { /* VS -> FS outputs. */ stats.outputs += shader->info.outinfo.param_exports + shader->info.outinfo.prim_param_exports; } else if (gfx_level <= GFX8) { /* VS -> TCS, VS -> GS outputs on GFX6-8 */ stats.outputs += shader->info.vs.num_linked_outputs; } break; case MESA_SHADER_TESS_CTRL: if (gfx_level >= GFX9) { /* VS -> TCS outputs on GFX9+ */ stats.outputs += shader->info.vs.num_linked_outputs; } /* TCS -> TES outputs */ stats.outputs += shader->info.tcs.io_info.highest_remapped_vram_output + shader->info.tcs.io_info.highest_remapped_vram_patch_output; break; case MESA_SHADER_TESS_EVAL: if (!shader->info.tes.as_es) { /* TES -> FS outputs */ stats.outputs += shader->info.outinfo.param_exports + shader->info.outinfo.prim_param_exports; } else if (gfx_level <= GFX8) { /* TES -> GS outputs on GFX6-8 */ stats.outputs += shader->info.tes.num_linked_outputs; } break; case MESA_SHADER_GEOMETRY: /* The IO stats of the GS copy shader are already reflected by GS and FS, so leave it empty. */ if (shader->info.type == RADV_SHADER_TYPE_GS_COPY) break; if (gfx_level >= GFX9) { if (shader->info.gs.es_type == MESA_SHADER_VERTEX) { /* VS -> GS outputs on GFX9+ */ stats.outputs += shader->info.vs.num_linked_outputs; } else if (shader->info.gs.es_type == MESA_SHADER_TESS_EVAL) { /* TES -> GS outputs on GFX9+ */ stats.outputs += shader->info.tes.num_linked_outputs; } } if (shader->info.is_ngg) { /* GS -> FS outputs (GFX10+ NGG) */ stats.outputs += shader->info.outinfo.param_exports + shader->info.outinfo.prim_param_exports; } else { /* GS -> FS outputs (GFX6-10.3 legacy) */ stats.outputs += DIV_ROUND_UP(((uint32_t)shader->info.gs.num_components_per_stream[0] + (uint32_t)shader->info.gs.num_components_per_stream[1] + (uint32_t)shader->info.gs.num_components_per_stream[2] + (uint32_t)shader->info.gs.num_components_per_stream[3]) * 4, 16); } break; case MESA_SHADER_MESH: /* MS -> FS outputs */ stats.outputs += shader->info.outinfo.param_exports + shader->info.outinfo.prim_param_exports; break; case MESA_SHADER_FRAGMENT: stats.outputs += DIV_ROUND_UP(util_bitcount(shader->info.ps.colors_written), 4) + !!shader->info.ps.writes_z + !!shader->info.ps.writes_stencil + !!shader->info.ps.writes_sample_mask + !!shader->info.ps.writes_mrt0_alpha; break; default: /* Other stages don't have IO or we are not interested in them. */ break; } vk_add_amd_stats(out, &stats); return vk_outarray_status(&out); } static VkResult radv_copy_representation(void *data, size_t *data_size, const char *src) { size_t total_size = strlen(src) + 1; if (!data) { *data_size = total_size; return VK_SUCCESS; } size_t size = MIN2(total_size, *data_size); memcpy(data, src, size); if (size) *((char *)data + size - 1) = 0; return size < total_size ? VK_INCOMPLETE : VK_SUCCESS; } VKAPI_ATTR VkResult VKAPI_CALL radv_GetPipelineExecutableInternalRepresentationsKHR( VkDevice _device, const VkPipelineExecutableInfoKHR *pExecutableInfo, uint32_t *pInternalRepresentationCount, VkPipelineExecutableInternalRepresentationKHR *pInternalRepresentations) { VK_FROM_HANDLE(radv_device, device, _device); VK_FROM_HANDLE(radv_pipeline, pipeline, pExecutableInfo->pipeline); const struct radv_physical_device *pdev = radv_device_physical(device); gl_shader_stage stage; struct radv_shader *shader = radv_get_shader_from_executable_index(pipeline, pExecutableInfo->executableIndex, &stage); VkPipelineExecutableInternalRepresentationKHR *p = pInternalRepresentations; VkPipelineExecutableInternalRepresentationKHR *end = p + (pInternalRepresentations ? *pInternalRepresentationCount : 0); VkResult result = VK_SUCCESS; /* optimized NIR */ if (p < end) { p->isText = true; VK_COPY_STR(p->name, "NIR Shader(s)"); VK_COPY_STR(p->description, "The optimized NIR shader(s)"); if (radv_copy_representation(p->pData, &p->dataSize, shader->nir_string) != VK_SUCCESS) result = VK_INCOMPLETE; } ++p; /* backend IR */ if (p < end) { p->isText = true; if (radv_use_llvm_for_stage(pdev, stage)) { VK_COPY_STR(p->name, "LLVM IR"); VK_COPY_STR(p->description, "The LLVM IR after some optimizations"); } else { VK_COPY_STR(p->name, "ACO IR"); VK_COPY_STR(p->description, "The ACO IR after some optimizations"); } if (radv_copy_representation(p->pData, &p->dataSize, shader->ir_string) != VK_SUCCESS) result = VK_INCOMPLETE; } ++p; /* Disassembler */ if (p < end && shader->disasm_string) { p->isText = true; VK_COPY_STR(p->name, "Assembly"); VK_COPY_STR(p->description, "Final Assembly"); if (radv_copy_representation(p->pData, &p->dataSize, shader->disasm_string) != VK_SUCCESS) result = VK_INCOMPLETE; } ++p; if (!pInternalRepresentations) *pInternalRepresentationCount = p - pInternalRepresentations; else if (p > end) { result = VK_INCOMPLETE; *pInternalRepresentationCount = end - pInternalRepresentations; } else { *pInternalRepresentationCount = p - pInternalRepresentations; } return result; } static void vk_shader_module_finish(void *_module) { struct vk_shader_module *module = _module; vk_object_base_finish(&module->base); } VkPipelineShaderStageCreateInfo * radv_copy_shader_stage_create_info(struct radv_device *device, uint32_t stageCount, const VkPipelineShaderStageCreateInfo *pStages, void *mem_ctx) { VkPipelineShaderStageCreateInfo *new_stages; size_t size = sizeof(VkPipelineShaderStageCreateInfo) * stageCount; new_stages = ralloc_size(mem_ctx, size); if (!new_stages) return NULL; if (size) memcpy(new_stages, pStages, size); for (uint32_t i = 0; i < stageCount; i++) { VK_FROM_HANDLE(vk_shader_module, module, new_stages[i].module); const VkShaderModuleCreateInfo *minfo = vk_find_struct_const(pStages[i].pNext, SHADER_MODULE_CREATE_INFO); if (module) { struct vk_shader_module *new_module = ralloc_size(mem_ctx, sizeof(struct vk_shader_module) + module->size); if (!new_module) return NULL; ralloc_set_destructor(new_module, vk_shader_module_finish); vk_object_base_init(&device->vk, &new_module->base, VK_OBJECT_TYPE_SHADER_MODULE); new_module->nir = NULL; memcpy(new_module->hash, module->hash, sizeof(module->hash)); new_module->size = module->size; memcpy(new_module->data, module->data, module->size); module = new_module; } else if (minfo) { module = ralloc_size(mem_ctx, sizeof(struct vk_shader_module) + minfo->codeSize); if (!module) return NULL; vk_shader_module_init(&device->vk, module, minfo); } if (module) { const VkSpecializationInfo *spec = new_stages[i].pSpecializationInfo; if (spec) { VkSpecializationInfo *new_spec = ralloc(mem_ctx, VkSpecializationInfo); if (!new_spec) return NULL; new_spec->mapEntryCount = spec->mapEntryCount; uint32_t map_entries_size = sizeof(VkSpecializationMapEntry) * spec->mapEntryCount; new_spec->pMapEntries = ralloc_size(mem_ctx, map_entries_size); if (!new_spec->pMapEntries) return NULL; memcpy((void *)new_spec->pMapEntries, spec->pMapEntries, map_entries_size); new_spec->dataSize = spec->dataSize; new_spec->pData = ralloc_size(mem_ctx, spec->dataSize); if (!new_spec->pData) return NULL; memcpy((void *)new_spec->pData, spec->pData, spec->dataSize); new_stages[i].pSpecializationInfo = new_spec; } new_stages[i].module = vk_shader_module_to_handle(module); new_stages[i].pName = ralloc_strdup(mem_ctx, new_stages[i].pName); if (!new_stages[i].pName) return NULL; new_stages[i].pNext = NULL; } } return new_stages; } void radv_pipeline_hash(const struct radv_device *device, const struct radv_pipeline_layout *pipeline_layout, struct mesa_sha1 *ctx) { _mesa_sha1_update(ctx, device->cache_hash, sizeof(device->cache_hash)); if (pipeline_layout) _mesa_sha1_update(ctx, pipeline_layout->hash, sizeof(pipeline_layout->hash)); } void radv_pipeline_hash_shader_stage(VkPipelineCreateFlags2 pipeline_flags, const VkPipelineShaderStageCreateInfo *sinfo, const struct radv_shader_stage_key *stage_key, struct mesa_sha1 *ctx) { unsigned char shader_sha1[SHA1_DIGEST_LENGTH]; vk_pipeline_hash_shader_stage(pipeline_flags, sinfo, NULL, shader_sha1); _mesa_sha1_update(ctx, shader_sha1, sizeof(shader_sha1)); _mesa_sha1_update(ctx, stage_key, sizeof(*stage_key)); } static void radv_print_pso_history(const struct radv_pipeline *pipeline, const struct radv_shader *shader, FILE *output) { const uint64_t start_addr = radv_shader_get_va(shader) & ((1ull << 48) - 1); const uint64_t end_addr = start_addr + shader->code_size; fprintf(output, "pipeline_hash=%.16llx, VA=%.16llx-%.16llx, stage=%s\n", (long long)pipeline->pipeline_hash, (long long)start_addr, (long long)end_addr, _mesa_shader_stage_to_string(shader->info.stage)); fflush(output); } void radv_pipeline_report_pso_history(const struct radv_device *device, struct radv_pipeline *pipeline) { const struct radv_physical_device *pdev = radv_device_physical(device); const struct radv_instance *instance = radv_physical_device_instance(pdev); FILE *output = instance->pso_history_logfile ? instance->pso_history_logfile : stderr; if (!(instance->debug_flags & RADV_DEBUG_PSO_HISTORY)) return; /* Only report PSO history for application pipelines. */ if (pipeline->is_internal) return; switch (pipeline->type) { case RADV_PIPELINE_GRAPHICS: for (uint32_t i = 0; i < MESA_VULKAN_SHADER_STAGES; i++) { const struct radv_shader *shader = pipeline->shaders[i]; if (shader) radv_print_pso_history(pipeline, shader, output); } if (pipeline->gs_copy_shader) radv_print_pso_history(pipeline, pipeline->gs_copy_shader, output); break; case RADV_PIPELINE_COMPUTE: radv_print_pso_history(pipeline, pipeline->shaders[MESA_SHADER_COMPUTE], output); break; case RADV_PIPELINE_RAY_TRACING: { struct radv_ray_tracing_pipeline *rt_pipeline = radv_pipeline_to_ray_tracing(pipeline); radv_print_pso_history(pipeline, rt_pipeline->prolog, output); for (uint32_t i = 0; i < rt_pipeline->stage_count; i++) { const struct radv_shader *shader = rt_pipeline->stages[i].shader; if (shader) radv_print_pso_history(pipeline, shader, output); } break; } default: break; } }