rev |
line source |
nuclear@8
|
1 #define OCL_CC_
|
nuclear@8
|
2
|
nuclear@0
|
3 #include <stdio.h>
|
nuclear@0
|
4 #include <stdlib.h>
|
nuclear@0
|
5 #include <string.h>
|
nuclear@8
|
6 #include <stdarg.h>
|
nuclear@0
|
7 #include <errno.h>
|
John@11
|
8 #ifndef _MSC_VER
|
nuclear@0
|
9 #include <alloca.h>
|
John@11
|
10 #else
|
John@11
|
11 #include <malloc.h>
|
John@11
|
12 #endif
|
nuclear@0
|
13 #include <sys/stat.h>
|
nuclear@0
|
14 #include "ocl.h"
|
nuclear@8
|
15 #include "ocl_errstr.h"
|
nuclear@0
|
16
|
nuclear@0
|
17
|
nuclear@0
|
18 class InitCL {
|
nuclear@0
|
19 public:
|
nuclear@0
|
20 InitCL();
|
nuclear@0
|
21 };
|
nuclear@0
|
22
|
nuclear@0
|
23 struct device_info {
|
nuclear@0
|
24 cl_device_id id;
|
nuclear@0
|
25 cl_device_type type;
|
nuclear@0
|
26 unsigned int units;
|
nuclear@0
|
27 unsigned int clock;
|
nuclear@0
|
28
|
nuclear@0
|
29 unsigned int dim;
|
nuclear@0
|
30 size_t *work_item_sizes;
|
nuclear@0
|
31 size_t work_group_size;
|
nuclear@0
|
32
|
nuclear@0
|
33 unsigned long mem_size;
|
nuclear@0
|
34 };
|
nuclear@0
|
35
|
nuclear@0
|
36 static bool init_opencl(void);
|
nuclear@0
|
37 static int select_device(struct device_info *di, int (*devcmp)(struct device_info*, struct device_info*));
|
nuclear@0
|
38 static int get_dev_info(cl_device_id dev, struct device_info *di);
|
nuclear@0
|
39 static int devcmp(struct device_info *a, struct device_info *b);
|
nuclear@0
|
40 static const char *devtypestr(cl_device_type type);
|
nuclear@0
|
41 static void print_memsize(FILE *out, unsigned long memsz);
|
nuclear@8
|
42 static const char *clstrerror(int err);
|
nuclear@0
|
43
|
nuclear@0
|
44
|
nuclear@0
|
45 static InitCL initcl;
|
nuclear@0
|
46 static cl_context ctx;
|
nuclear@0
|
47 static cl_command_queue cmdq;
|
nuclear@0
|
48 static device_info devinf;
|
nuclear@0
|
49
|
nuclear@0
|
50 InitCL::InitCL()
|
nuclear@0
|
51 {
|
nuclear@0
|
52 if(!init_opencl()) {
|
nuclear@0
|
53 exit(0);
|
nuclear@0
|
54 }
|
nuclear@0
|
55 }
|
nuclear@0
|
56
|
nuclear@0
|
57 static bool init_opencl(void)
|
nuclear@0
|
58 {
|
nuclear@0
|
59 if(select_device(&devinf, devcmp) == -1) {
|
nuclear@0
|
60 return false;
|
nuclear@0
|
61 }
|
nuclear@0
|
62
|
nuclear@0
|
63
|
nuclear@0
|
64 if(!(ctx = clCreateContext(0, 1, &devinf.id, 0, 0, 0))) {
|
nuclear@0
|
65 fprintf(stderr, "failed to create opencl context\n");
|
nuclear@0
|
66 return false;
|
nuclear@0
|
67 }
|
nuclear@0
|
68
|
nuclear@0
|
69 if(!(cmdq = clCreateCommandQueue(ctx, devinf.id, 0, 0))) {
|
nuclear@0
|
70 fprintf(stderr, "failed to create command queue\n");
|
nuclear@0
|
71 return false;
|
nuclear@0
|
72 }
|
nuclear@0
|
73 return true;
|
nuclear@0
|
74 }
|
nuclear@0
|
75
|
nuclear@0
|
76
|
nuclear@0
|
77 CLMemBuffer *create_mem_buffer(int rdwr, size_t sz, void *buf)
|
nuclear@0
|
78 {
|
nuclear@0
|
79 int err;
|
nuclear@0
|
80 cl_mem mem;
|
nuclear@12
|
81 cl_mem_flags flags = rdwr | CL_MEM_ALLOC_HOST_PTR;
|
nuclear@0
|
82
|
nuclear@12
|
83 if(buf) {
|
nuclear@12
|
84 flags |= CL_MEM_COPY_HOST_PTR;
|
nuclear@12
|
85 }
|
nuclear@0
|
86
|
nuclear@12
|
87
|
nuclear@12
|
88 if(!(mem = clCreateBuffer(ctx, flags, sz, buf, &err))) {
|
nuclear@8
|
89 fprintf(stderr, "failed to create memory buffer: %s\n", clstrerror(err));
|
nuclear@0
|
90 return 0;
|
nuclear@0
|
91 }
|
nuclear@0
|
92
|
nuclear@0
|
93 CLMemBuffer *mbuf = new CLMemBuffer;
|
nuclear@0
|
94 mbuf->mem = mem;
|
nuclear@0
|
95 mbuf->size = sz;
|
nuclear@12
|
96 mbuf->ptr = 0;
|
nuclear@0
|
97 return mbuf;
|
nuclear@0
|
98 }
|
nuclear@0
|
99
|
nuclear@0
|
100 void destroy_mem_buffer(CLMemBuffer *mbuf)
|
nuclear@0
|
101 {
|
nuclear@0
|
102 if(mbuf) {
|
nuclear@0
|
103 clReleaseMemObject(mbuf->mem);
|
nuclear@0
|
104 delete mbuf;
|
nuclear@0
|
105 }
|
nuclear@0
|
106 }
|
nuclear@0
|
107
|
nuclear@0
|
108 void *map_mem_buffer(CLMemBuffer *mbuf, int rdwr)
|
nuclear@0
|
109 {
|
nuclear@0
|
110 if(!mbuf) return 0;
|
nuclear@0
|
111
|
nuclear@12
|
112 #ifndef NDEBUG
|
nuclear@12
|
113 if(mbuf->ptr) {
|
nuclear@12
|
114 fprintf(stderr, "WARNING: map_mem_buffer called on already mapped buffer\n");
|
nuclear@12
|
115 }
|
nuclear@12
|
116 #endif
|
nuclear@12
|
117
|
nuclear@0
|
118 int err;
|
nuclear@0
|
119 mbuf->ptr = clEnqueueMapBuffer(cmdq, mbuf->mem, 1, rdwr, 0, mbuf->size, 0, 0, 0, &err);
|
nuclear@0
|
120 if(!mbuf->ptr) {
|
nuclear@8
|
121 fprintf(stderr, "failed to map buffer: %s\n", clstrerror(err));
|
nuclear@0
|
122 return 0;
|
nuclear@0
|
123 }
|
nuclear@0
|
124 return mbuf->ptr;
|
nuclear@0
|
125 }
|
nuclear@0
|
126
|
nuclear@0
|
127 void unmap_mem_buffer(CLMemBuffer *mbuf)
|
nuclear@0
|
128 {
|
nuclear@0
|
129 if(!mbuf || !mbuf->ptr) return;
|
nuclear@0
|
130 clEnqueueUnmapMemObject(cmdq, mbuf->mem, mbuf->ptr, 0, 0, 0);
|
nuclear@12
|
131 mbuf->ptr = 0;
|
nuclear@0
|
132 }
|
nuclear@0
|
133
|
nuclear@0
|
134 bool write_mem_buffer(CLMemBuffer *mbuf, size_t sz, void *src)
|
nuclear@0
|
135 {
|
nuclear@0
|
136 if(!mbuf) return false;
|
nuclear@0
|
137
|
nuclear@0
|
138 int err;
|
nuclear@0
|
139 if((err = clEnqueueWriteBuffer(cmdq, mbuf->mem, 1, 0, sz, src, 0, 0, 0)) != 0) {
|
nuclear@8
|
140 fprintf(stderr, "failed to write buffer: %s\n", clstrerror(err));
|
nuclear@0
|
141 return false;
|
nuclear@0
|
142 }
|
nuclear@0
|
143 return true;
|
nuclear@0
|
144 }
|
nuclear@0
|
145
|
nuclear@0
|
146 bool read_mem_buffer(CLMemBuffer *mbuf, size_t sz, void *dest)
|
nuclear@0
|
147 {
|
nuclear@0
|
148 if(!mbuf) return false;
|
nuclear@0
|
149
|
nuclear@0
|
150 int err;
|
nuclear@0
|
151 if((err = clEnqueueReadBuffer(cmdq, mbuf->mem, 1, 0, sz, dest, 0, 0, 0)) != 0) {
|
nuclear@8
|
152 fprintf(stderr, "failed to read buffer: %s\n", clstrerror(err));
|
nuclear@0
|
153 return false;
|
nuclear@0
|
154 }
|
nuclear@0
|
155 return true;
|
nuclear@0
|
156 }
|
nuclear@0
|
157
|
nuclear@0
|
158
|
nuclear@0
|
159 CLProgram::CLProgram(const char *kname)
|
nuclear@0
|
160 {
|
nuclear@0
|
161 prog = 0;
|
nuclear@0
|
162 kernel = 0;
|
nuclear@0
|
163 this->kname = kname;
|
nuclear@1
|
164 args.resize(16);
|
nuclear@0
|
165 built = false;
|
nuclear@0
|
166 }
|
nuclear@0
|
167
|
nuclear@0
|
168 CLProgram::~CLProgram()
|
nuclear@0
|
169 {
|
nuclear@0
|
170 if(prog) {
|
nuclear@0
|
171
|
nuclear@0
|
172 clReleaseProgram(prog);
|
nuclear@0
|
173 }
|
nuclear@0
|
174 if(kernel) {
|
nuclear@0
|
175
|
nuclear@0
|
176 clReleaseKernel(kernel);
|
nuclear@0
|
177 }
|
nuclear@1
|
178 for(size_t i=0; i<args.size(); i++) {
|
nuclear@1
|
179 if(args[i].type == ARGTYPE_MEM_BUF) {
|
nuclear@1
|
180 destroy_mem_buffer(args[i].v.mbuf);
|
nuclear@0
|
181 }
|
nuclear@0
|
182 }
|
nuclear@0
|
183 }
|
nuclear@0
|
184
|
nuclear@0
|
185 bool CLProgram::load(const char *fname)
|
nuclear@0
|
186 {
|
nuclear@0
|
187 FILE *fp;
|
nuclear@0
|
188 char *src;
|
nuclear@0
|
189 struct stat st;
|
nuclear@0
|
190
|
nuclear@0
|
191 printf("loading opencl program (%s)\n", fname);
|
nuclear@0
|
192
|
nuclear@0
|
193 if(!(fp = fopen(fname, "rb"))) {
|
nuclear@0
|
194 fprintf(stderr, "failed to open %s: %s\n", fname, strerror(errno));
|
nuclear@0
|
195 return false;
|
nuclear@0
|
196 }
|
nuclear@0
|
197
|
nuclear@0
|
198 fstat(fileno(fp), &st);
|
nuclear@0
|
199
|
nuclear@0
|
200 src = new char[st.st_size + 1];
|
nuclear@0
|
201
|
nuclear@0
|
202 fread(src, 1, st.st_size, fp);
|
nuclear@0
|
203 src[st.st_size] = 0;
|
nuclear@0
|
204 fclose(fp);
|
nuclear@0
|
205
|
nuclear@0
|
206
|
nuclear@0
|
207 if(!(prog = clCreateProgramWithSource(ctx, 1, (const char**)&src, 0, 0))) {
|
nuclear@0
|
208 fprintf(stderr, "error creating program object: %s\n", fname);
|
nuclear@0
|
209 delete [] src;
|
nuclear@0
|
210 return false;
|
nuclear@0
|
211 }
|
nuclear@0
|
212 delete [] src;
|
nuclear@0
|
213 return true;
|
nuclear@0
|
214 }
|
nuclear@0
|
215
|
nuclear@1
|
216 bool CLProgram::set_argi(int idx, int val)
|
nuclear@1
|
217 {
|
nuclear@1
|
218 if((int)args.size() <= idx) {
|
nuclear@1
|
219 args.resize(idx + 1);
|
nuclear@1
|
220 }
|
nuclear@1
|
221
|
nuclear@1
|
222 CLArg *arg = &args[idx];
|
nuclear@1
|
223 arg->type = ARGTYPE_INT;
|
nuclear@1
|
224 arg->v.ival = val;
|
nuclear@1
|
225 return true;
|
nuclear@1
|
226 }
|
nuclear@1
|
227
|
nuclear@1
|
228 bool CLProgram::set_argf(int idx, float val)
|
nuclear@1
|
229 {
|
nuclear@1
|
230 if((int)args.size() <= idx) {
|
nuclear@1
|
231 args.resize(idx + 1);
|
nuclear@1
|
232 }
|
nuclear@1
|
233
|
nuclear@1
|
234 CLArg *arg = &args[idx];
|
nuclear@1
|
235 arg->type = ARGTYPE_FLOAT;
|
nuclear@1
|
236 arg->v.fval = val;
|
nuclear@1
|
237 return true;
|
nuclear@1
|
238 }
|
nuclear@1
|
239
|
nuclear@1
|
240 bool CLProgram::set_arg_buffer(int idx, int rdwr, size_t sz, void *ptr)
|
nuclear@0
|
241 {
|
nuclear@13
|
242 printf("create argument %d buffer: %d bytes\n", idx, (int)sz);
|
nuclear@0
|
243 CLMemBuffer *buf;
|
nuclear@0
|
244
|
nuclear@0
|
245 if(!(buf = create_mem_buffer(rdwr, sz, ptr))) {
|
nuclear@0
|
246 return false;
|
nuclear@0
|
247 }
|
nuclear@0
|
248
|
nuclear@1
|
249 if((int)args.size() <= idx) {
|
nuclear@1
|
250 args.resize(idx + 1);
|
nuclear@0
|
251 }
|
nuclear@1
|
252 args[idx].type = ARGTYPE_MEM_BUF;
|
nuclear@1
|
253 args[idx].v.mbuf = buf;
|
nuclear@0
|
254 return true;
|
nuclear@0
|
255 }
|
nuclear@0
|
256
|
nuclear@0
|
257 CLMemBuffer *CLProgram::get_arg_buffer(int arg)
|
nuclear@0
|
258 {
|
nuclear@1
|
259 if(arg < 0 || arg >= (int)args.size() || args[arg].type != ARGTYPE_MEM_BUF) {
|
nuclear@0
|
260 return 0;
|
nuclear@0
|
261 }
|
nuclear@1
|
262 return args[arg].v.mbuf;
|
nuclear@0
|
263 }
|
nuclear@0
|
264
|
nuclear@0
|
265 bool CLProgram::build()
|
nuclear@0
|
266 {
|
nuclear@2
|
267 int err;
|
nuclear@0
|
268
|
nuclear@2
|
269 if((err = clBuildProgram(prog, 0, 0, 0, 0, 0)) != 0) {
|
nuclear@2
|
270 size_t sz;
|
nuclear@2
|
271 clGetProgramBuildInfo(prog, devinf.id, CL_PROGRAM_BUILD_LOG, 0, 0, &sz);
|
nuclear@0
|
272
|
nuclear@2
|
273 char *errlog = (char*)alloca(sz + 1);
|
nuclear@2
|
274 clGetProgramBuildInfo(prog, devinf.id, CL_PROGRAM_BUILD_LOG, sz, errlog, 0);
|
nuclear@8
|
275 fprintf(stderr, "failed to build program: %s\n%s\n", clstrerror(err), errlog);
|
nuclear@2
|
276
|
nuclear@0
|
277 clReleaseProgram(prog);
|
nuclear@0
|
278 prog = 0;
|
nuclear@0
|
279 return false;
|
nuclear@0
|
280 }
|
nuclear@0
|
281
|
nuclear@0
|
282
|
nuclear@0
|
283 if(!(kernel = clCreateKernel(prog, kname.c_str(), 0))) {
|
nuclear@0
|
284 fprintf(stderr, "failed to create kernel: %s\n", kname.c_str());
|
nuclear@0
|
285 clReleaseProgram(prog);
|
nuclear@0
|
286 prog = 0;
|
nuclear@0
|
287 return false;
|
nuclear@0
|
288 }
|
nuclear@0
|
289
|
nuclear@1
|
290 for(size_t i=0; i<args.size(); i++) {
|
nuclear@1
|
291 int err;
|
nuclear@0
|
292
|
nuclear@1
|
293 if(args[i].type == ARGTYPE_NONE) {
|
nuclear@1
|
294 break;
|
nuclear@1
|
295 }
|
nuclear@1
|
296
|
nuclear@1
|
297 switch(args[i].type) {
|
nuclear@1
|
298 case ARGTYPE_INT:
|
nuclear@1
|
299 if((err = clSetKernelArg(kernel, i, sizeof(int), &args[i].v.ival)) != 0) {
|
nuclear@8
|
300 fprintf(stderr, "failed to bind kernel argument %d: %s\n", (int)i, clstrerror(err));
|
nuclear@1
|
301 goto fail;
|
nuclear@1
|
302 }
|
nuclear@1
|
303 break;
|
nuclear@1
|
304
|
nuclear@1
|
305 case ARGTYPE_FLOAT:
|
nuclear@1
|
306 if((err = clSetKernelArg(kernel, i, sizeof(float), &args[i].v.fval)) != 0) {
|
nuclear@8
|
307 fprintf(stderr, "failed to bind kernel argument %d: %s\n", (int)i, clstrerror(err));
|
nuclear@1
|
308 goto fail;
|
nuclear@1
|
309 }
|
nuclear@1
|
310 break;
|
nuclear@1
|
311
|
nuclear@1
|
312 case ARGTYPE_MEM_BUF:
|
nuclear@1
|
313 {
|
nuclear@1
|
314 CLMemBuffer *mbuf = args[i].v.mbuf;
|
nuclear@1
|
315
|
nuclear@1
|
316 if((err = clSetKernelArg(kernel, i, sizeof mbuf->mem, &mbuf->mem)) != 0) {
|
nuclear@8
|
317 fprintf(stderr, "failed to bind kernel argument %d: %s\n", (int)i, clstrerror(err));
|
nuclear@1
|
318 goto fail;
|
nuclear@1
|
319 }
|
nuclear@1
|
320 }
|
nuclear@1
|
321 break;
|
nuclear@1
|
322
|
nuclear@1
|
323 default:
|
nuclear@1
|
324 break;
|
nuclear@0
|
325 }
|
nuclear@0
|
326 }
|
nuclear@0
|
327
|
nuclear@0
|
328 built = true;
|
nuclear@0
|
329 return true;
|
nuclear@1
|
330
|
nuclear@1
|
331 fail:
|
nuclear@1
|
332 clReleaseProgram(prog);
|
nuclear@1
|
333 clReleaseKernel(kernel);
|
nuclear@1
|
334 prog = 0;
|
nuclear@1
|
335 kernel = 0;
|
nuclear@1
|
336 return false;
|
nuclear@0
|
337 }
|
nuclear@0
|
338
|
nuclear@0
|
339 bool CLProgram::run() const
|
nuclear@0
|
340 {
|
nuclear@0
|
341 return run(1, 1);
|
nuclear@0
|
342 }
|
nuclear@0
|
343
|
nuclear@0
|
344 bool CLProgram::run(int dim, ...) const
|
nuclear@0
|
345 {
|
nuclear@0
|
346 if(!built) {
|
nuclear@0
|
347 if(!((CLProgram*)this)->build()) {
|
nuclear@0
|
348 return false;
|
nuclear@0
|
349 }
|
nuclear@0
|
350 }
|
nuclear@0
|
351
|
nuclear@0
|
352 va_list ap;
|
nuclear@0
|
353 size_t *global_size = (size_t*)alloca(dim * sizeof *global_size);
|
nuclear@0
|
354
|
nuclear@0
|
355 va_start(ap, dim);
|
nuclear@0
|
356 for(int i=0; i<dim; i++) {
|
nuclear@0
|
357 global_size[i] = va_arg(ap, int);
|
nuclear@0
|
358 }
|
nuclear@0
|
359 va_end(ap);
|
nuclear@0
|
360
|
nuclear@0
|
361 int err;
|
nuclear@0
|
362 if((err = clEnqueueNDRangeKernel(cmdq, kernel, dim, 0, global_size, 0, 0, 0, 0)) != 0) {
|
nuclear@8
|
363 fprintf(stderr, "error executing kernel: %s\n", clstrerror(err));
|
nuclear@0
|
364 return false;
|
nuclear@0
|
365 }
|
nuclear@0
|
366 return true;
|
nuclear@0
|
367 }
|
nuclear@0
|
368
|
nuclear@0
|
369 static int select_device(struct device_info *dev_inf, int (*devcmp)(struct device_info*, struct device_info*))
|
nuclear@0
|
370 {
|
nuclear@8
|
371 unsigned int i, j, num_dev, num_plat, sel, ret;
|
nuclear@0
|
372 cl_device_id dev[32];
|
nuclear@8
|
373 cl_platform_id plat[32];
|
nuclear@0
|
374
|
nuclear@0
|
375 dev_inf->work_item_sizes = 0;
|
nuclear@0
|
376
|
nuclear@8
|
377 if((ret = clGetPlatformIDs(32, plat, &num_plat)) != 0) {
|
nuclear@8
|
378 fprintf(stderr, "clGetPlatformIDs failed: %s\n", clstrerror(ret));
|
nuclear@8
|
379 return -1;
|
nuclear@8
|
380 }
|
nuclear@8
|
381 if(!num_plat) {
|
nuclear@8
|
382 fprintf(stderr, "OpenCL not available!\n");
|
nuclear@8
|
383 return -1;
|
nuclear@8
|
384 }
|
nuclear@0
|
385
|
nuclear@8
|
386 for(i=0; i<num_plat; i++) {
|
nuclear@8
|
387 char buf[512];
|
nuclear@8
|
388
|
nuclear@8
|
389 clGetPlatformInfo(plat[i], CL_PLATFORM_NAME, sizeof buf, buf, 0);
|
nuclear@8
|
390 printf("[%d]: %s", i, buf);
|
nuclear@8
|
391 clGetPlatformInfo(plat[i], CL_PLATFORM_VENDOR, sizeof buf, buf, 0);
|
nuclear@8
|
392 printf(", %s", buf);
|
nuclear@8
|
393 clGetPlatformInfo(plat[i], CL_PLATFORM_VERSION, sizeof buf, buf, 0);
|
nuclear@8
|
394 printf(" (%s)\n", buf);
|
nuclear@8
|
395 }
|
nuclear@8
|
396
|
nuclear@8
|
397 if((ret = clGetDeviceIDs(plat[0], CL_DEVICE_TYPE_ALL, 32, dev, &num_dev)) != 0) {
|
nuclear@8
|
398 fprintf(stderr, "clGetDeviceIDs failed: %s\n", clstrerror(ret));
|
nuclear@8
|
399 return -1;
|
nuclear@8
|
400 }
|
nuclear@0
|
401 printf("found %d cl devices.\n", num_dev);
|
nuclear@0
|
402
|
nuclear@0
|
403 for(i=0; i<num_dev; i++) {
|
nuclear@0
|
404 struct device_info di;
|
nuclear@0
|
405
|
nuclear@0
|
406 if(get_dev_info(dev[i], &di) == -1) {
|
nuclear@0
|
407 free(dev_inf->work_item_sizes);
|
nuclear@0
|
408 return -1;
|
nuclear@0
|
409 }
|
nuclear@0
|
410
|
nuclear@0
|
411 printf("--> device %u (%s)\n", i, devtypestr(di.type));
|
nuclear@0
|
412 printf("max compute units: %u\n", di.units);
|
nuclear@0
|
413 printf("max clock frequency: %u\n", di.clock);
|
nuclear@0
|
414 printf("max work item dimensions: %u\n", di.dim);
|
nuclear@0
|
415
|
nuclear@0
|
416 printf("max work item sizes: ");
|
nuclear@0
|
417 for(j=0; j<di.dim; j++) {
|
nuclear@0
|
418 printf("%u", (unsigned int)di.work_item_sizes[j]);
|
nuclear@0
|
419 if(di.dim - j > 1) {
|
nuclear@0
|
420 printf(", ");
|
nuclear@0
|
421 }
|
nuclear@0
|
422 }
|
nuclear@0
|
423 putchar('\n');
|
nuclear@0
|
424
|
nuclear@0
|
425 printf("max work group size: %u\n", (unsigned int)di.work_group_size);
|
nuclear@0
|
426 printf("max object allocation size: ");
|
nuclear@0
|
427 print_memsize(stdout, di.mem_size);
|
nuclear@0
|
428 putchar('\n');
|
nuclear@0
|
429
|
nuclear@0
|
430 if(devcmp(&di, dev_inf) > 0) {
|
nuclear@0
|
431 free(dev_inf->work_item_sizes);
|
nuclear@0
|
432 memcpy(dev_inf, &di, sizeof di);
|
nuclear@0
|
433 sel = i;
|
nuclear@0
|
434 }
|
nuclear@0
|
435 }
|
nuclear@0
|
436
|
nuclear@0
|
437 if(num_dev) {
|
nuclear@0
|
438 printf("\nusing device: %d\n", sel);
|
nuclear@0
|
439 return 0;
|
nuclear@0
|
440 }
|
nuclear@0
|
441
|
nuclear@0
|
442 return -1;
|
nuclear@0
|
443 }
|
nuclear@0
|
444
|
nuclear@0
|
445 static int get_dev_info(cl_device_id dev, struct device_info *di)
|
nuclear@0
|
446 {
|
nuclear@0
|
447 di->id = dev;
|
nuclear@0
|
448
|
nuclear@0
|
449
|
nuclear@0
|
450 clGetDeviceInfo(dev, CL_DEVICE_TYPE, sizeof di->type, &di->type, 0);
|
nuclear@0
|
451 clGetDeviceInfo(dev, CL_DEVICE_MAX_COMPUTE_UNITS, sizeof di->units, &di->units, 0);
|
nuclear@0
|
452 clGetDeviceInfo(dev, CL_DEVICE_MAX_CLOCK_FREQUENCY, sizeof di->clock, &di->clock, 0);
|
nuclear@0
|
453 clGetDeviceInfo(dev, CL_DEVICE_MAX_WORK_ITEM_DIMENSIONS, sizeof di->dim, &di->dim, 0);
|
nuclear@0
|
454
|
nuclear@0
|
455 di->work_item_sizes = new size_t[di->dim];
|
nuclear@0
|
456
|
nuclear@0
|
457 clGetDeviceInfo(dev, CL_DEVICE_MAX_WORK_ITEM_SIZES, di->dim * sizeof *di->work_item_sizes, di->work_item_sizes, 0);
|
nuclear@0
|
458 clGetDeviceInfo(dev, CL_DEVICE_MAX_WORK_GROUP_SIZE, sizeof di->work_group_size, &di->work_group_size, 0);
|
nuclear@0
|
459 clGetDeviceInfo(dev, CL_DEVICE_MAX_MEM_ALLOC_SIZE, sizeof di->mem_size, &di->mem_size, 0);
|
nuclear@0
|
460
|
nuclear@0
|
461 return 0;
|
nuclear@0
|
462 }
|
nuclear@0
|
463
|
nuclear@0
|
464 static int devcmp(struct device_info *a, struct device_info *b)
|
nuclear@0
|
465 {
|
nuclear@0
|
466 unsigned int aval = a->units * a->clock;
|
nuclear@0
|
467 unsigned int bval = b->units * b->clock;
|
nuclear@0
|
468
|
nuclear@0
|
469 return aval - bval;
|
nuclear@0
|
470 }
|
nuclear@0
|
471
|
nuclear@0
|
472 static const char *devtypestr(cl_device_type type)
|
nuclear@0
|
473 {
|
nuclear@0
|
474 switch(type) {
|
nuclear@0
|
475 case CL_DEVICE_TYPE_CPU:
|
nuclear@0
|
476 return "cpu";
|
nuclear@0
|
477 case CL_DEVICE_TYPE_GPU:
|
nuclear@0
|
478 return "gpu";
|
nuclear@0
|
479 case CL_DEVICE_TYPE_ACCELERATOR:
|
nuclear@0
|
480 return "accelerator";
|
nuclear@0
|
481 default:
|
nuclear@0
|
482 break;
|
nuclear@0
|
483 }
|
nuclear@0
|
484 return "unknown";
|
nuclear@0
|
485 }
|
nuclear@0
|
486
|
nuclear@0
|
487 static void print_memsize(FILE *out, unsigned long bytes)
|
nuclear@0
|
488 {
|
nuclear@0
|
489 int i;
|
nuclear@0
|
490 unsigned long memsz = bytes;
|
nuclear@0
|
491 const char *suffix[] = {"bytes", "kb", "mb", "gb", "tb", "pb", 0};
|
nuclear@0
|
492
|
nuclear@0
|
493 for(i=0; suffix[i]; i++) {
|
nuclear@0
|
494 if(memsz < 1024) {
|
nuclear@0
|
495 fprintf(out, "%lu %s", memsz, suffix[i]);
|
nuclear@0
|
496 if(i > 0) {
|
nuclear@0
|
497 fprintf(out, " (%lu bytes)", bytes);
|
nuclear@0
|
498 }
|
nuclear@0
|
499 return;
|
nuclear@0
|
500 }
|
nuclear@0
|
501
|
nuclear@0
|
502 memsz /= 1024;
|
nuclear@0
|
503 }
|
nuclear@0
|
504 }
|
nuclear@8
|
505
|
nuclear@8
|
506 static const char *clstrerror(int err)
|
nuclear@8
|
507 {
|
nuclear@8
|
508 if(err > 0) {
|
nuclear@8
|
509 return "<invalid error code>";
|
nuclear@8
|
510 }
|
nuclear@8
|
511 if(err <= -(int)(sizeof ocl_errstr / sizeof *ocl_errstr)) {
|
nuclear@8
|
512 return "<unknown error>";
|
nuclear@8
|
513 }
|
nuclear@8
|
514 return ocl_errstr[-err];
|
nuclear@8
|
515 }
|