[Mesa-dev] [PATCH 07/19] radv: Add shader info support for image deref instructions.

[Bas Nieuwenhuizen]

---
 src/amd/vulkan/radv_shader_info.c | 40 ++++++++++++++++++++++++++++---
 1 file changed, 37 insertions(+), 3 deletions(-)

diff --git a/src/amd/vulkan/radv_shader_info.c 
b/src/amd/vulkan/radv_shader_info.c
index 57175b3d37b..5b31703f9cf 100644
--- a/src/amd/vulkan/radv_shader_info.c
+++ b/src/amd/vulkan/radv_shader_info.c
@@ -156,9 +156,7 @@ gather_intrinsic_info(const nir_shader *nir, const 
nir_intrinsic_instr *instr,
        case nir_intrinsic_image_var_atomic_exchange:
        case nir_intrinsic_image_var_atomic_comp_swap:
        case nir_intrinsic_image_var_size: {
-               const struct glsl_type *type = instr->variables[0]->var->type;
-               if(instr->variables[0]->deref.child)
-                       type = instr->variables[0]->deref.child->type;
+               const struct glsl_type *type = 
glsl_without_array(instr->variables[0]->var->type);
 
                enum glsl_sampler_dim dim = glsl_get_sampler_dim(type);
                if (dim == GLSL_SAMPLER_DIM_SUBPASS ||
@@ -182,6 +180,42 @@ gather_intrinsic_info(const nir_shader *nir, const 
nir_intrinsic_instr *instr,
                }
                break;
        }
+       case nir_intrinsic_image_deref_load:
+       case nir_intrinsic_image_deref_store:
+       case nir_intrinsic_image_deref_atomic_add:
+       case nir_intrinsic_image_deref_atomic_min:
+       case nir_intrinsic_image_deref_atomic_max:
+       case nir_intrinsic_image_deref_atomic_and:
+       case nir_intrinsic_image_deref_atomic_or:
+       case nir_intrinsic_image_deref_atomic_xor:
+       case nir_intrinsic_image_deref_atomic_exchange:
+       case nir_intrinsic_image_deref_atomic_comp_swap:
+       case nir_intrinsic_image_deref_size: {
+               nir_variable *var = 
nir_deref_instr_get_variable(nir_instr_as_deref(instr->src[0].ssa->parent_instr));
+               const struct glsl_type *type = glsl_without_array(var->type);
+
+               enum glsl_sampler_dim dim = glsl_get_sampler_dim(type);
+               if (dim == GLSL_SAMPLER_DIM_SUBPASS ||
+                   dim == GLSL_SAMPLER_DIM_SUBPASS_MS) {
+                       info->ps.layer_input = true;
+                       info->ps.uses_input_attachments = true;
+               }
+               mark_sampler_desc(var, info);
+
+               if (nir_intrinsic_image_deref_store ||
+                   nir_intrinsic_image_deref_atomic_add ||
+                   nir_intrinsic_image_deref_atomic_min ||
+                   nir_intrinsic_image_deref_atomic_max ||
+                   nir_intrinsic_image_deref_atomic_and ||
+                   nir_intrinsic_image_deref_atomic_or ||
+                   nir_intrinsic_image_deref_atomic_xor ||
+                   nir_intrinsic_image_deref_atomic_exchange ||
+                   nir_intrinsic_image_deref_atomic_comp_swap) {
+                       if (nir->info.stage == MESA_SHADER_FRAGMENT)
+                               info->ps.writes_memory = true;
+               }
+               break;
+       }
        case nir_intrinsic_store_ssbo:
        case nir_intrinsic_ssbo_atomic_add:
        case nir_intrinsic_ssbo_atomic_imin:
-- 
2.17.0

_______________________________________________
mesa-dev mailing list
mesa-dev@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/mesa-dev