summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/gallium/drivers/radeonsi/si_compute.c6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/gallium/drivers/radeonsi/si_compute.c b/src/gallium/drivers/radeonsi/si_compute.c
index e75f6bd2333..c3e8a35ba1d 100644
--- a/src/gallium/drivers/radeonsi/si_compute.c
+++ b/src/gallium/drivers/radeonsi/si_compute.c
@@ -462,6 +462,12 @@ static void si_launch_grid(
si_decompress_compute_textures(sctx);
+ /* Add buffer sizes for memory checking in need_cs_space. */
+ r600_context_add_resource_size(ctx, &program->shader.bo->b.b);
+ if (info->indirect)
+ r600_context_add_resource_size(ctx, info->indirect);
+ /* TODO: add the scratch buffer */
+
si_need_cs_space(sctx);
if (!sctx->cs_shader_state.initialized)