diff options
Diffstat (limited to 'src/amd/compiler/aco_instruction_selection_setup.cpp')
-rw-r--r-- | src/amd/compiler/aco_instruction_selection_setup.cpp | 731 |
1 files changed, 184 insertions, 547 deletions
diff --git a/src/amd/compiler/aco_instruction_selection_setup.cpp b/src/amd/compiler/aco_instruction_selection_setup.cpp index 767e6fbf124..ba45464d063 100644 --- a/src/amd/compiler/aco_instruction_selection_setup.cpp +++ b/src/amd/compiler/aco_instruction_selection_setup.cpp @@ -1,32 +1,13 @@ /* * Copyright © 2018 Valve 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. - * + * SPDX-License-Identifier: MIT */ #include "aco_instruction_selection.h" -#include "common/ac_exp_param.h" +#include "common/ac_nir.h" #include "common/sid.h" -#include "vulkan/radv_descriptor_set.h" #include "nir_control_flow.h" @@ -36,80 +17,23 @@ namespace aco { namespace { -unsigned -get_interp_input(nir_intrinsic_op intrin, enum glsl_interp_mode interp) -{ - switch (interp) { - case INTERP_MODE_SMOOTH: - case INTERP_MODE_NONE: - if (intrin == nir_intrinsic_load_barycentric_pixel || - intrin == nir_intrinsic_load_barycentric_at_sample || - intrin == nir_intrinsic_load_barycentric_at_offset) - return S_0286CC_PERSP_CENTER_ENA(1); - else if (intrin == nir_intrinsic_load_barycentric_centroid) - return S_0286CC_PERSP_CENTROID_ENA(1); - else if (intrin == nir_intrinsic_load_barycentric_sample) - return S_0286CC_PERSP_SAMPLE_ENA(1); - break; - case INTERP_MODE_NOPERSPECTIVE: - if (intrin == nir_intrinsic_load_barycentric_pixel) - return S_0286CC_LINEAR_CENTER_ENA(1); - else if (intrin == nir_intrinsic_load_barycentric_centroid) - return S_0286CC_LINEAR_CENTROID_ENA(1); - else if (intrin == nir_intrinsic_load_barycentric_sample) - return S_0286CC_LINEAR_SAMPLE_ENA(1); - break; - default: break; - } - return 0; -} - -bool -is_loop_header_block(nir_block* block) -{ - return block->cf_node.parent->type == nir_cf_node_loop && - block == nir_loop_first_block(nir_cf_node_as_loop(block->cf_node.parent)); -} - -/* similar to nir_block_is_unreachable(), but does not require dominance information */ -bool -is_block_reachable(nir_function_impl* impl, nir_block* known_reachable, nir_block* block) -{ - if (block == nir_start_block(impl) || block == known_reachable) - return true; - - /* skip loop back-edges */ - if (is_loop_header_block(block)) { - nir_loop* loop = nir_cf_node_as_loop(block->cf_node.parent); - nir_block* preheader = nir_block_cf_tree_prev(nir_loop_first_block(loop)); - return is_block_reachable(impl, known_reachable, preheader); - } - - set_foreach (block->predecessors, entry) { - if (is_block_reachable(impl, known_reachable, (nir_block*)entry->key)) - return true; - } - - return false; -} - /* Check whether the given SSA def is only used by cross-lane instructions. */ bool -only_used_by_cross_lane_instrs(nir_ssa_def* ssa, bool follow_phis = true) +only_used_by_cross_lane_instrs(nir_def* ssa, bool follow_phis = true) { nir_foreach_use (src, ssa) { - switch (src->parent_instr->type) { + switch (nir_src_parent_instr(src)->type) { case nir_instr_type_alu: { - nir_alu_instr* alu = nir_instr_as_alu(src->parent_instr); + nir_alu_instr* alu = nir_instr_as_alu(nir_src_parent_instr(src)); if (alu->op != nir_op_unpack_64_2x32_split_x && alu->op != nir_op_unpack_64_2x32_split_y) return false; - if (!only_used_by_cross_lane_instrs(&alu->dest.dest.ssa, follow_phis)) + if (!only_used_by_cross_lane_instrs(&alu->def, follow_phis)) return false; continue; } case nir_instr_type_intrinsic: { - nir_intrinsic_instr* intrin = nir_instr_as_intrinsic(src->parent_instr); + nir_intrinsic_instr* intrin = nir_instr_as_intrinsic(nir_src_parent_instr(src)); if (intrin->intrinsic != nir_intrinsic_read_invocation && intrin->intrinsic != nir_intrinsic_read_first_invocation && intrin->intrinsic != nir_intrinsic_lane_permute_16_amd) @@ -122,8 +46,8 @@ only_used_by_cross_lane_instrs(nir_ssa_def* ssa, bool follow_phis = true) if (!follow_phis) return false; - nir_phi_instr* phi = nir_instr_as_phi(src->parent_instr); - if (!only_used_by_cross_lane_instrs(&phi->dest.ssa, false)) + nir_phi_instr* phi = nir_instr_as_phi(nir_src_parent_instr(src)); + if (!only_used_by_cross_lane_instrs(&phi->def, false)) return false; continue; @@ -138,25 +62,24 @@ only_used_by_cross_lane_instrs(nir_ssa_def* ssa, bool follow_phis = true) /* If one side of a divergent IF ends in a branch and the other doesn't, we * might have to emit the contents of the side without the branch at the merge * block instead. This is so that we can use any SGPR live-out of the side - * without the branch without creating a linear phi in the invert or merge block. */ + * without the branch without creating a linear phi in the invert or merge block. + * + * This also removes any unreachable merge blocks. + */ bool sanitize_if(nir_function_impl* impl, nir_if* nif) { - // TODO: skip this if the condition is uniform and there are no divergent breaks/continues? - nir_block* then_block = nir_if_last_then_block(nif); nir_block* else_block = nir_if_last_else_block(nif); - bool then_jump = nir_block_ends_in_jump(then_block) || - !is_block_reachable(impl, nir_if_first_then_block(nif), then_block); - bool else_jump = nir_block_ends_in_jump(else_block) || - !is_block_reachable(impl, nir_if_first_else_block(nif), else_block); - if (then_jump == else_jump) + bool then_jump = nir_block_ends_in_jump(then_block); + bool else_jump = nir_block_ends_in_jump(else_block); + if (!then_jump && !else_jump) return false; /* If the continue from block is empty then return as there is nothing to * move. */ - if (nir_cf_list_is_empty_block(else_jump ? &nif->then_list : &nif->else_list)) + if (nir_cf_list_is_empty_block(then_jump ? &nif->else_list : &nif->then_list)) return false; /* Even though this if statement has a jump on one side, we may still have @@ -167,9 +90,13 @@ sanitize_if(nir_function_impl* impl, nir_if* nif) nir_opt_remove_phis_block(nir_cf_node_as_block(nir_cf_node_next(&nif->cf_node))); /* Finally, move the continue from branch after the if-statement. */ - nir_block* last_continue_from_blk = else_jump ? then_block : else_block; + nir_block* last_continue_from_blk = then_jump ? else_block : then_block; nir_block* first_continue_from_blk = - else_jump ? nir_if_first_then_block(nif) : nir_if_first_else_block(nif); + then_jump ? nir_if_first_else_block(nif) : nir_if_first_then_block(nif); + + /* We don't need to repair SSA. nir_remove_after_cf_node() replaces any uses with undef. */ + if (then_jump && else_jump) + nir_remove_after_cf_node(&nif->cf_node); nir_cf_list tmp; nir_cf_extract(&tmp, nir_before_block(first_continue_from_blk), @@ -195,7 +122,26 @@ sanitize_cf_list(nir_function_impl* impl, struct exec_list* cf_list) } case nir_cf_node_loop: { nir_loop* loop = nir_cf_node_as_loop(cf_node); + assert(!nir_loop_has_continue_construct(loop)); progress |= sanitize_cf_list(impl, &loop->body); + + /* NIR seems to allow this, and even though the loop exit has no predecessors, SSA defs from the + * loop header are live. Handle this without complicating the ACO IR by creating a dummy break. + */ + if (nir_cf_node_cf_tree_next(&loop->cf_node)->predecessors->entries == 0) { + nir_builder b = nir_builder_create(impl); + b.cursor = nir_after_block_before_jump(nir_loop_last_block(loop)); + + nir_def *cond = nir_imm_false(&b); + /* We don't use block divergence information, so just this is enough. */ + cond->divergent = false; + + nir_push_if(&b, cond); + nir_jump(&b, nir_jump_break); + nir_pop_if(&b, NULL); + + progress = true; + } break; } case nir_cf_node_function: unreachable("Invalid cf type"); @@ -206,13 +152,13 @@ sanitize_cf_list(nir_function_impl* impl, struct exec_list* cf_list) } void -apply_nuw_to_ssa(isel_context* ctx, nir_ssa_def* ssa) +apply_nuw_to_ssa(isel_context* ctx, nir_def* ssa) { - nir_ssa_scalar scalar; + nir_scalar scalar; scalar.def = ssa; scalar.comp = 0; - if (!nir_ssa_scalar_is_alu(scalar) || nir_ssa_scalar_alu_op(scalar) != nir_op_iadd) + if (!nir_scalar_is_alu(scalar) || nir_scalar_alu_op(scalar) != nir_op_iadd) return; nir_alu_instr* add = nir_instr_as_alu(ssa->parent_instr); @@ -220,11 +166,11 @@ apply_nuw_to_ssa(isel_context* ctx, nir_ssa_def* ssa) if (add->no_unsigned_wrap) return; - nir_ssa_scalar src0 = nir_ssa_scalar_chase_alu_src(scalar, 0); - nir_ssa_scalar src1 = nir_ssa_scalar_chase_alu_src(scalar, 1); + nir_scalar src0 = nir_scalar_chase_alu_src(scalar, 0); + nir_scalar src1 = nir_scalar_chase_alu_src(scalar, 1); - if (nir_ssa_scalar_is_const(src0)) { - nir_ssa_scalar tmp = src0; + if (nir_scalar_is_const(src0)) { + nir_scalar tmp = src0; src0 = src1; src1 = tmp; } @@ -259,6 +205,9 @@ apply_nuw_to_offsets(isel_context* ctx, nir_function_impl* impl) if (!nir_src_is_divergent(intrin->src[2])) apply_nuw_to_ssa(ctx, intrin->src[2].ssa); break; + case nir_intrinsic_load_scratch: apply_nuw_to_ssa(ctx, intrin->src[0].ssa); break; + case nir_intrinsic_store_scratch: + case nir_intrinsic_load_smem_amd: apply_nuw_to_ssa(ctx, intrin->src[1].ssa); break; default: break; } } @@ -275,168 +224,23 @@ get_reg_class(isel_context* ctx, RegType type, unsigned components, unsigned bit } void -setup_vs_output_info(isel_context* ctx, nir_shader* nir, bool export_prim_id, - bool export_clip_dists, radv_vs_output_info* outinfo) -{ - memset(outinfo->vs_output_param_offset, AC_EXP_PARAM_UNDEFINED, - sizeof(outinfo->vs_output_param_offset)); - - outinfo->param_exports = 0; - int pos_written = 0x1; - bool writes_primitive_shading_rate = - outinfo->writes_primitive_shading_rate || ctx->options->force_vrs_rates; - if (outinfo->writes_pointsize || outinfo->writes_viewport_index || outinfo->writes_layer || - writes_primitive_shading_rate) - pos_written |= 1 << 1; - - uint64_t mask = nir->info.outputs_written; - while (mask) { - int idx = u_bit_scan64(&mask); - if (idx >= VARYING_SLOT_VAR0 || idx == VARYING_SLOT_LAYER || - idx == VARYING_SLOT_PRIMITIVE_ID || idx == VARYING_SLOT_VIEWPORT || - ((idx == VARYING_SLOT_CLIP_DIST0 || idx == VARYING_SLOT_CLIP_DIST1) && - export_clip_dists)) { - if (outinfo->vs_output_param_offset[idx] == AC_EXP_PARAM_UNDEFINED) - outinfo->vs_output_param_offset[idx] = outinfo->param_exports++; - } - } - if (outinfo->writes_layer && - outinfo->vs_output_param_offset[VARYING_SLOT_LAYER] == AC_EXP_PARAM_UNDEFINED) { - /* when ctx->options->key.has_multiview_view_index = true, the layer - * variable isn't declared in NIR and it's isel's job to get the layer */ - outinfo->vs_output_param_offset[VARYING_SLOT_LAYER] = outinfo->param_exports++; - } - - if (export_prim_id) { - assert(outinfo->vs_output_param_offset[VARYING_SLOT_PRIMITIVE_ID] == AC_EXP_PARAM_UNDEFINED); - outinfo->vs_output_param_offset[VARYING_SLOT_PRIMITIVE_ID] = outinfo->param_exports++; - } - - ctx->export_clip_dists = export_clip_dists; - ctx->num_clip_distances = util_bitcount(outinfo->clip_dist_mask); - ctx->num_cull_distances = util_bitcount(outinfo->cull_dist_mask); - - assert(ctx->num_clip_distances + ctx->num_cull_distances <= 8); - - if (ctx->num_clip_distances + ctx->num_cull_distances > 0) - pos_written |= 1 << 2; - if (ctx->num_clip_distances + ctx->num_cull_distances > 4) - pos_written |= 1 << 3; - - outinfo->pos_exports = util_bitcount(pos_written); - - /* GFX10+ early rasterization: - * When there are no param exports in an NGG (or legacy VS) shader, - * RADV sets NO_PC_EXPORT=1, which means the HW will start clipping and rasterization - * as soon as it encounters a DONE pos export. When this happens, PS waves can launch - * before the NGG (or VS) waves finish. - */ - ctx->program->early_rast = ctx->program->chip_class >= GFX10 && outinfo->param_exports == 0; -} - -void -setup_vs_variables(isel_context* ctx, nir_shader* nir) -{ - if (ctx->stage == vertex_vs || ctx->stage == vertex_ngg) { - radv_vs_output_info* outinfo = &ctx->program->info->vs.outinfo; - setup_vs_output_info(ctx, nir, outinfo->export_prim_id, - ctx->options->key.vs_common_out.export_clip_dists, outinfo); - - /* TODO: NGG streamout */ - if (ctx->stage.hw == HWStage::NGG) - assert(!ctx->args->shader_info->so.num_outputs); - } - - if (ctx->stage == vertex_ngg) { - ctx->program->config->lds_size = - DIV_ROUND_UP(nir->info.shared_size, ctx->program->dev.lds_encoding_granule); - assert((ctx->program->config->lds_size * ctx->program->dev.lds_encoding_granule) < - (32 * 1024)); - } -} - -void -setup_gs_variables(isel_context* ctx, nir_shader* nir) -{ - if (ctx->stage == vertex_geometry_gs || ctx->stage == tess_eval_geometry_gs) { - ctx->program->config->lds_size = - ctx->program->info->gs_ring_info.lds_size; /* Already in units of the alloc granularity */ - } else if (ctx->stage == vertex_geometry_ngg || ctx->stage == tess_eval_geometry_ngg) { - radv_vs_output_info* outinfo = &ctx->program->info->vs.outinfo; - setup_vs_output_info(ctx, nir, false, ctx->options->key.vs_common_out.export_clip_dists, - outinfo); - - ctx->program->config->lds_size = - DIV_ROUND_UP(nir->info.shared_size, ctx->program->dev.lds_encoding_granule); - } - - if (ctx->stage.has(SWStage::VS)) - ctx->program->info->gs.es_type = MESA_SHADER_VERTEX; - else if (ctx->stage.has(SWStage::TES)) - ctx->program->info->gs.es_type = MESA_SHADER_TESS_EVAL; -} - -void -setup_tcs_info(isel_context* ctx, nir_shader* nir, nir_shader* vs) -{ - ctx->tcs_in_out_eq = ctx->args->shader_info->vs.tcs_in_out_eq; - ctx->tcs_temp_only_inputs = ctx->args->shader_info->vs.tcs_temp_only_input_mask; - ctx->tcs_num_patches = ctx->args->shader_info->num_tess_patches; - ctx->program->config->lds_size = ctx->args->shader_info->tcs.num_lds_blocks; -} - -void -setup_tes_variables(isel_context* ctx, nir_shader* nir) +setup_tcs_info(isel_context* ctx) { - ctx->tcs_num_patches = ctx->args->shader_info->num_tess_patches; - - if (ctx->stage == tess_eval_vs || ctx->stage == tess_eval_ngg) { - radv_vs_output_info* outinfo = &ctx->program->info->tes.outinfo; - setup_vs_output_info(ctx, nir, outinfo->export_prim_id, - ctx->options->key.vs_common_out.export_clip_dists, outinfo); - - /* TODO: NGG streamout */ - if (ctx->stage.hw == HWStage::NGG) - assert(!ctx->args->shader_info->so.num_outputs); - } - - if (ctx->stage == tess_eval_ngg) { - ctx->program->config->lds_size = - DIV_ROUND_UP(nir->info.shared_size, ctx->program->dev.lds_encoding_granule); - assert((ctx->program->config->lds_size * ctx->program->dev.lds_encoding_granule) < - (32 * 1024)); - } + ctx->tcs_in_out_eq = ctx->program->info.vs.tcs_in_out_eq; + ctx->tcs_temp_only_inputs = ctx->program->info.vs.tcs_temp_only_input_mask; } void -setup_variables(isel_context* ctx, nir_shader* nir) +setup_lds_size(isel_context* ctx, nir_shader* nir) { - switch (nir->info.stage) { - case MESA_SHADER_FRAGMENT: { - break; - } - case MESA_SHADER_COMPUTE: { + /* TCS and GFX9 GS are special cases, already in units of the allocation granule. */ + if (ctx->stage.has(SWStage::TCS)) + ctx->program->config->lds_size = ctx->program->info.tcs.num_lds_blocks; + else if (ctx->stage.hw == AC_HW_LEGACY_GEOMETRY_SHADER && ctx->options->gfx_level >= GFX9) + ctx->program->config->lds_size = ctx->program->info.gfx9_gs_ring_lds_size; + else ctx->program->config->lds_size = DIV_ROUND_UP(nir->info.shared_size, ctx->program->dev.lds_encoding_granule); - break; - } - case MESA_SHADER_VERTEX: { - setup_vs_variables(ctx, nir); - break; - } - case MESA_SHADER_GEOMETRY: { - setup_gs_variables(ctx, nir); - break; - } - case MESA_SHADER_TESS_CTRL: { - break; - } - case MESA_SHADER_TESS_EVAL: { - setup_tes_variables(ctx, nir); - break; - } - default: unreachable("Unhandled shader stage."); - } /* Make sure we fit the available LDS space. */ assert((ctx->program->config->lds_size * ctx->program->dev.lds_encoding_granule) <= @@ -446,9 +250,6 @@ setup_variables(isel_context* ctx, nir_shader* nir) void setup_nir(isel_context* ctx, nir_shader* nir) { - /* the variable setup has to be done before lower_io / CSE */ - setup_variables(ctx, nir); - nir_convert_to_lcssa(nir, true, false); nir_lower_phis_to_scalar(nir, true); @@ -466,12 +267,8 @@ init_context(isel_context* ctx, nir_shader* shader) /* Init NIR range analysis. */ ctx->range_ht = _mesa_pointer_hash_table_create(NULL); - ctx->ub_config.min_subgroup_size = 64; - ctx->ub_config.max_subgroup_size = 64; - if (ctx->shader->info.stage == MESA_SHADER_COMPUTE && ctx->options->key.cs.subgroup_size) { - ctx->ub_config.min_subgroup_size = ctx->options->key.cs.subgroup_size; - ctx->ub_config.max_subgroup_size = ctx->options->key.cs.subgroup_size; - } + ctx->ub_config.min_subgroup_size = ctx->program->wave_size; + ctx->ub_config.max_subgroup_size = ctx->program->wave_size; ctx->ub_config.max_workgroup_invocations = 2048; ctx->ub_config.max_workgroup_count[0] = 65535; ctx->ub_config.max_workgroup_count[1] = 65535; @@ -479,38 +276,10 @@ init_context(isel_context* ctx, nir_shader* shader) ctx->ub_config.max_workgroup_size[0] = 2048; ctx->ub_config.max_workgroup_size[1] = 2048; ctx->ub_config.max_workgroup_size[2] = 2048; - for (unsigned i = 0; i < MAX_VERTEX_ATTRIBS; i++) { - unsigned attrib_format = ctx->options->key.vs.vertex_attribute_formats[i]; - unsigned dfmt = attrib_format & 0xf; - unsigned nfmt = (attrib_format >> 4) & 0x7; - - uint32_t max = UINT32_MAX; - if (nfmt == V_008F0C_BUF_NUM_FORMAT_UNORM) { - max = 0x3f800000u; - } else if (nfmt == V_008F0C_BUF_NUM_FORMAT_UINT || nfmt == V_008F0C_BUF_NUM_FORMAT_USCALED) { - bool uscaled = nfmt == V_008F0C_BUF_NUM_FORMAT_USCALED; - switch (dfmt) { - case V_008F0C_BUF_DATA_FORMAT_8: - case V_008F0C_BUF_DATA_FORMAT_8_8: - case V_008F0C_BUF_DATA_FORMAT_8_8_8_8: max = uscaled ? 0x437f0000u : UINT8_MAX; break; - case V_008F0C_BUF_DATA_FORMAT_10_10_10_2: - case V_008F0C_BUF_DATA_FORMAT_2_10_10_10: max = uscaled ? 0x447fc000u : 1023; break; - case V_008F0C_BUF_DATA_FORMAT_10_11_11: - case V_008F0C_BUF_DATA_FORMAT_11_11_10: max = uscaled ? 0x44ffe000u : 2047; break; - case V_008F0C_BUF_DATA_FORMAT_16: - case V_008F0C_BUF_DATA_FORMAT_16_16: - case V_008F0C_BUF_DATA_FORMAT_16_16_16_16: max = uscaled ? 0x477fff00u : UINT16_MAX; break; - case V_008F0C_BUF_DATA_FORMAT_32: - case V_008F0C_BUF_DATA_FORMAT_32_32: - case V_008F0C_BUF_DATA_FORMAT_32_32_32: - case V_008F0C_BUF_DATA_FORMAT_32_32_32_32: max = uscaled ? 0x4f800000u : UINT32_MAX; break; - } - } - ctx->ub_config.vertex_attrib_max[i] = max; - } nir_divergence_analysis(shader); - nir_opt_uniform_atomics(shader); + if (nir_opt_uniform_atomics(shader) && nir_lower_int64(shader)) + nir_divergence_analysis(shader); apply_nuw_to_offsets(ctx, impl); @@ -521,7 +290,7 @@ init_context(isel_context* ctx, nir_shader* shader) /* we'll need these for isel */ nir_metadata_require(impl, nir_metadata_block_index); - if (!ctx->stage.has(SWStage::GSCopy) && ctx->options->dump_preoptir) { + if (ctx->options->dump_preoptir) { fprintf(stderr, "NIR shader before instruction selection:\n"); nir_print_shader(shader, stderr); } @@ -530,8 +299,6 @@ init_context(isel_context* ctx, nir_shader* shader) ctx->program->allocateRange(impl->ssa_alloc); RegClass* regclasses = ctx->program->temp_rc.data() + ctx->first_temp_id; - unsigned spi_ps_inputs = 0; - std::unique_ptr<unsigned[]> nir_to_aco{new unsigned[impl->num_blocks]()}; /* TODO: make this recursive to improve compile times */ @@ -543,12 +310,14 @@ init_context(isel_context* ctx, nir_shader* shader) switch (instr->type) { case nir_instr_type_alu: { nir_alu_instr* alu_instr = nir_instr_as_alu(instr); - RegType type = - nir_dest_is_divergent(alu_instr->dest.dest) ? RegType::vgpr : RegType::sgpr; + RegType type = alu_instr->def.divergent ? RegType::vgpr : RegType::sgpr; switch (alu_instr->op) { case nir_op_fmul: + case nir_op_fmulz: case nir_op_fadd: case nir_op_fsub: + case nir_op_ffma: + case nir_op_ffmaz: case nir_op_fmax: case nir_op_fmin: case nir_op_fneg: @@ -565,8 +334,8 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_op_fceil: case nir_op_ftrunc: case nir_op_fround_even: - case nir_op_fsin: - case nir_op_fcos: + case nir_op_fsin_amd: + case nir_op_fcos_amd: case nir_op_f2f16: case nir_op_f2f16_rtz: case nir_op_f2f16_rtne: @@ -578,7 +347,12 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_op_i2f16: case nir_op_i2f32: case nir_op_i2f64: + case nir_op_pack_half_2x16_rtz_split: case nir_op_pack_half_2x16_split: + case nir_op_pack_unorm_2x16: + case nir_op_pack_snorm_2x16: + case nir_op_pack_uint_2x16: + case nir_op_pack_sint_2x16: case nir_op_unpack_half_2x16_split_x: case nir_op_unpack_half_2x16_split_y: case nir_op_fddx: @@ -591,14 +365,15 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_op_ldexp: case nir_op_frexp_sig: case nir_op_frexp_exp: - case nir_op_cube_face_index_amd: - case nir_op_cube_face_coord_amd: - case nir_op_sad_u8x4: - case nir_op_iadd_sat: + case nir_op_cube_amd: + case nir_op_msad_4x8: + case nir_op_mqsad_4x8: case nir_op_udot_4x8_uadd: case nir_op_sdot_4x8_iadd: + case nir_op_sudot_4x8_iadd: case nir_op_udot_4x8_uadd_sat: case nir_op_sdot_4x8_iadd_sat: + case nir_op_sudot_4x8_iadd_sat: case nir_op_udot_2x16_uadd: case nir_op_sdot_2x16_iadd: case nir_op_udot_2x16_uadd_sat: @@ -607,18 +382,20 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_op_f2u16: case nir_op_f2i32: case nir_op_f2u32: - case nir_op_f2i64: - case nir_op_f2u64: case nir_op_b2i8: case nir_op_b2i16: case nir_op_b2i32: - case nir_op_b2i64: case nir_op_b2b32: case nir_op_b2f16: case nir_op_b2f32: case nir_op_mov: break; + case nir_op_iabs: case nir_op_iadd: + case nir_op_iadd_sat: + case nir_op_uadd_sat: case nir_op_isub: + case nir_op_isub_sat: + case nir_op_usub_sat: case nir_op_imul: case nir_op_imin: case nir_op_imax: @@ -628,7 +405,7 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_op_ishr: case nir_op_ushr: /* packed 16bit instructions have to be VGPR */ - type = alu_instr->dest.dest.ssa.num_components == 2 ? RegType::vgpr : type; + type = alu_instr->def.num_components == 2 ? RegType::vgpr : type; FALLTHROUGH; default: for (unsigned i = 0; i < nir_op_infos[alu_instr->op].num_inputs; i++) { @@ -638,9 +415,9 @@ init_context(isel_context* ctx, nir_shader* shader) break; } - RegClass rc = get_reg_class(ctx, type, alu_instr->dest.dest.ssa.num_components, - alu_instr->dest.dest.ssa.bit_size); - regclasses[alu_instr->dest.dest.ssa.index] = rc; + RegClass rc = + get_reg_class(ctx, type, alu_instr->def.num_components, alu_instr->def.bit_size); + regclasses[alu_instr->def.index] = rc; break; } case nir_instr_type_load_const: { @@ -654,12 +431,20 @@ init_context(isel_context* ctx, nir_shader* shader) nir_intrinsic_instr* intrinsic = nir_instr_as_intrinsic(instr); if (!nir_intrinsic_infos[intrinsic->intrinsic].has_dest) break; + if (intrinsic->intrinsic == nir_intrinsic_strict_wqm_coord_amd) { + regclasses[intrinsic->def.index] = + RegClass::get(RegType::vgpr, intrinsic->def.num_components * 4 + + nir_intrinsic_base(intrinsic)) + .as_linear(); + break; + } RegType type = RegType::sgpr; switch (intrinsic->intrinsic) { case nir_intrinsic_load_push_constant: case nir_intrinsic_load_workgroup_id: case nir_intrinsic_load_num_workgroups: case nir_intrinsic_load_ray_launch_size: + case nir_intrinsic_load_sbt_base_amd: case nir_intrinsic_load_subgroup_id: case nir_intrinsic_load_num_subgroups: case nir_intrinsic_load_first_vertex: @@ -667,126 +452,71 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_intrinsic_vote_all: case nir_intrinsic_vote_any: case nir_intrinsic_read_first_invocation: + case nir_intrinsic_as_uniform: case nir_intrinsic_read_invocation: case nir_intrinsic_first_invocation: case nir_intrinsic_ballot: - case nir_intrinsic_load_ring_tess_factors_amd: - case nir_intrinsic_load_ring_tess_factors_offset_amd: - case nir_intrinsic_load_ring_tess_offchip_amd: - case nir_intrinsic_load_ring_tess_offchip_offset_amd: - case nir_intrinsic_load_ring_esgs_amd: - case nir_intrinsic_load_ring_es2gs_offset_amd: - case nir_intrinsic_image_deref_samples: - case nir_intrinsic_has_input_vertex_amd: - case nir_intrinsic_has_input_primitive_amd: - case nir_intrinsic_load_workgroup_num_input_vertices_amd: - case nir_intrinsic_load_workgroup_num_input_primitives_amd: - case nir_intrinsic_load_shader_query_enabled_amd: - case nir_intrinsic_load_cull_front_face_enabled_amd: - case nir_intrinsic_load_cull_back_face_enabled_amd: - case nir_intrinsic_load_cull_ccw_amd: - case nir_intrinsic_load_cull_small_primitives_enabled_amd: - case nir_intrinsic_load_cull_any_enabled_amd: - case nir_intrinsic_load_viewport_x_scale: - case nir_intrinsic_load_viewport_y_scale: - case nir_intrinsic_load_viewport_x_offset: - case nir_intrinsic_load_viewport_y_offset: type = RegType::sgpr; break; + case nir_intrinsic_ballot_relaxed: + case nir_intrinsic_bindless_image_samples: + case nir_intrinsic_load_scalar_arg_amd: + case nir_intrinsic_load_lds_ngg_scratch_base_amd: + case nir_intrinsic_load_lds_ngg_gs_out_vertex_base_amd: + case nir_intrinsic_load_smem_amd: + case nir_intrinsic_unit_test_uniform_amd: type = RegType::sgpr; break; case nir_intrinsic_load_sample_id: - case nir_intrinsic_load_sample_mask_in: case nir_intrinsic_load_input: case nir_intrinsic_load_output: case nir_intrinsic_load_input_vertex: case nir_intrinsic_load_per_vertex_input: case nir_intrinsic_load_per_vertex_output: - case nir_intrinsic_load_vertex_id: case nir_intrinsic_load_vertex_id_zero_base: case nir_intrinsic_load_barycentric_sample: case nir_intrinsic_load_barycentric_pixel: case nir_intrinsic_load_barycentric_model: case nir_intrinsic_load_barycentric_centroid: - case nir_intrinsic_load_barycentric_at_sample: case nir_intrinsic_load_barycentric_at_offset: case nir_intrinsic_load_interpolated_input: case nir_intrinsic_load_frag_coord: case nir_intrinsic_load_frag_shading_rate: case nir_intrinsic_load_sample_pos: - case nir_intrinsic_load_layer_id: case nir_intrinsic_load_local_invocation_id: case nir_intrinsic_load_local_invocation_index: case nir_intrinsic_load_subgroup_invocation: + case nir_intrinsic_load_ray_launch_id: case nir_intrinsic_load_tess_coord: case nir_intrinsic_write_invocation_amd: case nir_intrinsic_mbcnt_amd: - case nir_intrinsic_byte_permute_amd: case nir_intrinsic_lane_permute_16_amd: case nir_intrinsic_load_instance_id: - case nir_intrinsic_ssbo_atomic_add: - case nir_intrinsic_ssbo_atomic_imin: - case nir_intrinsic_ssbo_atomic_umin: - case nir_intrinsic_ssbo_atomic_imax: - case nir_intrinsic_ssbo_atomic_umax: - case nir_intrinsic_ssbo_atomic_and: - case nir_intrinsic_ssbo_atomic_or: - case nir_intrinsic_ssbo_atomic_xor: - case nir_intrinsic_ssbo_atomic_exchange: - case nir_intrinsic_ssbo_atomic_comp_swap: - case nir_intrinsic_ssbo_atomic_fmin: - case nir_intrinsic_ssbo_atomic_fmax: - case nir_intrinsic_global_atomic_add: - case nir_intrinsic_global_atomic_imin: - case nir_intrinsic_global_atomic_umin: - case nir_intrinsic_global_atomic_imax: - case nir_intrinsic_global_atomic_umax: - case nir_intrinsic_global_atomic_and: - case nir_intrinsic_global_atomic_or: - case nir_intrinsic_global_atomic_xor: - case nir_intrinsic_global_atomic_exchange: - case nir_intrinsic_global_atomic_comp_swap: - case nir_intrinsic_global_atomic_fmin: - case nir_intrinsic_global_atomic_fmax: - case nir_intrinsic_image_deref_atomic_add: - case nir_intrinsic_image_deref_atomic_umin: - case nir_intrinsic_image_deref_atomic_imin: - case nir_intrinsic_image_deref_atomic_umax: - case nir_intrinsic_image_deref_atomic_imax: - case nir_intrinsic_image_deref_atomic_and: - case nir_intrinsic_image_deref_atomic_or: - case nir_intrinsic_image_deref_atomic_xor: - case nir_intrinsic_image_deref_atomic_exchange: - case nir_intrinsic_image_deref_atomic_comp_swap: - case nir_intrinsic_image_deref_atomic_fmin: - case nir_intrinsic_image_deref_atomic_fmax: - case nir_intrinsic_image_deref_size: - case nir_intrinsic_shared_atomic_add: - case nir_intrinsic_shared_atomic_imin: - case nir_intrinsic_shared_atomic_umin: - case nir_intrinsic_shared_atomic_imax: - case nir_intrinsic_shared_atomic_umax: - case nir_intrinsic_shared_atomic_and: - case nir_intrinsic_shared_atomic_or: - case nir_intrinsic_shared_atomic_xor: - case nir_intrinsic_shared_atomic_exchange: - case nir_intrinsic_shared_atomic_comp_swap: - case nir_intrinsic_shared_atomic_fadd: - case nir_intrinsic_shared_atomic_fmin: - case nir_intrinsic_shared_atomic_fmax: + case nir_intrinsic_ssbo_atomic: + case nir_intrinsic_ssbo_atomic_swap: + case nir_intrinsic_global_atomic_amd: + case nir_intrinsic_global_atomic_swap_amd: + case nir_intrinsic_bindless_image_atomic: + case nir_intrinsic_bindless_image_atomic_swap: + case nir_intrinsic_bindless_image_size: + case nir_intrinsic_shared_atomic: + case nir_intrinsic_shared_atomic_swap: case nir_intrinsic_load_scratch: case nir_intrinsic_load_invocation_id: case nir_intrinsic_load_primitive_id: + case nir_intrinsic_load_typed_buffer_amd: case nir_intrinsic_load_buffer_amd: - case nir_intrinsic_load_tess_rel_patch_id_amd: - case nir_intrinsic_load_gs_vertex_offset_amd: case nir_intrinsic_load_initial_edgeflags_amd: - case nir_intrinsic_load_packed_passthrough_primitive_amd: case nir_intrinsic_gds_atomic_add_amd: case nir_intrinsic_bvh64_intersect_ray_amd: - case nir_intrinsic_load_cull_small_prim_precision_amd: type = RegType::vgpr; break; + case nir_intrinsic_load_vector_arg_amd: + case nir_intrinsic_load_rt_dynamic_callable_stack_base_amd: + case nir_intrinsic_ordered_xfb_counter_add_gfx11_amd: + case nir_intrinsic_cmat_muladd_amd: + case nir_intrinsic_unit_test_divergent_amd: type = RegType::vgpr; break; case nir_intrinsic_load_shared: + case nir_intrinsic_load_shared2_amd: /* When the result of these loads is only used by cross-lane instructions, * it is beneficial to use a VGPR destination. This is because this allows * to put the s_waitcnt further down, which decreases latency. */ - if (only_used_by_cross_lane_instrs(&intrinsic->dest.ssa)) { + if (only_used_by_cross_lane_instrs(&intrinsic->def)) { type = RegType::vgpr; break; } @@ -798,17 +528,14 @@ init_context(isel_context* ctx, nir_shader* shader) case nir_intrinsic_quad_swap_diagonal: case nir_intrinsic_quad_swizzle_amd: case nir_intrinsic_masked_swizzle_amd: + case nir_intrinsic_rotate: case nir_intrinsic_inclusive_scan: case nir_intrinsic_exclusive_scan: case nir_intrinsic_reduce: - case nir_intrinsic_load_sbt_amd: case nir_intrinsic_load_ubo: case nir_intrinsic_load_ssbo: - case nir_intrinsic_load_global: - case nir_intrinsic_load_global_constant: - case nir_intrinsic_vulkan_resource_index: - case nir_intrinsic_get_ssbo_size: - type = nir_dest_is_divergent(intrinsic->dest) ? RegType::vgpr : RegType::sgpr; + case nir_intrinsic_load_global_amd: + type = intrinsic->def.divergent ? RegType::vgpr : RegType::sgpr; break; case nir_intrinsic_load_view_index: type = ctx->stage == fragment_fs ? RegType::vgpr : RegType::sgpr; @@ -821,90 +548,38 @@ init_context(isel_context* ctx, nir_shader* shader) } break; } - RegClass rc = get_reg_class(ctx, type, intrinsic->dest.ssa.num_components, - intrinsic->dest.ssa.bit_size); - regclasses[intrinsic->dest.ssa.index] = rc; - - switch (intrinsic->intrinsic) { - case nir_intrinsic_load_barycentric_sample: - case nir_intrinsic_load_barycentric_pixel: - case nir_intrinsic_load_barycentric_centroid: - case nir_intrinsic_load_barycentric_at_sample: - case nir_intrinsic_load_barycentric_at_offset: { - glsl_interp_mode mode = (glsl_interp_mode)nir_intrinsic_interp_mode(intrinsic); - spi_ps_inputs |= get_interp_input(intrinsic->intrinsic, mode); - break; - } - case nir_intrinsic_load_barycentric_model: - spi_ps_inputs |= S_0286CC_PERSP_PULL_MODEL_ENA(1); - break; - case nir_intrinsic_load_front_face: - spi_ps_inputs |= S_0286CC_FRONT_FACE_ENA(1); - break; - case nir_intrinsic_load_frag_coord: - case nir_intrinsic_load_sample_pos: { - uint8_t mask = nir_ssa_def_components_read(&intrinsic->dest.ssa); - for (unsigned i = 0; i < 4; i++) { - if (mask & (1 << i)) - spi_ps_inputs |= S_0286CC_POS_X_FLOAT_ENA(1) << i; - } - - if (ctx->options->adjust_frag_coord_z && - intrinsic->intrinsic == nir_intrinsic_load_frag_coord && - G_0286CC_POS_Z_FLOAT_ENA(spi_ps_inputs)) { - /* Enable ancillary for adjusting gl_FragCoord.z for - * VRS due to a hw bug on some GFX10.3 chips. - */ - spi_ps_inputs |= S_0286CC_ANCILLARY_ENA(1); - } - break; - } - case nir_intrinsic_load_sample_id: - case nir_intrinsic_load_frag_shading_rate: - spi_ps_inputs |= S_0286CC_ANCILLARY_ENA(1); - break; - case nir_intrinsic_load_sample_mask_in: - spi_ps_inputs |= S_0286CC_ANCILLARY_ENA(1); - spi_ps_inputs |= S_0286CC_SAMPLE_COVERAGE_ENA(1); - break; - default: break; - } + RegClass rc = + get_reg_class(ctx, type, intrinsic->def.num_components, intrinsic->def.bit_size); + regclasses[intrinsic->def.index] = rc; break; } case nir_instr_type_tex: { nir_tex_instr* tex = nir_instr_as_tex(instr); - RegType type = nir_dest_is_divergent(tex->dest) ? RegType::vgpr : RegType::sgpr; + RegType type = tex->def.divergent ? RegType::vgpr : RegType::sgpr; if (tex->op == nir_texop_texture_samples) { - assert(!tex->dest.ssa.divergent); + assert(!tex->def.divergent); } - RegClass rc = - get_reg_class(ctx, type, tex->dest.ssa.num_components, tex->dest.ssa.bit_size); - regclasses[tex->dest.ssa.index] = rc; - break; - } - case nir_instr_type_parallel_copy: { - nir_foreach_parallel_copy_entry (entry, nir_instr_as_parallel_copy(instr)) { - regclasses[entry->dest.ssa.index] = regclasses[entry->src.ssa->index]; - } + RegClass rc = get_reg_class(ctx, type, tex->def.num_components, tex->def.bit_size); + regclasses[tex->def.index] = rc; break; } - case nir_instr_type_ssa_undef: { - unsigned num_components = nir_instr_as_ssa_undef(instr)->def.num_components; - unsigned bit_size = nir_instr_as_ssa_undef(instr)->def.bit_size; + case nir_instr_type_undef: { + unsigned num_components = nir_instr_as_undef(instr)->def.num_components; + unsigned bit_size = nir_instr_as_undef(instr)->def.bit_size; RegClass rc = get_reg_class(ctx, RegType::sgpr, num_components, bit_size); - regclasses[nir_instr_as_ssa_undef(instr)->def.index] = rc; + regclasses[nir_instr_as_undef(instr)->def.index] = rc; break; } case nir_instr_type_phi: { nir_phi_instr* phi = nir_instr_as_phi(instr); RegType type = RegType::sgpr; - unsigned num_components = phi->dest.ssa.num_components; - assert((phi->dest.ssa.bit_size != 1 || num_components == 1) && + unsigned num_components = phi->def.num_components; + assert((phi->def.bit_size != 1 || num_components == 1) && "Multiple components not supported on boolean phis."); - if (nir_dest_is_divergent(phi->dest)) { + if (phi->def.divergent) { type = RegType::vgpr; } else { nir_foreach_phi_src (src, phi) { @@ -913,10 +588,10 @@ init_context(isel_context* ctx, nir_shader* shader) } } - RegClass rc = get_reg_class(ctx, type, num_components, phi->dest.ssa.bit_size); - if (rc != regclasses[phi->dest.ssa.index]) + RegClass rc = get_reg_class(ctx, type, num_components, phi->def.bit_size); + if (rc != regclasses[phi->def.index]) done = false; - regclasses[phi->dest.ssa.index] = rc; + regclasses[phi->def.index] = rc; break; } default: break; @@ -925,18 +600,8 @@ init_context(isel_context* ctx, nir_shader* shader) } } - if (G_0286CC_POS_W_FLOAT_ENA(spi_ps_inputs)) { - /* If POS_W_FLOAT (11) is enabled, at least one of PERSP_* must be enabled too */ - spi_ps_inputs |= S_0286CC_PERSP_CENTER_ENA(1); - } - - if (!(spi_ps_inputs & 0x7F)) { - /* At least one of PERSP_* (0xF) or LINEAR_* (0x70) must be enabled */ - spi_ps_inputs |= S_0286CC_PERSP_CENTER_ENA(1); - } - - ctx->program->config->spi_ps_input_ena = spi_ps_inputs; - ctx->program->config->spi_ps_input_addr = spi_ps_inputs; + ctx->program->config->spi_ps_input_ena = ctx->program->info.ps.spi_ps_input_ena; + ctx->program->config->spi_ps_input_addr = ctx->program->info.ps.spi_ps_input_addr; ctx->cf_info.nir_to_aco = std::move(nir_to_aco); @@ -947,6 +612,8 @@ init_context(isel_context* ctx, nir_shader* shader) ctx->program->constant_data.insert(ctx->program->constant_data.end(), (uint8_t*)shader->constant_data, (uint8_t*)shader->constant_data + shader->constant_data_size); + + BITSET_CLEAR_RANGE(ctx->output_args, 0, BITSET_SIZE(ctx->output_args)); } void @@ -957,97 +624,67 @@ cleanup_context(isel_context* ctx) isel_context setup_isel_context(Program* program, unsigned shader_count, struct nir_shader* const* shaders, - ac_shader_config* config, struct radv_shader_args* args, bool is_gs_copy_shader) + ac_shader_config* config, const struct aco_compiler_options* options, + const struct aco_shader_info* info, const struct ac_shader_args* args, + SWStage sw_stage) { - SWStage sw_stage = SWStage::None; for (unsigned i = 0; i < shader_count; i++) { switch (shaders[i]->info.stage) { case MESA_SHADER_VERTEX: sw_stage = sw_stage | SWStage::VS; break; case MESA_SHADER_TESS_CTRL: sw_stage = sw_stage | SWStage::TCS; break; case MESA_SHADER_TESS_EVAL: sw_stage = sw_stage | SWStage::TES; break; - case MESA_SHADER_GEOMETRY: - sw_stage = sw_stage | (is_gs_copy_shader ? SWStage::GSCopy : SWStage::GS); - break; + case MESA_SHADER_GEOMETRY: sw_stage = sw_stage | SWStage::GS; break; case MESA_SHADER_FRAGMENT: sw_stage = sw_stage | SWStage::FS; break; + case MESA_SHADER_KERNEL: case MESA_SHADER_COMPUTE: sw_stage = sw_stage | SWStage::CS; break; + case MESA_SHADER_TASK: sw_stage = sw_stage | SWStage::TS; break; + case MESA_SHADER_MESH: sw_stage = sw_stage | SWStage::MS; break; + case MESA_SHADER_RAYGEN: + case MESA_SHADER_CLOSEST_HIT: + case MESA_SHADER_MISS: + case MESA_SHADER_CALLABLE: + case MESA_SHADER_INTERSECTION: + case MESA_SHADER_ANY_HIT: sw_stage = SWStage::RT; break; default: unreachable("Shader stage not implemented"); } } - bool gfx9_plus = args->options->chip_class >= GFX9; - bool ngg = args->shader_info->is_ngg && args->options->chip_class >= GFX10; - HWStage hw_stage{}; - if (sw_stage == SWStage::VS && args->shader_info->vs.as_es && !ngg) - hw_stage = HWStage::ES; - else if (sw_stage == SWStage::VS && !args->shader_info->vs.as_ls && !ngg) - hw_stage = HWStage::VS; - else if (sw_stage == SWStage::VS && ngg) - hw_stage = HWStage::NGG; /* GFX10/NGG: VS without GS uses the HW GS stage */ - else if (sw_stage == SWStage::GS) - hw_stage = HWStage::GS; - else if (sw_stage == SWStage::FS) - hw_stage = HWStage::FS; - else if (sw_stage == SWStage::CS) - hw_stage = HWStage::CS; - else if (sw_stage == SWStage::GSCopy) - hw_stage = HWStage::VS; - else if (sw_stage == SWStage::VS_GS && gfx9_plus && !ngg) - hw_stage = HWStage::GS; /* GFX6-9: VS+GS merged into a GS (and GFX10/legacy) */ - else if (sw_stage == SWStage::VS_GS && ngg) - hw_stage = HWStage::NGG; /* GFX10+: VS+GS merged into an NGG GS */ - else if (sw_stage == SWStage::VS && args->shader_info->vs.as_ls) - hw_stage = HWStage::LS; /* GFX6-8: VS is a Local Shader, when tessellation is used */ - else if (sw_stage == SWStage::TCS) - hw_stage = HWStage::HS; /* GFX6-8: TCS is a Hull Shader */ - else if (sw_stage == SWStage::VS_TCS) - hw_stage = HWStage::HS; /* GFX9-10: VS+TCS merged into a Hull Shader */ - else if (sw_stage == SWStage::TES && !args->shader_info->tes.as_es && !ngg) - hw_stage = HWStage::VS; /* GFX6-9: TES without GS uses the HW VS stage (and GFX10/legacy) */ - else if (sw_stage == SWStage::TES && !args->shader_info->tes.as_es && ngg) - hw_stage = HWStage::NGG; /* GFX10/NGG: TES without GS */ - else if (sw_stage == SWStage::TES && args->shader_info->tes.as_es && !ngg) - hw_stage = HWStage::ES; /* GFX6-8: TES is an Export Shader */ - else if (sw_stage == SWStage::TES_GS && gfx9_plus && !ngg) - hw_stage = HWStage::GS; /* GFX9: TES+GS merged into a GS (and GFX10/legacy) */ - else if (sw_stage == SWStage::TES_GS && ngg) - hw_stage = HWStage::NGG; /* GFX10+: TES+GS merged into an NGG GS */ - else - unreachable("Shader stage not implemented"); - init_program(program, Stage{hw_stage, sw_stage}, args->shader_info, args->options->chip_class, - args->options->family, args->options->wgp_mode, config); + init_program(program, Stage{info->hw_stage, sw_stage}, info, options->gfx_level, options->family, + options->wgp_mode, config); isel_context ctx = {}; ctx.program = program; ctx.args = args; - ctx.options = args->options; + ctx.options = options; ctx.stage = program->stage; - program->workgroup_size = args->shader_info->workgroup_size; + program->workgroup_size = program->info.workgroup_size; assert(program->workgroup_size); - if (ctx.stage == tess_control_hs) - setup_tcs_info(&ctx, shaders[0], NULL); - else if (ctx.stage == vertex_tess_control_hs) - setup_tcs_info(&ctx, shaders[1], shaders[0]); + /* Mesh shading only works on GFX10.3+. */ + ASSERTED bool mesh_shading = ctx.stage.has(SWStage::TS) || ctx.stage.has(SWStage::MS); + assert(!mesh_shading || ctx.program->gfx_level >= GFX10_3); + + setup_tcs_info(&ctx); calc_min_waves(program); unsigned scratch_size = 0; - if (program->stage == gs_copy_vs) { - assert(shader_count == 1); - setup_vs_output_info(&ctx, shaders[0], false, true, &args->shader_info->vs.outinfo); - } else { - for (unsigned i = 0; i < shader_count; i++) { - nir_shader* nir = shaders[i]; - setup_nir(&ctx, nir); - } - - for (unsigned i = 0; i < shader_count; i++) - scratch_size = std::max(scratch_size, shaders[i]->scratch_size); + for (unsigned i = 0; i < shader_count; i++) { + nir_shader* nir = shaders[i]; + setup_nir(&ctx, nir); + setup_lds_size(&ctx, nir); } - ctx.program->config->scratch_bytes_per_wave = align(scratch_size * ctx.program->wave_size, 1024); + for (unsigned i = 0; i < shader_count; i++) + scratch_size = std::max(scratch_size, shaders[i]->scratch_size); + + ctx.program->config->scratch_bytes_per_wave = scratch_size * ctx.program->wave_size; + unsigned nir_num_blocks = 0; + for (unsigned i = 0; i < shader_count; i++) + nir_num_blocks += nir_shader_get_entrypoint(shaders[i])->num_blocks; + ctx.program->blocks.reserve(nir_num_blocks * 2); ctx.block = ctx.program->create_and_insert_block(); ctx.block->kind = block_kind_top_level; |