[3c2c2f0] | 1 | #include <assert.h>
|
---|
| 2 | #include <stddef.h>
|
---|
| 3 | struct $thread;
|
---|
| 4 | struct __thread_desc_link {
|
---|
| 5 | struct $thread * next;
|
---|
| 6 | struct $thread * prev;
|
---|
| 7 | volatile unsigned long long ts;
|
---|
| 8 | int preferred;
|
---|
| 9 | };
|
---|
| 10 | struct $thread {
|
---|
| 11 | int f[64]; // FIXME: make "is volatile" consistent; given bug #TBD
|
---|
| 12 | struct __thread_desc_link link;
|
---|
| 13 | };
|
---|
| 14 |
|
---|
| 15 | // Intrusives lanes which are used by the relaxed ready queue
|
---|
| 16 | struct __attribute__((aligned(128))) __intrusive_lane_t {
|
---|
| 17 |
|
---|
| 18 | #if defined(USE_MPSC)
|
---|
| 19 | mpsc_queue($thread) queue;
|
---|
| 20 | __attribute__((aligned(128)))
|
---|
| 21 | #else
|
---|
| 22 | // anchor for the head and the tail of the queue
|
---|
| 23 | __attribute__((aligned(128))) struct __sentinel_t {
|
---|
| 24 | // Link lists fields
|
---|
| 25 | // instrusive link field for threads
|
---|
| 26 | // must be exactly as in $thread
|
---|
| 27 | __thread_desc_link link;
|
---|
| 28 | } before, after;
|
---|
| 29 | #endif
|
---|
| 30 |
|
---|
| 31 | // spin lock protecting the queue
|
---|
| 32 | volatile bool lock;
|
---|
| 33 |
|
---|
| 34 | // Optional statistic counters
|
---|
| 35 | #if !defined(__CFA_NO_SCHED_STATS__)
|
---|
| 36 | struct __attribute__((aligned(64))) {
|
---|
| 37 | // difference between number of push and pops
|
---|
| 38 | ssize_t diff;
|
---|
| 39 |
|
---|
| 40 | // total number of pushes and pops
|
---|
| 41 | size_t push;
|
---|
| 42 | size_t pop ;
|
---|
| 43 | } stat;
|
---|
| 44 | #endif
|
---|
| 45 | };
|
---|
| 46 |
|
---|
| 47 | void ?{}(__intrusive_lane_t & this);
|
---|
| 48 | void ^?{}(__intrusive_lane_t & this);
|
---|
| 49 |
|
---|
| 50 | // Get the head pointer (one before the first element) from the anchor
|
---|
| 51 | static inline $thread * head(const __intrusive_lane_t & this) {
|
---|
| 52 | #if defined(USE_MPSC)
|
---|
| 53 | return this.queue.head;
|
---|
| 54 | #else
|
---|
| 55 | $thread * rhead = ($thread *)(
|
---|
| 56 | (uintptr_t)( &this.before ) - offsetof( $thread, link )
|
---|
| 57 | );
|
---|
| 58 | /* paranoid */ verify(rhead);
|
---|
| 59 | return rhead;
|
---|
| 60 | #endif
|
---|
| 61 | }
|
---|
| 62 |
|
---|
| 63 | // Get the tail pointer (one after the last element) from the anchor
|
---|
| 64 | static inline $thread * tail(const __intrusive_lane_t & this) {
|
---|
| 65 | #if defined(USE_MPSC)
|
---|
| 66 | return this.queue.tail;
|
---|
| 67 | #else
|
---|
| 68 | $thread * rtail = ($thread *)(
|
---|
| 69 | (uintptr_t)( &this.after ) - offsetof( $thread, link )
|
---|
| 70 | );
|
---|
| 71 | /* paranoid */ verify(rtail);
|
---|
| 72 | return rtail;
|
---|
| 73 | #endif
|
---|
| 74 | }
|
---|
| 75 |
|
---|
| 76 | // Ctor
|
---|
| 77 | void ?{}( __intrusive_lane_t & this ) {
|
---|
| 78 | this.lock = false;
|
---|
| 79 |
|
---|
| 80 | #if !defined(USE_MPSC)
|
---|
| 81 | this.before.link.prev = 0p;
|
---|
| 82 | this.before.link.next = tail(this);
|
---|
| 83 | this.before.link.ts = 0;
|
---|
| 84 |
|
---|
| 85 | this.after .link.prev = head(this);
|
---|
| 86 | this.after .link.next = 0p;
|
---|
| 87 | this.after .link.ts = 0;
|
---|
| 88 |
|
---|
| 89 | #if !defined(__CFA_NO_SCHED_STATS__)
|
---|
| 90 | this.stat.diff = 0;
|
---|
| 91 | this.stat.push = 0;
|
---|
| 92 | this.stat.pop = 0;
|
---|
| 93 | #endif
|
---|
| 94 |
|
---|
| 95 | // We add a boat-load of assertions here because the anchor code is very fragile
|
---|
| 96 | /* paranoid */ verify(((uintptr_t)( head(this) ) + offsetof( $thread, link )) == (uintptr_t)(&this.before));
|
---|
| 97 | /* paranoid */ verify(((uintptr_t)( tail(this) ) + offsetof( $thread, link )) == (uintptr_t)(&this.after ));
|
---|
| 98 | /* paranoid */ verify(head(this)->link.prev == 0p );
|
---|
| 99 | /* paranoid */ verify(head(this)->link.next == tail(this) );
|
---|
| 100 | /* paranoid */ verify(tail(this)->link.next == 0p );
|
---|
| 101 | /* paranoid */ verify(tail(this)->link.prev == head(this) );
|
---|
| 102 | /* paranoid */ verify(&head(this)->link.prev == &this.before.link.prev );
|
---|
| 103 | /* paranoid */ verify(&head(this)->link.next == &this.before.link.next );
|
---|
| 104 | /* paranoid */ verify(&tail(this)->link.prev == &this.after .link.prev );
|
---|
| 105 | /* paranoid */ verify(&tail(this)->link.next == &this.after .link.next );
|
---|
| 106 | /* paranoid */ verify(__alignof__(__intrusive_lane_t) == 128);
|
---|
| 107 | /* paranoid */ verify(__alignof__(this) == 128);
|
---|
| 108 | /* paranoid */ verifyf(((intptr_t)(&this) % 128) == 0, "Expected address to be aligned %p %% 128 == %zd", &this, ((intptr_t)(&this) % 128));
|
---|
| 109 | #endif
|
---|
| 110 | }
|
---|
| 111 |
|
---|
| 112 | // Dtor is trivial
|
---|
| 113 | void ^?{}( __intrusive_lane_t & this ) {
|
---|
| 114 | #if !defined(USE_MPSC)
|
---|
| 115 | // Make sure the list is empty
|
---|
| 116 | /* paranoid */ verify(head(this)->link.prev == 0p );
|
---|
| 117 | /* paranoid */ verify(head(this)->link.next == tail(this) );
|
---|
| 118 | /* paranoid */ verify(tail(this)->link.next == 0p );
|
---|
| 119 | /* paranoid */ verify(tail(this)->link.prev == head(this) );
|
---|
| 120 | #endif
|
---|
| 121 | }
|
---|
| 122 |
|
---|
| 123 | // Push a thread onto this lane
|
---|
| 124 | // returns true of lane was empty before push, false otherwise
|
---|
| 125 | bool push(__intrusive_lane_t & this, $thread * node) {
|
---|
| 126 | #if defined(USE_MPSC)
|
---|
| 127 | inline $thread * volatile & ?`next ( $thread * this ) __attribute__((const)) {
|
---|
| 128 | return this->link.next;
|
---|
| 129 | }
|
---|
| 130 | push(this.queue, node);
|
---|
| 131 | #else
|
---|
| 132 | #if defined(__CFA_WITH_VERIFY__)
|
---|
| 133 | /* paranoid */ verify(this.lock);
|
---|
| 134 | /* paranoid */ verify(node->link.ts != 0);
|
---|
| 135 | /* paranoid */ verify(node->link.next == 0p);
|
---|
| 136 | /* paranoid */ verify(node->link.prev == 0p);
|
---|
| 137 | /* paranoid */ verify(tail(this)->link.next == 0p);
|
---|
| 138 | /* paranoid */ verify(head(this)->link.prev == 0p);
|
---|
| 139 |
|
---|
| 140 | if(this.before.link.ts == 0l) {
|
---|
| 141 | /* paranoid */ verify(tail(this)->link.prev == head(this));
|
---|
| 142 | /* paranoid */ verify(head(this)->link.next == tail(this));
|
---|
| 143 | } else {
|
---|
| 144 | /* paranoid */ verify(tail(this)->link.prev != head(this));
|
---|
| 145 | /* paranoid */ verify(head(this)->link.next != tail(this));
|
---|
| 146 | }
|
---|
| 147 | #endif
|
---|
| 148 |
|
---|
| 149 | // Get the relevant nodes locally
|
---|
| 150 | $thread * tail = tail(this);
|
---|
| 151 | $thread * prev = tail->link.prev;
|
---|
| 152 |
|
---|
| 153 | // Do the push
|
---|
| 154 | node->link.next = tail;
|
---|
| 155 | node->link.prev = prev;
|
---|
| 156 | prev->link.next = node;
|
---|
| 157 | tail->link.prev = node;
|
---|
| 158 |
|
---|
| 159 | // Update stats
|
---|
| 160 | #if !defined(__CFA_NO_SCHED_STATS__)
|
---|
| 161 | this.stat.diff++;
|
---|
| 162 | this.stat.push++;
|
---|
| 163 | #endif
|
---|
| 164 |
|
---|
| 165 | verify(node->link.next == tail(this));
|
---|
| 166 |
|
---|
| 167 | // Check if the queue used to be empty
|
---|
| 168 | if(this.before.link.ts == 0l) {
|
---|
| 169 | this.before.link.ts = node->link.ts;
|
---|
| 170 | /* paranoid */ verify(node->link.prev == head(this));
|
---|
| 171 | return true;
|
---|
| 172 | }
|
---|
| 173 | return false;
|
---|
| 174 | #endif
|
---|
| 175 | }
|
---|
| 176 |
|
---|
| 177 | // Pop a thread from this lane (must be non-empty)
|
---|
| 178 | // returns popped
|
---|
| 179 | // returns true of lane was empty before push, false otherwise
|
---|
| 180 | $thread * pop(__intrusive_lane_t & this) {
|
---|
| 181 | /* paranoid */ verify(this.lock);
|
---|
| 182 | #if defined(USE_MPSC)
|
---|
| 183 | inline $thread * volatile & ?`next ( $thread * this ) __attribute__((const)) {
|
---|
| 184 | return this->link.next;
|
---|
| 185 | }
|
---|
| 186 | return pop(this.queue);
|
---|
| 187 | #else
|
---|
| 188 | /* paranoid */ verify(this.before.link.ts != 0ul);
|
---|
| 189 |
|
---|
| 190 | // Get anchors locally
|
---|
| 191 | $thread * head = head(this);
|
---|
| 192 | $thread * tail = tail(this);
|
---|
| 193 |
|
---|
| 194 | // Get the relevant nodes locally
|
---|
| 195 | $thread * node = head->link.next;
|
---|
| 196 | $thread * next = node->link.next;
|
---|
| 197 |
|
---|
| 198 | /* paranoid */ verify(node != tail);
|
---|
| 199 | /* paranoid */ verify(node);
|
---|
| 200 |
|
---|
| 201 | // Do the pop
|
---|
| 202 | head->link.next = next;
|
---|
| 203 | next->link.prev = head;
|
---|
| 204 | node->link.next = 0p;
|
---|
| 205 | node->link.prev = 0p;
|
---|
| 206 |
|
---|
| 207 | // Update head time stamp
|
---|
| 208 | this.before.link.ts = next->link.ts;
|
---|
| 209 |
|
---|
| 210 | // Update stats
|
---|
| 211 | #ifndef __CFA_NO_SCHED_STATS__
|
---|
| 212 | this.stat.diff--;
|
---|
| 213 | this.stat.pop ++;
|
---|
| 214 | #endif
|
---|
| 215 |
|
---|
| 216 | // Check if we emptied list and return accordingly
|
---|
| 217 | /* paranoid */ verify(tail(this)->link.next == 0p);
|
---|
| 218 | /* paranoid */ verify(head(this)->link.prev == 0p);
|
---|
| 219 | if(next == tail) {
|
---|
| 220 | /* paranoid */ verify(this.before.link.ts == 0);
|
---|
| 221 | /* paranoid */ verify(tail(this)->link.prev == head(this));
|
---|
| 222 | /* paranoid */ verify(head(this)->link.next == tail(this));
|
---|
| 223 | return node;
|
---|
| 224 | }
|
---|
| 225 | else {
|
---|
| 226 | /* paranoid */ verify(next->link.ts != 0);
|
---|
| 227 | /* paranoid */ verify(tail(this)->link.prev != head(this));
|
---|
| 228 | /* paranoid */ verify(head(this)->link.next != tail(this));
|
---|
| 229 | /* paranoid */ verify(this.before.link.ts != 0);
|
---|
| 230 | return node;
|
---|
| 231 | }
|
---|
| 232 | #endif
|
---|
| 233 | }
|
---|
| 234 |
|
---|
| 235 | // Check whether or not list is empty
|
---|
| 236 | static inline bool is_empty(__intrusive_lane_t & this) {
|
---|
| 237 | #if defined(USE_MPSC)
|
---|
| 238 | return this.queue.head == 0p;
|
---|
| 239 | #else
|
---|
| 240 | // Cannot verify here since it may not be locked
|
---|
| 241 | return this.before.link.ts == 0;
|
---|
| 242 | #endif
|
---|
| 243 | }
|
---|
| 244 |
|
---|
| 245 | // Return the timestamp
|
---|
| 246 | static inline unsigned long long ts(__intrusive_lane_t & this) {
|
---|
| 247 | #if defined(USE_MPSC)
|
---|
| 248 | $thread * tl = this.queue.head;
|
---|
| 249 | if(!tl) return -1ull;
|
---|
| 250 | return tl->link.ts;
|
---|
| 251 | #else
|
---|
| 252 | // Cannot verify here since it may not be locked
|
---|
| 253 | return this.before.link.ts;
|
---|
| 254 | #endif
|
---|
| 255 | }
|
---|