nir: handle get_buffer_size in nir_lower_atomics_to_ssbo

Overlooked initially, be we need to remap the SSBO index for this as
well.

Signed-off-by: Rob Clark <robdclark@gmail.com>
Reviewed-by: Jason Ekstrand <jason@jlekstrand.net>
Reviewed-by: Kenneth Graunke <kenneth@whitecape.org>

diff --git a/src/compiler/nir/nir_lower_atomics_to_ssbo.c b/src/compiler/nir/nir_lower_atomics_to_ssbo.c
index 371eb0b9d15c3dce4f4b76997546c608883c7f32..934ae81d750595af5b9427b44e7b2f9801d8c6d7 100644 (file)
--- a/src/compiler/nir/nir_lower_atomics_to_ssbo.c
+++ b/src/compiler/nir/nir_lower_atomics_to_ssbo.c
@@ -59,6 +59,7 @@ lower_instr(nir_intrinsic_instr *instr, unsigned ssbo_offset, nir_builder *b)
    case nir_intrinsic_ssbo_atomic_comp_swap:
    case nir_intrinsic_store_ssbo:
    case nir_intrinsic_load_ssbo:
+   case nir_intrinsic_get_buffer_size:
       /* easy case, keep same opcode and just remap SSBO buffer index: */
       op = instr->intrinsic;
       idx_src = (op == nir_intrinsic_store_ssbo) ? 1 : 0;