nvk: Lower nir_intrinsic_load_view_index

Lower view_index intrinsics to a load_ubo using NIR.

Signed-off-by: Rebecca Mckeever <rebecca.mckeever@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/24326>
This commit is contained in:
Rebecca Mckeever
2023-02-16 18:40:45 -08:00
committed by Marge Bot
parent f0c86365da
commit ec917719bf
@@ -198,6 +198,27 @@ lower_load_push_constant(nir_builder *b, nir_intrinsic_instr *load,
return true;
}
static bool
lower_load_view_index(nir_builder *b, nir_intrinsic_instr *load,
const struct lower_descriptors_ctx *ctx)
{
const uint32_t root_table_offset =
nvk_root_descriptor_offset(draw.view_index);
b->cursor = nir_instr_remove(&load->instr);
nir_ssa_def *val = nir_load_ubo(b, 1, 32,
nir_imm_int(b, 0),
nir_imm_int(b, root_table_offset),
.align_mul = 4,
.align_offset = 0,
.range = root_table_offset + 4);
nir_ssa_def_rewrite_uses(&load->dest.ssa, val);
return true;
}
static void
get_resource_deref_binding(nir_builder *b, nir_deref_instr *deref,
uint32_t *set, uint32_t *binding,
@@ -273,6 +294,9 @@ lower_intrin(nir_builder *b, nir_intrinsic_instr *intrin,
case nir_intrinsic_load_push_constant:
return lower_load_push_constant(b, intrin, ctx);
case nir_intrinsic_load_view_index:
return lower_load_view_index(b, intrin, ctx);
case nir_intrinsic_image_deref_load:
case nir_intrinsic_image_deref_store:
case nir_intrinsic_image_deref_atomic: