1#include <clc/clc.h>
2
3#pragma OPENCL EXTENSION cl_khr_byte_addressable_store : enable
4
5#define VSTORE_VECTORIZE(PRIM_TYPE, ADDR_SPACE) \
6  typedef PRIM_TYPE##2 less_aligned_##ADDR_SPACE##PRIM_TYPE##2 __attribute__ ((aligned (sizeof(PRIM_TYPE))));\
7  _CLC_OVERLOAD _CLC_DEF void vstore2(PRIM_TYPE##2 vec, size_t offset, ADDR_SPACE PRIM_TYPE *mem) { \
8    *((ADDR_SPACE less_aligned_##ADDR_SPACE##PRIM_TYPE##2*) (&mem[2*offset])) = vec; \
9  } \
10\
11  _CLC_OVERLOAD _CLC_DEF void vstore3(PRIM_TYPE##3 vec, size_t offset, ADDR_SPACE PRIM_TYPE *mem) { \
12    *((ADDR_SPACE less_aligned_##ADDR_SPACE##PRIM_TYPE##2*) (&mem[3*offset])) = (PRIM_TYPE##2)(vec.s0, vec.s1); \
13    mem[3 * offset + 2] = vec.s2;\
14  } \
15\
16  typedef PRIM_TYPE##4 less_aligned_##ADDR_SPACE##PRIM_TYPE##4 __attribute__ ((aligned (sizeof(PRIM_TYPE))));\
17  _CLC_OVERLOAD _CLC_DEF void vstore4(PRIM_TYPE##4 vec, size_t offset, ADDR_SPACE PRIM_TYPE *mem) { \
18    *((ADDR_SPACE less_aligned_##ADDR_SPACE##PRIM_TYPE##4*) (&mem[4*offset])) = vec; \
19  } \
20\
21  typedef PRIM_TYPE##8 less_aligned_##ADDR_SPACE##PRIM_TYPE##8 __attribute__ ((aligned (sizeof(PRIM_TYPE))));\
22  _CLC_OVERLOAD _CLC_DEF void vstore8(PRIM_TYPE##8 vec, size_t offset, ADDR_SPACE PRIM_TYPE *mem) { \
23    *((ADDR_SPACE less_aligned_##ADDR_SPACE##PRIM_TYPE##8*) (&mem[8*offset])) = vec; \
24  } \
25\
26  typedef PRIM_TYPE##16 less_aligned_##ADDR_SPACE##PRIM_TYPE##16 __attribute__ ((aligned (sizeof(PRIM_TYPE))));\
27  _CLC_OVERLOAD _CLC_DEF void vstore16(PRIM_TYPE##16 vec, size_t offset, ADDR_SPACE PRIM_TYPE *mem) { \
28    *((ADDR_SPACE less_aligned_##ADDR_SPACE##PRIM_TYPE##16*) (&mem[16*offset])) = vec; \
29  } \
30
31#define VSTORE_ADDR_SPACES(__CLC_SCALAR___CLC_GENTYPE) \
32    VSTORE_VECTORIZE(__CLC_SCALAR___CLC_GENTYPE, __private) \
33    VSTORE_VECTORIZE(__CLC_SCALAR___CLC_GENTYPE, __local) \
34    VSTORE_VECTORIZE(__CLC_SCALAR___CLC_GENTYPE, __global) \
35
36#define VSTORE_TYPES() \
37    VSTORE_ADDR_SPACES(char) \
38    VSTORE_ADDR_SPACES(uchar) \
39    VSTORE_ADDR_SPACES(short) \
40    VSTORE_ADDR_SPACES(ushort) \
41    VSTORE_ADDR_SPACES(int) \
42    VSTORE_ADDR_SPACES(uint) \
43    VSTORE_ADDR_SPACES(long) \
44    VSTORE_ADDR_SPACES(ulong) \
45    VSTORE_ADDR_SPACES(float) \
46
47VSTORE_TYPES()
48
49#ifdef cl_khr_fp64
50#pragma OPENCL EXTENSION cl_khr_fp64 : enable
51    VSTORE_ADDR_SPACES(double)
52#endif
53
54/* vstore_half are legal even without cl_khr_fp16 */
55#define DECLARE_HELPER(STYPE, AS) void __clc_vstore_half_##STYPE##_helper##AS(STYPE, AS half *);
56
57DECLARE_HELPER(float, __private);
58DECLARE_HELPER(float, __global);
59DECLARE_HELPER(float, __local);
60
61#ifdef cl_khr_fp64
62#pragma OPENCL EXTENSION cl_khr_fp64 : enable
63DECLARE_HELPER(double, __private);
64DECLARE_HELPER(double, __global);
65DECLARE_HELPER(double, __local);
66#endif
67
68
69#define VEC_STORE1(STYPE, AS, val) __clc_vstore_half_##STYPE##_helper##AS (val, &mem[offset++]);
70#define VEC_STORE2(STYPE, AS, val) \
71	VEC_STORE1(STYPE, AS, val.lo) \
72	VEC_STORE1(STYPE, AS, val.hi)
73#define VEC_STORE3(STYPE, AS, val) \
74	VEC_STORE1(STYPE, AS, val.s0) \
75	VEC_STORE1(STYPE, AS, val.s1) \
76	VEC_STORE1(STYPE, AS, val.s2)
77#define VEC_STORE4(STYPE, AS, val) \
78	VEC_STORE2(STYPE, AS, val.lo) \
79	VEC_STORE2(STYPE, AS, val.hi)
80#define VEC_STORE8(STYPE, AS, val) \
81	VEC_STORE4(STYPE, AS, val.lo) \
82	VEC_STORE4(STYPE, AS, val.hi)
83#define VEC_STORE16(STYPE, AS, val) \
84	VEC_STORE8(STYPE, AS, val.lo) \
85	VEC_STORE8(STYPE, AS, val.hi)
86
87#define __FUNC(SUFFIX, VEC_SIZE, TYPE, STYPE, AS) \
88  _CLC_OVERLOAD _CLC_DEF void vstore_half##SUFFIX(TYPE vec, size_t offset, AS half *mem) { \
89    offset *= VEC_SIZE; \
90    VEC_STORE##VEC_SIZE(STYPE, AS, vec) \
91  }
92
93#define FUNC(SUFFIX, VEC_SIZE, TYPE, STYPE, AS) __FUNC(SUFFIX, VEC_SIZE, TYPE, STYPE, AS)
94
95#define __CLC_BODY "vstore_half.inc"
96#include <clc/math/gentype.inc>
97
98