For some reasons, InstanceID is VGPR3 although StepRate0 is set to 1.

v3: fix instanceID input VGPR for geometry
v2: fix instanceID

Signed-off-by: Samuel Pitoiset <samuel.pitoi...@gmail.com>
---
 src/amd/vulkan/radv_nir_to_llvm.c | 12 +++++++++---
 src/amd/vulkan/radv_shader.c      |  8 ++++++--
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/amd/vulkan/radv_nir_to_llvm.c 
b/src/amd/vulkan/radv_nir_to_llvm.c
index 336bae28614..cf73cdc692b 100644
--- a/src/amd/vulkan/radv_nir_to_llvm.c
+++ b/src/amd/vulkan/radv_nir_to_llvm.c
@@ -852,9 +852,15 @@ declare_vs_input_vgprs(struct radv_shader_context *ctx, 
struct arg_info *args)
                        }
                } else {
                        if (ctx->ac.chip_class >= GFX10) {
-                               add_arg(args, ARG_VGPR, ctx->ac.i32, NULL); /* 
user vgpr */
-                               add_arg(args, ARG_VGPR, ctx->ac.i32, NULL); /* 
user vgpr */
-                               add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->abi.instance_id);
+                               if (ctx->options->key.vs_common_out.as_ngg) {
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
NULL); /* user vgpr */
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
NULL); /* user vgpr */
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->abi.instance_id);
+                               } else {
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
NULL); /* unused */
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->vs_prim_id);
+                                       add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->abi.instance_id);
+                               }
                        } else {
                                add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->abi.instance_id);
                                add_arg(args, ARG_VGPR, ctx->ac.i32, 
&ctx->vs_prim_id);
diff --git a/src/amd/vulkan/radv_shader.c b/src/amd/vulkan/radv_shader.c
index 3adaf52e152..06122664a13 100644
--- a/src/amd/vulkan/radv_shader.c
+++ b/src/amd/vulkan/radv_shader.c
@@ -765,7 +765,7 @@ static void radv_postprocess_config(const struct 
radv_physical_device *pdevice,
                        if (info->vs.export_prim_id) {
                                vgpr_comp_cnt = 2;
                        } else if (info->info.vs.needs_instance_id) {
-                               vgpr_comp_cnt = 1;
+                               vgpr_comp_cnt = pdevice->rad_info.chip_class >= 
GFX10 ? 3 : 1;
                        } else {
                                vgpr_comp_cnt = 0;
                        }
@@ -837,7 +837,11 @@ static void radv_postprocess_config(const struct 
radv_physical_device *pdevice,
 
                if (es_type == MESA_SHADER_VERTEX) {
                        /* VGPR0-3: (VertexID, InstanceID / StepRate0, ...) */
-                       es_vgpr_comp_cnt = info->info.vs.needs_instance_id ? 1 
: 0;
+                       if (info->info.vs.needs_instance_id) {
+                               es_vgpr_comp_cnt = pdevice->rad_info.chip_class 
>= GFX10 ? 3 : 1;
+                       } else {
+                               es_vgpr_comp_cnt = 0;
+                       }
                } else if (es_type == MESA_SHADER_TESS_EVAL) {
                        es_vgpr_comp_cnt = info->info.uses_prim_id ? 3 : 2;
                } else {
-- 
2.22.0

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

Reply via email to