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
|
#include "utest_helper.hpp"
#include <cmath>
static int *tmp = NULL;
static struct float4 {float x,y,z,w;} *src = NULL, *dst = NULL;
static int w = 0;
static int h = 0;
static int sz = 0;
static const size_t chunk = 64;
static void compiler_box_blur_float()
{
OCL_CREATE_KERNEL("compiler_box_blur_float");
/* Load the picture */
tmp = cl_read_bmp("sample.bmp", &w, &h);
if(tmp == NULL)
return;
sz = w * h * sizeof(float[4]);
src = (float4*)malloc(sz);
/* RGBA -> float4 conversion */
const int n = w*h;
for (int i = 0; i < n; ++i) {
src[i].x = (float) (tmp[i] & 0xff);
src[i].y = (float) ((tmp[i] >> 8) & 0xff);
src[i].z = (float) ((tmp[i] >> 16) & 0xff);
src[i].w = 0.f;
}
free(tmp);
/* Run the kernel */
OCL_CREATE_BUFFER(buf[0], CL_MEM_COPY_HOST_PTR, sz, src);
OCL_CREATE_BUFFER(buf[1], 0, sz, NULL);
OCL_SET_ARG(0, sizeof(cl_mem), &buf[0]);
OCL_SET_ARG(1, sizeof(cl_mem), &buf[1]);
OCL_SET_ARG(2, sizeof(int), &w);
OCL_SET_ARG(3, sizeof(int), &h);
OCL_SET_ARG(4, sizeof(int), &chunk);
globals[0] = size_t(w);
globals[1] = h/chunk + ((h%chunk)?1:0);
locals[0] = 16;
locals[1] = 1;
free(src);
OCL_NDRANGE(2);
OCL_MAP_BUFFER(1);
dst = (float4*) buf_data[1];
/* Convert back to RGBA and save */
int *tmp = (int*) malloc(n*sizeof(int));
for (int i = 0; i < n; ++i) {
int to = int(std::min(dst[i].x, 255.f));
to |= int(std::min(dst[i].y, 255.f)) << 8;
to |= int(std::min(dst[i].z, 255.f)) << 16;
tmp[i] = to;
}
/* Save the image (for debug purpose) */
cl_write_bmp(tmp, w, h, "compiler_box_blur_float.bmp");
/* Compare with the golden image */
OCL_CHECK_IMAGE(tmp, w, h, "compiler_box_blur_float_ref.bmp");
free(tmp);
}
MAKE_UTEST_FROM_FUNCTION(compiler_box_blur_float);
|
