mirror of
https://github.com/vortexgpgpu/vortex.git
synced 2025-04-24 22:07:41 -04:00
75 lines
No EOL
1.5 KiB
C
75 lines
No EOL
1.5 KiB
C
|
|
#include "../intrinsics/vx_intrinsics.h"
|
|
#include "vx_api.h"
|
|
#include <inttypes.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
func_t global_function_pointer;
|
|
void * global_argument_struct;
|
|
unsigned global_num_threads;
|
|
|
|
void setup_call() {
|
|
vx_tmc(global_num_threads);
|
|
global_function_pointer(global_argument_struct);
|
|
|
|
unsigned wid = vx_warpID();
|
|
if (wid != 0) {
|
|
vx_tmc(0); // Halt Warp Execution
|
|
} else {
|
|
vx_tmc(1); // Only activate one thread
|
|
}
|
|
}
|
|
|
|
void vx_spawnWarps(unsigned numWarps, unsigned numThreads, func_t func_ptr, void * args) {
|
|
global_function_pointer = func_ptr;
|
|
global_argument_struct = args;
|
|
global_num_threads = numThreads;
|
|
vx_wspawn(numWarps, (unsigned) setup_call);
|
|
setup_call();
|
|
}
|
|
|
|
unsigned pocl_threads;
|
|
struct context_t * pocl_ctx;
|
|
vx_pocl_workgroup_func pocl_pfn;
|
|
const void * pocl_args;
|
|
|
|
void pocl_spawn_runonce() {
|
|
|
|
vx_tmc(pocl_threads);
|
|
|
|
int x = vx_threadID();
|
|
int y = vx_warpID();
|
|
|
|
(pocl_pfn)(pocl_args, pocl_ctx, x, y, 0);
|
|
|
|
if (y != 0) {
|
|
vx_tmc(0);
|
|
}
|
|
|
|
vx_tmc(1);
|
|
}
|
|
|
|
void pocl_spawn(struct context_t * ctx, vx_pocl_workgroup_func pfn, const void * args) {
|
|
if (ctx->num_groups[2] > 1) {
|
|
printf("ERROR: pocl_spawn doesn't support Z dimension yet!\n");
|
|
return;
|
|
}
|
|
|
|
pocl_threads = ctx->num_groups[0];
|
|
pocl_ctx = ctx;
|
|
pocl_pfn = pfn;
|
|
pocl_args = args;
|
|
|
|
if (ctx->num_groups[1] > 1) {
|
|
vx_wspawn(ctx->num_groups[1], (unsigned)&pocl_spawn_runonce);
|
|
}
|
|
|
|
pocl_spawn_runonce();
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif |