adding work groups support to spawntasks API

This commit is contained in:
Blaise Tine 2024-05-06 01:25:13 -07:00
parent 0cabd24f08
commit 79f5824c74
8 changed files with 581 additions and 246 deletions

View file

@ -21,33 +21,18 @@
extern "C" {
#endif
typedef struct {
uint32_t num_groups[3];
uint32_t global_offset[3];
uint32_t local_size[3];
uint32_t printf_buffer;
uint32_t printf_buffer_position;
uint32_t printf_buffer_capacity;
uint32_t work_dim;
} pocl_kernel_context_t;
typedef void (*pocl_kernel_cb) (
const void * /* arg */,
const pocl_kernel_context_t * /* context */,
uint32_t /* group_x */,
uint32_t /* group_y */,
uint32_t /* group_z */,
uint32_t /* local_offset */
);
typedef void (*vx_spawn_tasks_cb)(int task_id, void *arg);
typedef void (*vx_spawn_tasks_ex_cb)(int local_task_id, int group_id, void *arg);
typedef void (*vx_serial_cb)(void *arg);
void vx_spawn_pocl_kernel(pocl_kernel_context_t * ctx, pocl_kernel_cb callback, void * arg);
void vx_spawn_tasks(int num_tasks, vx_spawn_tasks_cb callback, void * arg);
void vx_spawn_tasks_ex(int num_groups, int group_size, vx_spawn_tasks_ex_cb callback, void * arg);
void vx_syncthreads(int barrier_id);
void vx_serial(vx_serial_cb callback, void * arg);
#ifdef __cplusplus