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
113
114
115
116
117
118
|
// Copyright © Tavian Barnes <tavianator@tavianator.com>
// SPDX-License-Identifier: 0BSD
/**
* Shorthand for standard C atomic operations.
*/
#ifndef BFS_ATOMIC_H
#define BFS_ATOMIC_H
#include "prelude.h"
#include "sanity.h"
#include <stdatomic.h>
/**
* Prettier spelling of _Atomic.
*/
#define atomic _Atomic
/**
* Shorthand for atomic_load_explicit().
*
* @param obj
* A pointer to the atomic object.
* @param order
* The memory ordering to use, without the memory_order_ prefix.
* @return
* The loaded value.
*/
#define load(obj, order) \
atomic_load_explicit(obj, memory_order_##order)
/**
* Shorthand for atomic_store_explicit().
*/
#define store(obj, value, order) \
atomic_store_explicit(obj, value, memory_order_##order)
/**
* Shorthand for atomic_exchange_explicit().
*/
#define exchange(obj, value, order) \
atomic_exchange_explicit(obj, value, memory_order_##order)
/**
* Shorthand for atomic_compare_exchange_weak_explicit().
*/
#define compare_exchange_weak(obj, expected, desired, succ, fail) \
atomic_compare_exchange_weak_explicit(obj, expected, desired, memory_order_##succ, memory_order_##fail)
/**
* Shorthand for atomic_compare_exchange_strong_explicit().
*/
#define compare_exchange_strong(obj, expected, desired, succ, fail) \
atomic_compare_exchange_strong_explicit(obj, expected, desired, memory_order_##succ, memory_order_##fail)
/**
* Shorthand for atomic_fetch_add_explicit().
*/
#define fetch_add(obj, arg, order) \
atomic_fetch_add_explicit(obj, arg, memory_order_##order)
/**
* Shorthand for atomic_fetch_sub_explicit().
*/
#define fetch_sub(obj, arg, order) \
atomic_fetch_sub_explicit(obj, arg, memory_order_##order)
/**
* Shorthand for atomic_fetch_or_explicit().
*/
#define fetch_or(obj, arg, order) \
atomic_fetch_or_explicit(obj, arg, memory_order_##order)
/**
* Shorthand for atomic_fetch_xor_explicit().
*/
#define fetch_xor(obj, arg, order) \
atomic_fetch_xor_explicit(obj, arg, memory_order_##order)
/**
* Shorthand for atomic_fetch_and_explicit().
*/
#define fetch_and(obj, arg, order) \
atomic_fetch_and_explicit(obj, arg, memory_order_##order)
/**
* Shorthand for atomic_thread_fence().
*/
#if SANITIZE_THREAD
// TSan doesn't support fences: https://github.com/google/sanitizers/issues/1415
# define thread_fence(obj, order) \
fetch_add(obj, 0, order)
#else
# define thread_fence(obj, order) \
atomic_thread_fence(memory_order_##order)
#endif
/**
* Shorthand for atomic_signal_fence().
*/
#define signal_fence(order) \
atomic_signal_fence(memory_order_##order)
/**
* A hint to the CPU to relax while it spins.
*/
#if __has_builtin(__builtin_ia32_pause)
# define spin_loop() __builtin_ia32_pause()
#elif __has_builtin(__builtin_arm_yield)
# define spin_loop() __builtin_arm_yield()
#elif __has_builtin(__builtin_riscv_pause)
# define spin_loop() __builtin_riscv_pause()
#else
# define spin_loop() ((void)0)
#endif
#endif // BFS_ATOMIC_H
|
