radv: gather the number of output components per stream

This will be also used for splitting the GS->VS ring buffer.

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Dave Airlie <airlied@redhat.com>

diff --git a/src/amd/vulkan/radv_shader.h b/src/amd/vulkan/radv_shader.h
index 465c8d160fef744520a0ee747ecf0176851a2a21..c199b3c27af91a0e38f8c3293cfb3395cc967303 100644 (file)
--- a/src/amd/vulkan/radv_shader.h
+++ b/src/amd/vulkan/radv_shader.h
@@ -159,6 +159,7 @@ struct radv_shader_info {
        } vs;
        struct {
                uint8_t output_usage_mask[VARYING_SLOT_VAR31 + 1];
+               uint8_t num_stream_output_components[4];
                uint8_t max_stream;
        } gs;
        struct {

diff --git a/src/amd/vulkan/radv_shader_info.c b/src/amd/vulkan/radv_shader_info.c
index 00bc2ca5db11df4fb326f4598a481cb8576539b4..126785120d91170c593f42acc8563b9cf8468cfa 100644 (file)
--- a/src/amd/vulkan/radv_shader_info.c
+++ b/src/amd/vulkan/radv_shader_info.c
@@ -438,11 +438,13 @@ static void
 gather_info_output_decl_gs(const nir_shader *nir, const nir_variable *var,
                           struct radv_shader_info *info)
 {
+       unsigned num_components = glsl_get_component_slots(var->type);
        unsigned stream = var->data.stream;
 
        assert(stream < 4);
 
        info->gs.max_stream = MAX2(info->gs.max_stream, stream);
+       info->gs.num_stream_output_components[stream] += num_components;
 }
 
 static void