From 7e83962e8534b5e72eda35027081d03deb32e728 Mon Sep 17 00:00:00 2001
From: Qiang Yu <yuq825@gmail.com>
Date: Tue, 29 Apr 2025 15:30:03 +0800
Subject: [PATCH] radeonsi: update scratch va for mesh shader
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Reviewed-by: Marek Olšák <marek.olsak@amd.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/37505>
---
 src/gallium/drivers/radeonsi/si_state_shaders.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/gallium/drivers/radeonsi/si_state_shaders.cpp b/src/gallium/drivers/radeonsi/si_state_shaders.cpp
index 4eede60627d..2b42bae7d38 100644
--- a/src/gallium/drivers/radeonsi/si_state_shaders.cpp
+++ b/src/gallium/drivers/radeonsi/si_state_shaders.cpp
@@ -4497,6 +4497,12 @@ static bool si_update_scratch_relocs(struct si_context *sctx)
          si_pm4_bind_state(sctx, vs, sctx->shader.tes.current);
    }
 
+   r = si_update_scratch_buffer(sctx, sctx->ms_shader_state.current);
+   if (r < 0)
+      return false;
+   if (r == 1)
+      si_pm4_bind_state(sctx, gs, sctx->ms_shader_state.current);
+
    return true;
 }