kernels/compiler_generic_atomic.cl


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

#define GENERIC_KERNEL(T)                                              \
kernel void compiler_generic_atomic_##T(global T *src, global T *dst)  \
{                                                                      \
  size_t gid = get_global_id(0);                                       \
  size_t lid = get_local_id(0);                                        \
  private T pdata[16];                                                 \
  local T ldata[16];                                                   \
  generic T * p1 = &pdata[lid];                                        \
  generic T * p2 = &ldata[lid];                                        \
  generic T *p = (gid & 1) ? p1 : p2;                                  \
  /* below expression is not supported by clang now   */               \
  /* generic T *p = (gid & 1) ? p1 : (T *)&ldata[lid]; */              \
  *p = src[gid];                                                       \
  /* fill other data */                                                \
  if(gid&1) {                                                          \
    ldata[lid] = 20;                                                   \
  } else {                                                             \
    for (int i = 0; i < 16; i++) {                                     \
      pdata[i] = src[lid];;                                            \
    }                                                                  \
  }                                                                    \
  barrier(CLK_LOCAL_MEM_FENCE);                                        \
                                                                       \
  generic T * q1 = &pdata[lid];                                        \
  generic T * q2 = &ldata[lid];                                        \
  generic T *q = (gid & 1) ? q1 : q2;                                  \
  atomic_fetch_add((atomic_int*)q , pdata[lid]);                       \
  dst[gid] = *q;                                                       \
}

GENERIC_KERNEL(int)
//GENERIC_KERNEL(long)