| | |
|---|
| | | #ifdef GPU |
|---|
| | | #ifndef OPENCL_H |
|---|
| | | #define OPENCL_H |
|---|
| | | extern int gpu_index; |
|---|
| | | #ifdef GPU |
|---|
| | | #ifdef __APPLE__ |
|---|
| | | #include <OpenCL/opencl.h> |
|---|
| | | #else |
|---|
| | | #include <CL/cl.h> |
|---|
| | | #endif |
|---|
| | | |
|---|
| | | #define NUM_QUEUES 8 |
|---|
| | | |
|---|
| | | typedef struct { |
|---|
| | | int initialized; |
|---|
| | |
|---|
| | | cl_device_id device; |
|---|
| | | cl_context context; |
|---|
| | | cl_command_queue queue; |
|---|
| | | cl_command_queue queues[NUM_QUEUES]; |
|---|
| | | }cl_info; |
|---|
| | | |
|---|
| | | extern cl_info cl; |
|---|
| | | |
|---|
| | | void cl_setup(); |
|---|
| | | void wait_for_queues(); |
|---|
| | | void check_error(cl_info info); |
|---|
| | | cl_kernel get_kernel(char *filename, char *kernelname, char *options); |
|---|
| | | void cl_read_array(cl_mem mem, float *x, int n); |
|---|
| | |
|---|
| | | cl_mem cl_make_int_array(int *x, int n); |
|---|
| | | void cl_copy_array(cl_mem src, cl_mem dst, int n); |
|---|
| | | cl_mem cl_sub_array(cl_mem src, int offset, int size); |
|---|
| | | float cl_checksum(cl_mem mem, int n); |
|---|
| | | #endif |
|---|
| | | #endif |
|---|
