panfrost: Invoke compute shader according to grid info

We already have helpers for packing invocations (due to its role in
instanced vertex shaders), so we can reuse this drop in for compute
shaders.

Signed-off-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>

diff --git a/src/gallium/drivers/panfrost/pan_compute.c b/src/gallium/drivers/panfrost/pan_compute.c
index 34d27904b3f2b85996005c69d588a15a3fce5aaf..ee3aa5edb367014b36760f0cf1a136ae44687b87 100644 (file)
--- a/src/gallium/drivers/panfrost/pan_compute.c
+++ b/src/gallium/drivers/panfrost/pan_compute.c
@@ -85,6 +85,12 @@ panfrost_launch_grid(struct pipe_context *pipe,
         payload->postfix.framebuffer =
                 panfrost_upload_transient(ctx, &compute_fbd, sizeof(compute_fbd));
 
+        /* Invoke according to the grid info */
+
+        panfrost_pack_work_groups_compute(&payload->prefix,
+                        info->grid[0], info->grid[1], info->grid[2],
+                        info->block[0], info->block[1], info->block[2]);
+
         /* Upload the payload */
 
         struct panfrost_transfer transfer = panfrost_allocate_transient(ctx, sizeof(job) + sizeof(*payload));