From: Antia Puentes <apuen...@igalia.com>
When constructing NIR if we have a SPIR-V uint variable and the
storage class is SpvStorageClassAtomicCounter, we store as NIR's
glsl_type an atomic_uint to reflect the fact that the variable is an
atomic counter.
However, we were tweaking the type only for atomic_uint scalars, we
have to do it as well for atomic_uint arrays and atomic_uint arrays of
arrays of any depth.
Signed-off-by: Antia Puentes <apuen...@igalia.com>
Signed-off-by: Alejandro Piñeiro <apinhe...@igalia.com>
v2: update after deref patches got pushed (Alejandro Piñeiro)
---
src/compiler/spirv/vtn_variables.c | 40 +++++++++++++++++++++++++++++++++++---
1 file changed, 37 insertions(+), 3 deletions(-)
diff --git a/src/compiler/spirv/vtn_variables.c
b/src/compiler/spirv/vtn_variables.c
index a40c30c8a75..c75492ef43f 100644
--- a/src/compiler/spirv/vtn_variables.c
+++ b/src/compiler/spirv/vtn_variables.c
@@ -374,6 +374,41 @@ vtn_pointer_for_variable(struct vtn_builder *b,
return pointer;
}
+/* Returns an atomic_uint type based on the original uint type. The returned
+ * type will be equivalent to the original one but will have an atomic_uint
+ * type as leaf instead of an uint.
+ *
+ * Manages uint scalars, arrays, and arrays of arrays of any nested depth.
+ */
+static const struct glsl_type *
+repair_atomic_type(const struct glsl_type *type, SpvStorageClass storage_class)
+{
+ assert(storage_class == SpvStorageClassAtomicCounter);
+ assert(glsl_get_base_type(glsl_without_array(type)) == GLSL_TYPE_UINT);
+ assert(glsl_type_is_scalar(glsl_without_array(type)));
+
+ const struct glsl_type *atomic = glsl_atomic_uint_type();
+ unsigned depth = glsl_array_depth(type);
+
+ if (depth > 0) {
+ unsigned *lengths = calloc(depth, sizeof(unsigned));
+ unsigned i = depth;
+
+ while (glsl_type_is_array(type)) {
+ i--;
+ lengths[i] = glsl_get_length(type);
+ type = glsl_get_array_element(type);
+ }
+
+ for (i = 0; i < depth; i++)
+ atomic = glsl_array_type(atomic, lengths[i]);
+
+ free(lengths);
+ }
+
+ return atomic;
+}
+
nir_deref_instr *
vtn_pointer_to_deref(struct vtn_builder *b, struct vtn_pointer *ptr)
{
@@ -1648,9 +1683,8 @@ vtn_create_variable(struct vtn_builder *b, struct
vtn_value *val,
* the access to storage_class, that is the one that points us that is
* an atomic uint.
*/
- if (glsl_get_base_type(var->type->type) == GLSL_TYPE_UINT &&
- storage_class == SpvStorageClassAtomicCounter) {
- var->var->type = glsl_atomic_uint_type();
+ if (storage_class == SpvStorageClassAtomicCounter) {
+ var->var->type = repair_atomic_type(var->type->type, storage_class);
} else {
var->var->type = var->type->type;
}
--
2.14.1
_______________________________________________
mesa-dev mailing list
mesa-dev@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/mesa-dev
