fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-20 07:20:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

radeonsi: implement task ring nir intrinsic lower

Browse source 
Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/38044>
This commit is contained in:
Qiang Yu 2025-06-10 09:59:56 +08:00 • committed by Marge Bot
parent e819554b2a
commit f1138c18ac
1 changed files with 38 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
src/gallium/drivers/radeonsi/si_nir_lower_abi.c
View file

@ -185,6 +185,38 @@ static bool build_gsvs_ring_desc(nir_builder *b, struct lower_abi_state *s)
return false;
}
static nir_def *build_task_ring_desc(nir_builder *b, struct lower_abi_state *s,
bool payload)
{
struct si_screen *screen = s->shader->selector->screen;
struct ac_task_info *info = &screen->task_info;
unsigned entry_size = payload ? info->payload_entry_size : AC_TASK_DRAW_ENTRY_BYTES;
const struct ac_buffer_state ac_state = {
.va = (uint64_t)screen->info.address32_hi << 32,
.size = screen->task_info.num_entries * entry_size,
.format = PIPE_FORMAT_R32_FLOAT,
.swizzle = { PIPE_SWIZZLE_X, PIPE_SWIZZLE_Y, PIPE_SWIZZLE_Z, PIPE_SWIZZLE_W },
.gfx10_oob_select = V_008F0C_OOB_SELECT_DISABLED,
};
unsigned desc[4];
ac_build_buffer_descriptor(screen->info.gfx_level, &ac_state, desc);
nir_def *addr = ac_nir_load_arg(b, &s->args->ac, s->args->task_ring_addr);
unsigned offset = payload ? info->payload_ring_offset : info->draw_ring_offset;
addr = nir_iadd_imm(b, addr, offset);
nir_def *comp[] = {
addr,
nir_imm_int(b, desc[1]),
nir_imm_int(b, desc[2]),
nir_imm_int(b, desc[3]),
};
return nir_vec(b, comp, 4);
}
static bool preload_reusable_variables(nir_builder *b, struct lower_abi_state *s)
{
const struct si_shader_selector *sel = s->shader->selector;
@ -629,6 +661,12 @@ static bool lower_intrinsic(nir_builder *b, nir_instr *instr, struct lower_abi_s
case nir_intrinsic_load_lds_ngg_gs_out_vertex_base_amd:
replacement = nir_imul_imm(b, GET_FIELD_NIR(GS_STATE_GS_OUT_LDS_OFFSET_256B), 256);
break;
case nir_intrinsic_load_ring_task_draw_amd:
replacement = build_task_ring_desc(b, s, false);
break;
case nir_intrinsic_load_ring_task_payload_amd:
replacement = build_task_ring_desc(b, s, true);
break;
default:
return false;
}