Mesa (main): panfrost: Add shader_stage helper

[GitLab Mirror]

Module: Mesa
Branch: main
Commit: 4d04437a3be47dba05ef8a3cc48447a5756ad790
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=4d04437a3be47dba05ef8a3cc48447a5756ad790

Author: Alyssa Rosenzweig <alyssa at collabora.com>
Date:   Tue Apr  5 12:35:25 2022 -0400

panfrost: Add shader_stage helper

For Valhall, which specifies these in the shader program descriptor.

Signed-off-by: Alyssa Rosenzweig <alyssa at collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/15795>

---

 src/panfrost/lib/pan_shader.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/panfrost/lib/pan_shader.h b/src/panfrost/lib/pan_shader.h
index 2764d2587bf..bd10b00cfb0 100644
--- a/src/panfrost/lib/pan_shader.h
+++ b/src/panfrost/lib/pan_shader.h
@@ -43,6 +43,21 @@ GENX(pan_shader_compile)(nir_shader *nir,
                          struct util_dynarray *binary,
                          struct pan_shader_info *info);
 
+#if PAN_ARCH >= 9
+static inline enum mali_shader_stage
+pan_shader_stage(const struct pan_shader_info *info)
+{
+        switch (info->stage) {
+        case MESA_SHADER_VERTEX:
+                return MALI_SHADER_STAGE_VERTEX;
+        case MESA_SHADER_FRAGMENT:
+                return MALI_SHADER_STAGE_FRAGMENT;
+        default:
+                return MALI_SHADER_STAGE_COMPUTE;
+        }
+}
+#endif
+
 #if PAN_ARCH <= 5
 static inline void
 pan_shader_prepare_midgard_rsd(const struct pan_shader_info *info,