From: Samuel Pitoiset Date: Tue, 6 Sep 2016 18:15:00 +0000 (+0200) Subject: st/mesa: add support for dispatching a variable local size X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=dfd7734cb785268acd914590733dcb5a30e873a1;p=mesa.git st/mesa: add support for dispatching a variable local size Signed-off-by: Samuel Pitoiset Reviewed-by: Marek Olšák Reviewed-by: Nicolai Hähnle --- diff --git a/src/mesa/state_tracker/st_cb_compute.c b/src/mesa/state_tracker/st_cb_compute.c index 88c1ee2cd1c..ccc5dc2b004 100644 --- a/src/mesa/state_tracker/st_cb_compute.c +++ b/src/mesa/state_tracker/st_cb_compute.c @@ -36,6 +36,7 @@ static void st_dispatch_compute_common(struct gl_context *ctx, const GLuint *num_groups, + const GLuint *group_size, struct pipe_resource *indirect, GLintptr indirect_offset) { @@ -56,7 +57,7 @@ static void st_dispatch_compute_common(struct gl_context *ctx, st_validate_state(st, ST_PIPELINE_COMPUTE); for (unsigned i = 0; i < 3; i++) { - info.block[i] = prog->Comp.LocalSize[i]; + info.block[i] = group_size ? group_size[i] : prog->Comp.LocalSize[i]; info.grid[i] = num_groups ? num_groups[i] : 0; } @@ -71,7 +72,7 @@ static void st_dispatch_compute_common(struct gl_context *ctx, static void st_dispatch_compute(struct gl_context *ctx, const GLuint *num_groups) { - st_dispatch_compute_common(ctx, num_groups, NULL, 0); + st_dispatch_compute_common(ctx, num_groups, NULL, NULL, 0); } static void st_dispatch_compute_indirect(struct gl_context *ctx, @@ -80,11 +81,19 @@ static void st_dispatch_compute_indirect(struct gl_context *ctx, struct gl_buffer_object *indirect_buffer = ctx->DispatchIndirectBuffer; struct pipe_resource *indirect = st_buffer_object(indirect_buffer)->buffer; - st_dispatch_compute_common(ctx, NULL, indirect, indirect_offset); + st_dispatch_compute_common(ctx, NULL, NULL, indirect, indirect_offset); +} + +static void st_dispatch_compute_group_size(struct gl_context *ctx, + const GLuint *num_groups, + const GLuint *group_size) +{ + st_dispatch_compute_common(ctx, num_groups, group_size, NULL, 0); } void st_init_compute_functions(struct dd_function_table *functions) { functions->DispatchCompute = st_dispatch_compute; functions->DispatchComputeIndirect = st_dispatch_compute_indirect; + functions->DispatchComputeGroupSize = st_dispatch_compute_group_size; }