1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#include "boring_parts.h"
#define RETURN_IF_FAIL(expr) do { \
int res=(expr); \
if ( res != 0 ) return res; \
} while(0)
// TODO : print streamsdk::getOpenCLErrorCodeStr(res)
#define CL_RETURN_VAL_IF_FAIL(val, expr) do { \
cl_int res=(expr); \
if ( res != CL_SUCCESS ) { \
std::cerr << "file " << __FILE__ << ": line " << __LINE__ << " (" << __PRETTY_FUNCTION__ \
<< "): '" << "expr" << "' failed (return code : " << res << ")" << std::endl; \
return val; \
} \
} while(0)
int initLibs() {
// FIXME : unused
#ifdef HAS_OPENCL
RETURN_IF_FAIL( initOpenCL() );
#endif /*HAS_OPENCL*/
return 0;
}
#ifdef HAS_OPENCL
int initOpenCL() {
cl_uint id, numPlatforms, numDevices;
char pbuf[100];
std::string dTypeStr;
cl_platform_id *platforms, platform;
cl_device_id *devices, device;
// Get platform count
CL_RETURN_VAL_IF_FAIL(1,
clGetPlatformIDs(0, NULL, &numPlatforms)
);
std::cout << "Detected " << numPlatforms << " platform(s)" << std::endl;
if ( ! ( numPlatforms > 0 ) ) return 2;
// Allocate room for all platform IDs
platforms = new cl_platform_id[numPlatforms];
// Get platform IDs
CL_RETURN_VAL_IF_FAIL(3,
clGetPlatformIDs(numPlatforms, platforms, &numPlatforms)
);
// Enumerate platforms and grab informations
for(id=0;id<numPlatforms;id++) {
CL_RETURN_VAL_IF_FAIL(4,
clGetPlatformInfo(platforms[id], CL_PLATFORM_VENDOR, sizeof(pbuf), pbuf, NULL)
);
std::cout << "Platform " << id << " : " << pbuf << std::endl;
}
// Take the first one FIXME : do more !
platform = platforms[0];
// Get device count
CL_RETURN_VAL_IF_FAIL(11,
clGetDeviceIDs(platform,CL_DEVICE_TYPE_ALL,0,NULL,&numDevices)
);
std::cout << "\tDetected " << numDevices << " device(s) for platform " << platform << std::endl;
if ( ! ( numDevices > 0 ) ) return 12;
// Allocate room for all devices IDs
devices = new cl_device_id[numDevices];
// Get devices IDs
CL_RETURN_VAL_IF_FAIL(13,
clGetDeviceIDs(platform,CL_DEVICE_TYPE_ALL,numDevices,devices,&numDevices)
);
// Enumerate devices and grab informations
for(id=0;id<numDevices;id++) {
cl_device_type dType;
CL_RETURN_VAL_IF_FAIL(14,
clGetDeviceInfo(devices[id], CL_DEVICE_TYPE, sizeof(dType), &dType, NULL)
);
switch(dType) {
case CL_DEVICE_TYPE_GPU: dTypeStr="gpu"; break;
case CL_DEVICE_TYPE_CPU: dTypeStr="cpu"; break;
default: dTypeStr="(not supported)"; break;
}
std::cout << "\tDevice " << id << " type is " << dTypeStr << std::endl;
}
// Take the first one FIXME : do more !
device=devices[0];
std::cout << "\tWill using device 0 (" << device << ")" << std::endl;
delete [] devices;
delete [] platforms;
return 0;
}
#endif /*HAS_OPENCL*/
|
