1 /**************************************************************************
3 * Copyright 2016 Samuel Pitoiset
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sub license, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
14 * The above copyright notice and this permission notice (including the
15 * next paragraph) shall be included in all copies or substantial portions
18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
21 * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
26 **************************************************************************/
28 #include "main/state.h"
30 #include "st_context.h"
31 #include "st_cb_bitmap.h"
32 #include "st_cb_bufferobjects.h"
33 #include "st_cb_compute.h"
36 #include "pipe/p_context.h"
38 static void st_dispatch_compute_common(struct gl_context
*ctx
,
39 const GLuint
*num_groups
,
40 const GLuint
*group_size
,
41 struct pipe_resource
*indirect
,
42 GLintptr indirect_offset
)
44 struct gl_program
*prog
=
45 ctx
->_Shader
->CurrentProgram
[MESA_SHADER_COMPUTE
];
46 struct st_context
*st
= st_context(ctx
);
47 struct pipe_context
*pipe
= st
->pipe
;
48 struct pipe_grid_info info
= { 0 };
50 st_flush_bitmap_cache(st
);
51 st_invalidate_readpix_cache(st
);
54 _mesa_update_state(ctx
);
56 if ((st
->dirty
| ctx
->NewDriverState
) & ST_PIPELINE_COMPUTE_STATE_MASK
||
57 st
->compute_shader_may_be_dirty
)
58 st_validate_state(st
, ST_PIPELINE_COMPUTE
);
60 for (unsigned i
= 0; i
< 3; i
++) {
61 info
.block
[i
] = group_size
? group_size
[i
] : prog
->info
.cs
.local_size
[i
];
62 info
.grid
[i
] = num_groups
? num_groups
[i
] : 0;
66 info
.indirect
= indirect
;
67 info
.indirect_offset
= indirect_offset
;
70 pipe
->launch_grid(pipe
, &info
);
73 static void st_dispatch_compute(struct gl_context
*ctx
,
74 const GLuint
*num_groups
)
76 st_dispatch_compute_common(ctx
, num_groups
, NULL
, NULL
, 0);
79 static void st_dispatch_compute_indirect(struct gl_context
*ctx
,
80 GLintptr indirect_offset
)
82 struct gl_buffer_object
*indirect_buffer
= ctx
->DispatchIndirectBuffer
;
83 struct pipe_resource
*indirect
= st_buffer_object(indirect_buffer
)->buffer
;
85 st_dispatch_compute_common(ctx
, NULL
, NULL
, indirect
, indirect_offset
);
88 static void st_dispatch_compute_group_size(struct gl_context
*ctx
,
89 const GLuint
*num_groups
,
90 const GLuint
*group_size
)
92 st_dispatch_compute_common(ctx
, num_groups
, group_size
, NULL
, 0);
95 void st_init_compute_functions(struct dd_function_table
*functions
)
97 functions
->DispatchCompute
= st_dispatch_compute
;
98 functions
->DispatchComputeIndirect
= st_dispatch_compute_indirect
;
99 functions
->DispatchComputeGroupSize
= st_dispatch_compute_group_size
;