From: Marek Olšák <marek.ol...@amd.com>

---
 src/gallium/drivers/radeonsi/si_shader.h | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/gallium/drivers/radeonsi/si_shader.h 
b/src/gallium/drivers/radeonsi/si_shader.h
index f1e162e..d6b19a1 100644
--- a/src/gallium/drivers/radeonsi/si_shader.h
+++ b/src/gallium/drivers/radeonsi/si_shader.h
@@ -266,14 +266,20 @@ struct si_shader {
 
 static inline struct tgsi_shader_info *si_get_vs_info(struct si_context *sctx)
 {
-       return sctx->gs_shader ? &sctx->gs_shader->info
-                               : &sctx->vs_shader->info;
+       if (sctx->gs_shader)
+               return &sctx->gs_shader->info;
+       else if (sctx->tes_shader)
+               return &sctx->tes_shader->info;
+       else
+               return &sctx->vs_shader->info;
 }
 
 static inline struct si_shader* si_get_vs_state(struct si_context *sctx)
 {
        if (sctx->gs_shader)
                return sctx->gs_shader->current->gs_copy_shader;
+       else if (sctx->tes_shader)
+               return sctx->tes_shader->current;
        else
                return sctx->vs_shader->current;
 }
-- 
2.1.0

_______________________________________________
mesa-dev mailing list
mesa-dev@lists.freedesktop.org
http://lists.freedesktop.org/mailman/listinfo/mesa-dev

Reply via email to