[Mesa-dev] [PATCH 2/2] radeonsi/nir: set TGSI_PROPERTY_NEXT_SHADER

[Timothy Arceri]

---
 src/gallium/drivers/radeonsi/si_shader_nir.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/gallium/drivers/radeonsi/si_shader_nir.c b/src/gallium/drivers/radeonsi/si_shader_nir.c
index d410a6c2d6..57f8dc6f03 100644
--- a/src/gallium/drivers/radeonsi/si_shader_nir.c
+++ b/src/gallium/drivers/radeonsi/si_shader_nir.c
@@ -247,6 +247,15 @@ void si_nir_scan_shader(const struct nir_shader *nir,
 	info->num_tokens = 2; /* indicate that the shader is non-empty */
 	info->num_instructions = 2;
 
+	/* Set the next shader stage hint for VS and TES. */
+	if (!nir->info.separate_shader &&
+	    (nir->info.stage == MESA_SHADER_VERTEX ||
+	     nir->info.stage == MESA_SHADER_TESS_EVAL)) {
+
+		info->properties[TGSI_PROPERTY_NEXT_SHADER] =
+			pipe_shader_type_from_mesa(nir->info.next_stage);
+	}
+
 	if (nir->info.stage == MESA_SHADER_TESS_CTRL) {
 		info->properties[TGSI_PROPERTY_TCS_VERTICES_OUT] =
 			nir->info.tess.tcs_vertices_out;
-- 
2.14.3