clray
annotate src/ocl.h @ 1:0b0e4d18d53f
added non-buffer args
| author | John Tsiombikas <nuclear@member.fsf.org> |
|---|---|
| date | Mon, 12 Jul 2010 07:00:19 +0300 |
| parents | 5767277e049f |
| children | deaf85acf6af |
| rev | line source |
|---|---|
| nuclear@0 | 1 #ifndef OCL_H_ |
| nuclear@0 | 2 #define OCL_H_ |
| nuclear@0 | 3 |
| nuclear@0 | 4 #include <vector> |
| nuclear@0 | 5 #include <string> |
| nuclear@0 | 6 #ifndef __APPLE__ |
| nuclear@0 | 7 #include <CL/opencl.h> |
| nuclear@0 | 8 #else |
| nuclear@0 | 9 #include <OpenCL/opencl.h> |
| nuclear@0 | 10 #endif |
| nuclear@0 | 11 |
| nuclear@0 | 12 enum { |
| nuclear@0 | 13 ARG_RD = CL_MEM_READ_ONLY, |
| nuclear@0 | 14 ARG_WR = CL_MEM_WRITE_ONLY, |
| nuclear@0 | 15 ARG_RDWR = CL_MEM_READ_WRITE |
| nuclear@0 | 16 }; |
| nuclear@0 | 17 |
| nuclear@0 | 18 enum { |
| nuclear@0 | 19 MAP_RD = CL_MAP_READ, |
| nuclear@0 | 20 MAP_WR = CL_MAP_WRITE, |
| nuclear@0 | 21 MAP_RDWR = CL_MAP_READ | CL_MAP_WRITE |
| nuclear@0 | 22 }; |
| nuclear@0 | 23 |
| nuclear@0 | 24 struct CLMemBuffer { |
| nuclear@0 | 25 cl_mem mem; |
| nuclear@0 | 26 size_t size; |
| nuclear@0 | 27 void *ptr; |
| nuclear@0 | 28 }; |
| nuclear@0 | 29 |
| nuclear@0 | 30 CLMemBuffer *create_mem_buffer(int rdwr, size_t sz, void *buf); |
| nuclear@0 | 31 void destroy_mem_buffer(CLMemBuffer *mbuf); |
| nuclear@0 | 32 |
| nuclear@0 | 33 void *map_mem_buffer(CLMemBuffer *mbuf, int rdwr); |
| nuclear@0 | 34 void unmap_mem_buffer(CLMemBuffer *mbuf); |
| nuclear@0 | 35 |
| nuclear@0 | 36 bool write_mem_buffer(CLMemBuffer *mbuf, size_t sz, void *src); |
| nuclear@0 | 37 bool read_mem_buffer(CLMemBuffer *mbuf, size_t sz, void *dest); |
| nuclear@0 | 38 |
| nuclear@1 | 39 enum { |
| nuclear@1 | 40 ARGTYPE_NONE, |
| nuclear@1 | 41 |
| nuclear@1 | 42 ARGTYPE_INT, |
| nuclear@1 | 43 ARGTYPE_FLOAT, |
| nuclear@1 | 44 ARGTYPE_FLOAT4, |
| nuclear@1 | 45 ARGTYPE_MEM_BUF |
| nuclear@1 | 46 }; |
| nuclear@1 | 47 |
| nuclear@1 | 48 struct CLArg { |
| nuclear@1 | 49 int type; |
| nuclear@1 | 50 union { |
| nuclear@1 | 51 int ival; |
| nuclear@1 | 52 float fval; |
| nuclear@1 | 53 cl_float4 vval; |
| nuclear@1 | 54 CLMemBuffer *mbuf; |
| nuclear@1 | 55 } v; |
| nuclear@1 | 56 }; |
| nuclear@1 | 57 |
| nuclear@0 | 58 |
| nuclear@0 | 59 class CLProgram { |
| nuclear@0 | 60 private: |
| nuclear@0 | 61 std::string kname; |
| nuclear@0 | 62 cl_program prog; |
| nuclear@0 | 63 cl_kernel kernel; |
| nuclear@1 | 64 std::vector<CLArg> args; |
| nuclear@0 | 65 bool built; |
| nuclear@0 | 66 |
| nuclear@0 | 67 public: |
| nuclear@0 | 68 CLProgram(const char *kname); |
| nuclear@0 | 69 ~CLProgram(); |
| nuclear@0 | 70 |
| nuclear@0 | 71 bool load(const char *fname); |
| nuclear@0 | 72 |
| nuclear@1 | 73 bool set_argi(int arg, int val); |
| nuclear@1 | 74 bool set_argf(int arg, float val); |
| nuclear@1 | 75 bool set_arg_buffer(int arg, int rdwr, size_t sz, void *buf); |
| nuclear@0 | 76 CLMemBuffer *get_arg_buffer(int arg); |
| nuclear@0 | 77 |
| nuclear@0 | 78 bool build(); |
| nuclear@0 | 79 |
| nuclear@0 | 80 bool run() const; |
| nuclear@0 | 81 bool run(int dim, ...) const; |
| nuclear@0 | 82 }; |
| nuclear@0 | 83 |
| nuclear@0 | 84 #endif /* OCL_H_ */ |