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
|
#include "pocl_opencl.h"
#define CL_HPP_ENABLE_EXCEPTIONS
#define CL_HPP_MINIMUM_OPENCL_VERSION 120
#define CL_HPP_TARGET_OPENCL_VERSION 120
#include <CL/opencl.hpp>
#include <cassert>
#include <iostream>
using namespace std;
const char *SOURCE = R"RAW(
__kernel void test(__global float* buf){
size_t gid[2];
gid[0] = get_global_id(0);
gid[1] = get_global_id(1);
size_t i, j, k;
float tempResult[3][3][2];
for (i = 0; i < 3; ++i) {
tempResult[i][0][0] = 1.0f;
tempResult[i][0][1] = 1.0f;
tempResult[i][1][0] = 1.0f;
tempResult[i][1][1] = 1.0f;
tempResult[i][2][0] = 1.0f;
tempResult[i][2][1] = 1.0f;
}
if (gid[0] == 0 && gid[1] == 0)
buf[0] = tempResult[1][1][1];
}
)RAW";
#define ARRAY_SIZE 16
int main(int argc, char *argv[]) {
cl::Platform platform = cl::Platform::getDefault();
cl::Device device = cl::Device::getDefault();
float out[ARRAY_SIZE] = {0.0f};
try {
cl::CommandQueue queue = cl::CommandQueue::getDefault();
cl::Program program(SOURCE);
program.build("-cl-std=CL1.2");
cl::Buffer outbuf((cl_mem_flags)(CL_MEM_WRITE_ONLY | CL_MEM_USE_HOST_PTR),
(ARRAY_SIZE * sizeof(float)), out);
auto kernel = cl::KernelFunctor<cl::Buffer>(program, "test");
kernel(cl::EnqueueArgs(queue, cl::NDRange(1, 4), cl::NDRange(1, 4)),
outbuf);
queue.enqueueReadBuffer(outbuf, 1, 0, (ARRAY_SIZE * sizeof(float)), out);
queue.finish();
} catch (cl::Error &err) {
std::cerr << "ERROR: " << err.what() << "(" << err.err() << ")"
<< std::endl;
return EXIT_FAILURE;
}
platform.unloadCompiler();
if (out[0] == 1.0f) {
printf("OK\n");
return EXIT_SUCCESS;
} else {
printf("FAIL\n");
return EXIT_FAILURE;
}
}
|
