| Aaron Watry | 50a7bcb | 2013-09-05 16:04:01 +0000 | [diff] [blame] | 1 | #include <clc/clc.h> |
| 2 | |
| Aaron Watry | ea32a57 | 2014-09-16 22:34:53 +0000 | [diff] [blame] | 3 | #define ATOMIC_FUNC_DEFINE(RET_SIGN, ARG_SIGN, TYPE, CL_FUNCTION, CLC_FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 4 | _CLC_OVERLOAD _CLC_DEF RET_SIGN TYPE CL_FUNCTION (volatile CL_ADDRSPACE RET_SIGN TYPE *p, RET_SIGN TYPE val) { \ |
| 5 | return (RET_SIGN TYPE)__clc_##CLC_FUNCTION##_addr##LLVM_ADDRSPACE((volatile CL_ADDRSPACE ARG_SIGN TYPE*)p, (ARG_SIGN TYPE)val); \ |
| Aaron Watry | 50a7bcb | 2013-09-05 16:04:01 +0000 | [diff] [blame] | 6 | } |
| 7 | |
| Aaron Watry | ea32a57 | 2014-09-16 22:34:53 +0000 | [diff] [blame] | 8 | /* For atomic functions that don't need different bitcode dependending on argument signedness */ |
| Aaron Watry | 50a7bcb | 2013-09-05 16:04:01 +0000 | [diff] [blame] | 9 | #define ATOMIC_FUNC_SIGN(TYPE, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 10 | _CLC_DECL signed TYPE __clc_##FUNCTION##_addr##LLVM_ADDRSPACE(volatile CL_ADDRSPACE signed TYPE*, signed TYPE); \ |
| Aaron Watry | ea32a57 | 2014-09-16 22:34:53 +0000 | [diff] [blame] | 11 | ATOMIC_FUNC_DEFINE(signed, signed, TYPE, FUNCTION, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 12 | ATOMIC_FUNC_DEFINE(unsigned, signed, TYPE, FUNCTION, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) |
| Aaron Watry | 50a7bcb | 2013-09-05 16:04:01 +0000 | [diff] [blame] | 13 | |
| 14 | #define ATOMIC_FUNC_ADDRSPACE(TYPE, FUNCTION) \ |
| 15 | ATOMIC_FUNC_SIGN(TYPE, FUNCTION, global, 1) \ |
| 16 | ATOMIC_FUNC_SIGN(TYPE, FUNCTION, local, 3) |
| 17 | |
| 18 | #define ATOMIC_FUNC(FUNCTION) \ |
| 19 | ATOMIC_FUNC_ADDRSPACE(int, FUNCTION) |
| 20 | |
| Aaron Watry | f4133b8 | 2014-09-16 22:34:59 +0000 | [diff] [blame] | 21 | #define ATOMIC_FUNC_DEFINE_3_ARG(RET_SIGN, ARG_SIGN, TYPE, CL_FUNCTION, CLC_FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 22 | _CLC_OVERLOAD _CLC_DEF RET_SIGN TYPE CL_FUNCTION (volatile CL_ADDRSPACE RET_SIGN TYPE *p, RET_SIGN TYPE cmp, RET_SIGN TYPE val) { \ |
| 23 | return (RET_SIGN TYPE)__clc_##CLC_FUNCTION##_addr##LLVM_ADDRSPACE((volatile CL_ADDRSPACE ARG_SIGN TYPE*)p, (ARG_SIGN TYPE)cmp, (ARG_SIGN TYPE)val); \ |
| 24 | } |
| 25 | |
| 26 | /* For atomic functions that don't need different bitcode dependending on argument signedness */ |
| 27 | #define ATOMIC_FUNC_SIGN_3_ARG(TYPE, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 28 | _CLC_DECL signed TYPE __clc_##FUNCTION##_addr##LLVM_ADDRSPACE(volatile CL_ADDRSPACE signed TYPE*, signed TYPE, signed TYPE); \ |
| 29 | ATOMIC_FUNC_DEFINE_3_ARG(signed, signed, TYPE, FUNCTION, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) \ |
| 30 | ATOMIC_FUNC_DEFINE_3_ARG(unsigned, signed, TYPE, FUNCTION, FUNCTION, CL_ADDRSPACE, LLVM_ADDRSPACE) |
| 31 | |
| 32 | #define ATOMIC_FUNC_ADDRSPACE_3_ARG(TYPE, FUNCTION) \ |
| 33 | ATOMIC_FUNC_SIGN_3_ARG(TYPE, FUNCTION, global, 1) \ |
| 34 | ATOMIC_FUNC_SIGN_3_ARG(TYPE, FUNCTION, local, 3) |
| 35 | |
| 36 | #define ATOMIC_FUNC_3_ARG(FUNCTION) \ |
| 37 | ATOMIC_FUNC_ADDRSPACE_3_ARG(int, FUNCTION) |
| 38 | |
| Aaron Watry | 50a7bcb | 2013-09-05 16:04:01 +0000 | [diff] [blame] | 39 | ATOMIC_FUNC(atomic_add) |
| Aaron Watry | 348db3c | 2014-09-16 22:34:51 +0000 | [diff] [blame] | 40 | ATOMIC_FUNC(atomic_and) |
| Aaron Watry | 5ab82be | 2014-09-16 22:34:52 +0000 | [diff] [blame] | 41 | ATOMIC_FUNC(atomic_or) |
| Aaron Watry | 283e3fa | 2013-09-06 20:20:21 +0000 | [diff] [blame] | 42 | ATOMIC_FUNC(atomic_sub) |
| Aaron Watry | e210cae | 2014-09-16 22:34:58 +0000 | [diff] [blame] | 43 | ATOMIC_FUNC(atomic_xchg) |
| Aaron Watry | dd754f4 | 2014-09-16 22:34:55 +0000 | [diff] [blame] | 44 | ATOMIC_FUNC(atomic_xor) |
| Aaron Watry | f4133b8 | 2014-09-16 22:34:59 +0000 | [diff] [blame] | 45 | ATOMIC_FUNC_3_ARG(atomic_cmpxchg) |
| Aaron Watry | ea32a57 | 2014-09-16 22:34:53 +0000 | [diff] [blame] | 46 | |
| 47 | _CLC_DECL signed int __clc_atomic_max_addr1(volatile global signed int*, signed int); |
| 48 | _CLC_DECL signed int __clc_atomic_max_addr3(volatile local signed int*, signed int); |
| 49 | _CLC_DECL uint __clc_atomic_umax_addr1(volatile global uint*, uint); |
| 50 | _CLC_DECL uint __clc_atomic_umax_addr3(volatile local uint*, uint); |
| 51 | |
| 52 | ATOMIC_FUNC_DEFINE(signed, signed, int, atomic_max, atomic_max, global, 1) |
| 53 | ATOMIC_FUNC_DEFINE(signed, signed, int, atomic_max, atomic_max, local, 3) |
| 54 | ATOMIC_FUNC_DEFINE(unsigned, unsigned, int, atomic_max, atomic_umax, global, 1) |
| 55 | ATOMIC_FUNC_DEFINE(unsigned, unsigned, int, atomic_max, atomic_umax, local, 3) |
| Aaron Watry | 0545fa3 | 2014-09-16 22:34:56 +0000 | [diff] [blame] | 56 | |
| 57 | _CLC_DECL signed int __clc_atomic_min_addr1(volatile global signed int*, signed int); |
| 58 | _CLC_DECL signed int __clc_atomic_min_addr3(volatile local signed int*, signed int); |
| 59 | _CLC_DECL uint __clc_atomic_umin_addr1(volatile global uint*, uint); |
| 60 | _CLC_DECL uint __clc_atomic_umin_addr3(volatile local uint*, uint); |
| 61 | |
| 62 | ATOMIC_FUNC_DEFINE(signed, signed, int, atomic_min, atomic_min, global, 1) |
| 63 | ATOMIC_FUNC_DEFINE(signed, signed, int, atomic_min, atomic_min, local, 3) |
| 64 | ATOMIC_FUNC_DEFINE(unsigned, unsigned, int, atomic_min, atomic_umin, global, 1) |
| 65 | ATOMIC_FUNC_DEFINE(unsigned, unsigned, int, atomic_min, atomic_umin, local, 3) |