radeonsi/nir: perform lowering of input/output driver locations

3 files changed, 29 insertions, 0 deletions

diff --git a/src/gallium/drivers/radeonsi/si_shader.h b/src/gallium/drivers/radeonsi/si_shader.h
index 4881bae1f7..f1a1af3fdf 100644
--- a/src/gallium/drivers/radeonsi/si_shader.h
+++ b/src/gallium/drivers/radeonsi/si_shader.h
@@ -622,6 +622,7 @@ const char *si_get_shader_name(const struct si_shader *shader, unsigned processo
 /* si_shader_nir.c */
 void si_nir_scan_shader(const struct nir_shader *nir,
 			struct tgsi_shader_info *info);
+void si_lower_nir(struct si_shader_selector *sel);
 
 /* Inline helpers. */
 
diff --git a/src/gallium/drivers/radeonsi/si_shader_nir.c b/src/gallium/drivers/radeonsi/si_shader_nir.c
index f02abab151..dc2ef8bd40 100644
--- a/src/gallium/drivers/radeonsi/si_shader_nir.c
+++ b/src/gallium/drivers/radeonsi/si_shader_nir.c
@@ -313,6 +313,32 @@ void si_nir_scan_shader(const struct nir_shader *nir,
 	}
 }
 
+/**
+ * Perform "lowering" operations on the NIR that are run once when the shader
+ * selector is created.
+ */
+void
+si_lower_nir(struct si_shader_selector* sel)
+{
+	/* Adjust the driver location of inputs and outputs. The state tracker
+	 * interprets them as slots, while the ac/nir backend interprets them
+	 * as individual components.
+	 */
+	nir_foreach_variable(variable, &sel->nir->inputs)
+		variable->data.driver_location *= 4;
+
+	nir_foreach_variable(variable, &sel->nir->outputs) {
+		variable->data.driver_location *= 4;
+
+		if (sel->nir->stage == MESA_SHADER_FRAGMENT) {
+			if (variable->data.location == FRAG_RESULT_DEPTH)
+				variable->data.driver_location += 2;
+			else if (variable->data.location == FRAG_RESULT_STENCIL)
+				variable->data.driver_location += 1;
+		}
+	}
+}
+
 static void declare_nir_input_vs(struct si_shader_context *ctx,
 				 struct nir_variable *variable, unsigned rel,
 				 LLVMValueRef out[4])
diff --git a/src/gallium/drivers/radeonsi/si_state_shaders.c b/src/gallium/drivers/radeonsi/si_state_shaders.c
index e1a31adadd..afa78ebe36 100644
--- a/src/gallium/drivers/radeonsi/si_state_shaders.c
+++ b/src/gallium/drivers/radeonsi/si_state_shaders.c
@@ -1986,6 +1986,8 @@ static void *si_create_shader_selector(struct pipe_context *ctx,
 		sel->nir = state->ir.nir;
 
 		si_nir_scan_shader(sel->nir, &sel->info);
+
+		si_lower_nir(sel);
 	}
 
 	sel->type = sel->info.processor;