| 1 | // sequential equivalent swap
 | 
|---|
| 2 | void swap( uint victim_idx, uint my_idx  ) {
 | 
|---|
| 3 |     // Step 0:
 | 
|---|
| 4 |     work_queue * my_queue = request_queues[my_idx];
 | 
|---|
| 5 |     work_queue * vic_queue = request_queues[victim_idx];
 | 
|---|
| 6 |     // Step 2:
 | 
|---|
| 7 |     request_queues[my_idx] = 0p;
 | 
|---|
| 8 |     // Step 3:
 | 
|---|
| 9 |     request_queues[victim_idx] = my_queue;
 | 
|---|
| 10 |     // Step 4:
 | 
|---|
| 11 |     request_queues[my_idx] = vic_queue;
 | 
|---|
| 12 | }
 | 
|---|
| 13 | 
 | 
|---|
| 14 | // This routine is atomic
 | 
|---|
| 15 | bool CAS( work_queue ** ptr, work_queue ** old, work_queue * new ) {
 | 
|---|
| 16 |     if ( *ptr != *old )
 | 
|---|
| 17 |         return false;
 | 
|---|
| 18 |     *ptr = new;
 | 
|---|
| 19 |     return true;
 | 
|---|
| 20 | }
 | 
|---|
| 21 | 
 | 
|---|
| 22 | bool try_swap_queues( worker & this, uint victim_idx, uint my_idx ) with(this) {
 | 
|---|
| 23 |     // Step 0:
 | 
|---|
| 24 |     // request_queues is the shared array of all sharded queues
 | 
|---|
| 25 |     work_queue * my_queue = request_queues[my_idx];
 | 
|---|
| 26 |     work_queue * vic_queue = request_queues[victim_idx];
 | 
|---|
| 27 | 
 | 
|---|
| 28 |     // Step 1:
 | 
|---|
| 29 |     // If either queue is 0p then they are in the process of being stolen
 | 
|---|
| 30 |     // 0p is CForAll's equivalent of C++'s nullptr
 | 
|---|
| 31 |     if ( vic_queue == 0p ) return false;
 | 
|---|
| 32 | 
 | 
|---|
| 33 |     // Step 2:
 | 
|---|
| 34 |     // Try to set thief's queue ptr to be 0p.
 | 
|---|
| 35 |     // If this CAS fails someone stole thief's queue so return false
 | 
|---|
| 36 |     if ( !CAS( &request_queues[my_idx], &my_queue, 0p ) )
 | 
|---|
| 37 |         return false;
 | 
|---|
| 38 |     
 | 
|---|
| 39 |     // Step 3:
 | 
|---|
| 40 |     // Try to set victim queue ptr to be thief's queue ptr.
 | 
|---|
| 41 |     // If it fails someone stole the other queue, so fix up then return false
 | 
|---|
| 42 |     if ( !CAS( &request_queues[victim_idx], &vic_queue, my_queue ) ) {
 | 
|---|
| 43 |         request_queues[my_idx] = my_queue; // reset queue ptr back to prev val
 | 
|---|
| 44 |         return false;
 | 
|---|
| 45 |     }
 | 
|---|
| 46 | 
 | 
|---|
| 47 |     // Step 4:
 | 
|---|
| 48 |     // Successfully swapped.
 | 
|---|
| 49 |     // Thief's ptr is 0p so no one will touch it
 | 
|---|
| 50 |     // Write back without CAS is safe
 | 
|---|
| 51 |     request_queues[my_idx] = vic_queue;
 | 
|---|
| 52 |     return true;
 | 
|---|
| 53 | }
 | 
|---|