radeonsi: implement vertex color clamping for tess and GS

1 files changed, 6 insertions, 4 deletions

diff --git a/src/gallium/drivers/radeonsi/si_descriptors.c b/src/gallium/drivers/radeonsi/si_descriptors.c
index e8c028061d1..83f62e4ac93 100644
--- a/src/gallium/drivers/radeonsi/si_descriptors.c
+++ b/src/gallium/drivers/radeonsi/si_descriptors.c
@@ -2002,12 +2002,14 @@ static void si_set_user_data_base(struct si_context *sctx,
 	if (*base != new_base) {
 		*base = new_base;
 
-		if (new_base) {
+		if (new_base)
 			si_mark_shader_pointers_dirty(sctx, shader);
 
-			if (shader == PIPE_SHADER_VERTEX)
-				sctx->last_vs_state = ~0;
-		}
+		/* Any change in enabled shader stages requires re-emitting
+		 * the VS state SGPR, because it contains the clamp_vertex_color
+		 * state, which can be done in VS, TES, and GS.
+		 */
+		sctx->last_vs_state = ~0;
 	}
 }