freedreno/ir3: image atomics use image-store path

image reads are handled via tex state, whereas image writes and atomics
are handled via SSBO state block.  Previously we were only considering
image write, and not image atomics which also uses the SSBO state block.

Signed-off-by: Rob Clark <robdclark@gmail.com>

diff --git a/src/gallium/drivers/freedreno/ir3/ir3_nir.c b/src/gallium/drivers/freedreno/ir3/ir3_nir.c
index cd1f9c5..a3896a3 100644 (file)
--- a/src/gallium/drivers/freedreno/ir3/ir3_nir.c
+++ b/src/gallium/drivers/freedreno/ir3/ir3_nir.c
@@ -242,6 +242,14 @@ ir3_nir_scan_driver_consts(nir_shader *shader,
                                                layout->ssbo_size.count;
                                        layout->ssbo_size.count += 1; /* one const per */
                                        break;
+                               case nir_intrinsic_image_var_atomic_add:
+                               case nir_intrinsic_image_var_atomic_min:
+                               case nir_intrinsic_image_var_atomic_max:
+                               case nir_intrinsic_image_var_atomic_and:
+                               case nir_intrinsic_image_var_atomic_or:
+                               case nir_intrinsic_image_var_atomic_xor:
+                               case nir_intrinsic_image_var_atomic_exchange:
+                               case nir_intrinsic_image_var_atomic_comp_swap:
                                case nir_intrinsic_image_var_store:
                                        idx = intr->variables[0]->var->data.driver_location;
                                        if (layout->image_dims.mask & (1 << idx))