From dcc64fcfed6904f6be41ee4b742c818e76b85712 Mon Sep 17 00:00:00 2001 From: Connor Abbott Date: Tue, 27 Aug 2019 10:54:12 +0200 Subject: [PATCH] nir: Fix num_ssbos when lowering atomic counters MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Otherwise it's impossible to know the maximum SSBO index for both internal TGSI shaders from TTN (which don't have any notion of atomic counters and no offset) as well as shaders from GLSL. I fixed everything I could find while grepping for num_ssbos and num_abos, which hopefully is everything (iris was the only user I could find that uses it in a meaningful way). Reviewed-by: Marek Olšák --- src/broadcom/compiler/vir.c | 3 +-- src/compiler/nir/nir_lower_atomics_to_ssbo.c | 21 ++++++++++++++++++++ src/gallium/drivers/iris/iris_program.c | 5 +---- 3 files changed, 23 insertions(+), 6 deletions(-) diff --git a/src/broadcom/compiler/vir.c b/src/broadcom/compiler/vir.c index 78362a2949c..917ccfeaef1 100644 --- a/src/broadcom/compiler/vir.c +++ b/src/broadcom/compiler/vir.c @@ -812,8 +812,7 @@ v3d_nir_lower_fs_early(struct v3d_compile *c) * enabling early_fragment_tests even if the user didn't. */ if (!(c->s->info.num_images || - c->s->info.num_ssbos || - c->s->info.num_abos)) { + c->s->info.num_ssbos)) { c->s->info.fs.early_fragment_tests = true; } } diff --git a/src/compiler/nir/nir_lower_atomics_to_ssbo.c b/src/compiler/nir/nir_lower_atomics_to_ssbo.c index 66040c5f7f8..918f060a8df 100644 --- a/src/compiler/nir/nir_lower_atomics_to_ssbo.c +++ b/src/compiler/nir/nir_lower_atomics_to_ssbo.c @@ -239,6 +239,27 @@ nir_lower_atomics_to_ssbo(nir_shader *shader, unsigned ssbo_offset) replaced |= (1 << var->data.binding); } } + + /* Make sure that shader->info.num_ssbos still reflects the maximum SSBO + * index that can be used in the shader. + */ + if (shader->info.num_ssbos > 0) { + shader->info.num_ssbos += ssbo_offset; + } else { + /* We can't use num_abos, because it only represents the number of + * active atomic counters, and currently unlike SSBO's they aren't + * compacted so num_abos actually isn't a bound on the index passed + * to nir_intrinsic_atomic_counter_*. e.g. if we have a single atomic + * counter declared like: + * + * layout(binding=1) atomic_uint counter0; + * + * then when we lower accesses to it the atomic_counter_* intrinsics + * will have 1 as the index but num_abos will still be 1. + * */ + shader->info.num_ssbos = util_last_bit(replaced); + } + shader->info.num_abos = 0; } return progress; diff --git a/src/gallium/drivers/iris/iris_program.c b/src/gallium/drivers/iris/iris_program.c index b1830fbdcbf..bf409e5cd9c 100644 --- a/src/gallium/drivers/iris/iris_program.c +++ b/src/gallium/drivers/iris/iris_program.c @@ -691,10 +691,7 @@ iris_setup_binding_table(const struct gen_device_info *devinfo, */ bt->sizes[IRIS_SURFACE_GROUP_UBO] = num_cbufs + 1; - /* The first IRIS_MAX_ABOs indices in the SSBO group are for atomics, real - * SSBOs start after that. Compaction will remove unused ABOs. - */ - bt->sizes[IRIS_SURFACE_GROUP_SSBO] = IRIS_MAX_ABOS + info->num_ssbos; + bt->sizes[IRIS_SURFACE_GROUP_SSBO] = info->num_ssbos; for (int i = 0; i < IRIS_SURFACE_GROUP_COUNT; i++) assert(bt->sizes[i] <= SURFACE_GROUP_MAX_ELEMENTS); -- 2.30.2