From: Michel Dänzer <michel.daen...@amd.com>

Signed-off-by: Michel Dänzer <michel.daen...@amd.com>
---
 src/gallium/drivers/radeonsi/si_shader.c     | 15 ++++++++++++++-
 src/gallium/drivers/radeonsi/si_shader.h     |  1 +
 src/gallium/drivers/radeonsi/si_state_draw.c |  2 +-
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/gallium/drivers/radeonsi/si_shader.c 
b/src/gallium/drivers/radeonsi/si_shader.c
index 62ebf13..35a646b 100644
--- a/src/gallium/drivers/radeonsi/si_shader.c
+++ b/src/gallium/drivers/radeonsi/si_shader.c
@@ -177,6 +177,7 @@ static int si_store_shader_io_attribs(struct si_shader 
*shader,
                assert(i < Elements(shader->input));
                shader->input[i].name = d->Semantic.Name;
                shader->input[i].sid = d->Semantic.Index;
+               shader->input[i].index = d->Range.First;
                shader->input[i].interpolate = d->Interp.Interpolate;
                shader->input[i].centroid = d->Interp.Centroid;
                return -1;
@@ -268,7 +269,9 @@ static void declare_input_gs(
        struct si_shader *shader = &si_shader_ctx->shader->shader;
 
        si_store_shader_io_attribs(shader, decl);
-       shader->input[input_index].param_offset = shader->nparam++;
+
+       if (decl->Semantic.Name != TGSI_SEMANTIC_PRIMID)
+               shader->input[input_index].param_offset = shader->nparam++;
 }
 
 static LLVMValueRef fetch_input_gs(
@@ -289,6 +292,15 @@ static LLVMValueRef fetch_input_gs(
        LLVMValueRef args[9];
        unsigned vtx_offset_param;
 
+       if (swizzle != ~0 &&
+           shader->input[reg->Register.Index].name == TGSI_SEMANTIC_PRIMID) {
+               if (swizzle == 0)
+                       return LLVMGetParam(si_shader_ctx->radeon_bld.main_fn,
+                                           SI_PARAM_PRIMITIVE_ID);
+               else
+                       return uint->zero;
+       }
+
        if (!reg->Register.Dimension)
                return NULL;
 
@@ -1052,6 +1064,7 @@ handle_semantic:
                case TGSI_SEMANTIC_CLIPVERTEX:
                        si_llvm_emit_clipvertex(bld_base, pos_args, 
outputs[i].values);
                        continue;
+               case TGSI_SEMANTIC_PRIMID:
                case TGSI_SEMANTIC_FOG:
                case TGSI_SEMANTIC_GENERIC:
                        target = V_008DFC_SQ_EXP_PARAM + param_count;
diff --git a/src/gallium/drivers/radeonsi/si_shader.h 
b/src/gallium/drivers/radeonsi/si_shader.h
index 3299906..d11d346 100644
--- a/src/gallium/drivers/radeonsi/si_shader.h
+++ b/src/gallium/drivers/radeonsi/si_shader.h
@@ -93,6 +93,7 @@ struct si_shader_input {
        unsigned                name;
        int                     sid;
        unsigned                param_offset;
+       unsigned                index;
        unsigned                interpolate;
        bool                    centroid;
 };
diff --git a/src/gallium/drivers/radeonsi/si_state_draw.c 
b/src/gallium/drivers/radeonsi/si_state_draw.c
index 3fc5074..2185f16 100644
--- a/src/gallium/drivers/radeonsi/si_state_draw.c
+++ b/src/gallium/drivers/radeonsi/si_state_draw.c
@@ -148,7 +148,7 @@ static void si_pipe_shader_gs(struct pipe_context *ctx, 
struct si_pipe_shader *s
                       si_conv_prim_to_gs_out(shader->shader.gs_output_prim));
 
        si_pm4_set_reg(pm4, R_028AAC_VGT_ESGS_RING_ITEMSIZE,
-                      shader->shader.ninput * (16 >> 2));
+                      shader->shader.nparam * (16 >> 2));
        si_pm4_set_reg(pm4, R_028AB0_VGT_GSVS_RING_ITEMSIZE, gsvs_itemsize);
 
        si_pm4_set_reg(pm4, R_028B38_VGT_GS_MAX_VERT_OUT, gs_max_vert_out);
-- 
1.8.5.3

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

Reply via email to