65 lines
1.1 KiB
C
65 lines
1.1 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 (func_t)(void *)
|
|
|
|
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();
|
|
|
|
}
|
|
|
|
void pocl_spawn(context_t * ctx, const void * pfn, void * arguments)
|
|
{
|
|
|
|
vx_pocl_workgroup_func use_pfn = (vx_pocl_workgroup_func) pfn;
|
|
int z;
|
|
int y;
|
|
int x;
|
|
for (z = 0; z < ctx->num_groups[2]; ++z)
|
|
{
|
|
for (y = 0; y < ctx->num_groups[1]; ++y)
|
|
{
|
|
for (x = 0; x < ctx->num_groups[0]; ++x)
|
|
{
|
|
(use_pfn)((uint8_t *)arguments, (uint8_t *)ctx, x, y, z);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|