diff options
-rw-r--r-- | src/gallium/drivers/radeonsi/si_compute.c | 6 |
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) |