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
106
107
108
109
110
111
112
|
#include "utest_helper.hpp"
namespace {
constexpr int n = 16;
// declaration only, we should create each template specification for each type.
template<typename T>
T get_data(int idx, int part);
/* the format of test data is as follows:
* the first column is A
* the second column is B
* the third column is the expected result.
*/
#define DEF_TEMPLATE(TYPE, NAME) \
template <> \
TYPE get_data<TYPE>(int idx, int part) \
{ \
static TYPE test_data[n][3] = { \
{ 0, 0, 0 }, \
{ 0, 1, 1 }, \
{ 0, 2, 2 }, \
{ -1, 1, 0 }, \
{ 1, -2, -1 }, \
{ 0, 110, 110 }, \
{ -10, -10, -20 }, \
{ CL_##NAME##_MIN, CL_##NAME##_MIN, CL_##NAME##_MIN }, \
{ CL_##NAME##_MIN, CL_##NAME##_MAX, -1 }, \
{ CL_##NAME##_MAX, 0, CL_##NAME##_MAX }, \
{ CL_##NAME##_MAX, 1, CL_##NAME##_MAX }, \
{ CL_##NAME##_MAX, 2, CL_##NAME##_MAX }, \
{ CL_##NAME##_MAX, CL_##NAME##_MAX, CL_##NAME##_MAX }, \
{ CL_##NAME##_MAX/2, CL_##NAME##_MAX/2, CL_##NAME##_MAX-1 }, \
{ CL_##NAME##_MAX/2, CL_##NAME##_MAX/2+1, CL_##NAME##_MAX }, \
{ CL_##NAME##_MAX/2+1, CL_##NAME##_MAX/2+1, CL_##NAME##_MAX } \
}; \
return test_data[idx][part]; \
} \
\
template <> \
u##TYPE get_data<u##TYPE>(int idx, int part) \
{ \
static u##TYPE test_data[n][3] = { \
{ 0, 0, 0 }, \
{ CL_U##NAME##_MAX, 0, CL_U##NAME##_MAX }, \
{ CL_U##NAME##_MAX, 1, CL_U##NAME##_MAX }, \
{ CL_U##NAME##_MAX, 2, CL_U##NAME##_MAX }, \
{ CL_U##NAME##_MAX, CL_U##NAME##_MAX, CL_U##NAME##_MAX }, \
{ CL_U##NAME##_MAX/2, CL_U##NAME##_MAX/2, CL_U##NAME##_MAX-1 }, \
{ CL_U##NAME##_MAX/2, CL_U##NAME##_MAX/2+1, CL_U##NAME##_MAX }, \
{ CL_U##NAME##_MAX/2+1, CL_U##NAME##_MAX/2+1, CL_U##NAME##_MAX }\
}; \
return test_data[idx][part]; \
}
DEF_TEMPLATE(int8_t, CHAR)
DEF_TEMPLATE(int16_t, SHRT)
DEF_TEMPLATE(int32_t, INT)
//DEF_TEMPLATE(int64_t, LONG)
template<typename T>
void test(const char *kernel_name)
{
T C[n] = { 0 };
T A[n] = { 0 };
T B[n] = { 0 };
for (int i = 0; i < n; i++) {
A[i] = get_data<T>(i, 0);
B[i] = get_data<T>(i, 1);
}
OCL_CREATE_KERNEL_FROM_FILE("compiler_saturate", kernel_name);
OCL_CREATE_BUFFER(buf[0], CL_MEM_COPY_HOST_PTR, n * sizeof(T), &C[0]);
OCL_CREATE_BUFFER(buf[1], CL_MEM_COPY_HOST_PTR, n * sizeof(T), &A[0]);
OCL_CREATE_BUFFER(buf[2], CL_MEM_COPY_HOST_PTR, n * sizeof(T), &B[0]);
OCL_SET_ARG(0, sizeof(cl_mem), &buf[0]);
OCL_SET_ARG(1, sizeof(cl_mem), &buf[1]);
OCL_SET_ARG(2, sizeof(cl_mem), &buf[2]);
globals[0] = n;
locals[0] = n;
OCL_NDRANGE(1);
OCL_MAP_BUFFER(0);
for (int i = 0; i < n; i++) {
OCL_ASSERT(((T*)buf_data[0])[i] == get_data<T>(i, 2));
}
OCL_UNMAP_BUFFER(0);
}
}
static void compiler_saturate(void)
{
test<int8_t>("test_char");
test<uint8_t>("test_uchar");
test<int16_t>("test_short");
test<uint16_t>("test_ushort");
test<int32_t>("test_int");
test<uint32_t>("test_uint");
// test<int64_t>("test_long");
// test<uint64_t>("test_ulong");
}
MAKE_UTEST_FROM_FUNCTION(compiler_saturate);
|
