radv/llvm: gather VS input usage mask from load_input

Signed-off-by: Samuel Pitoiset <samuel.pitoiset@gmail.com>
Reviewed-by: Bas Nieuwenhuizen <bas@basnieuwenhuizen.nl>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/6912>
diff --git a/src/amd/vulkan/radv_shader_info.c b/src/amd/vulkan/radv_shader_info.c
index 5e302e5..709485c 100644
--- a/src/amd/vulkan/radv_shader_info.c
+++ b/src/amd/vulkan/radv_shader_info.c
@@ -112,6 +112,25 @@
 	}
 }
 
+static void
+gather_intrinsic_load_input_info(const nir_shader *nir,
+			       const nir_intrinsic_instr *instr,
+			       struct radv_shader_info *info)
+{
+	switch (nir->info.stage) {
+	case MESA_SHADER_VERTEX: {
+		unsigned idx = nir_intrinsic_io_semantics(instr).location;
+		unsigned component = nir_intrinsic_component(instr);
+		unsigned mask = nir_ssa_def_components_read(&instr->dest.ssa);
+
+		info->vs.input_usage_mask[idx] |= mask << component;
+		break;
+	}
+	default:
+		break;
+	}
+}
+
 static uint32_t
 widen_writemask(uint32_t wrmask)
 {
@@ -384,6 +403,9 @@
 	case nir_intrinsic_global_atomic_comp_swap:
 		set_writes_memory(nir, info);
 		break;
+	case nir_intrinsic_load_input:
+		gather_intrinsic_load_input_info(nir, instr, info);
+		break;
 	case nir_intrinsic_load_deref:
 		gather_intrinsic_load_deref_info(nir, instr, info);
 		break;