source: libcfa/src/concurrency/actor.hfa @ 508671e

Last change on this file since 508671e was 508671e, checked in by caparsons <caparson@…>, 12 months ago

added termination check for non-stealing actor system

  • Property mode set to 100644
File size: 26.7 KB
Line 
1#pragma once
2
3#include <locks.hfa>
4#include <limits.hfa>
5#include <kernel.hfa>
6#include <iofwd.hfa>
7#include <virtual_dtor.hfa>
8
9#ifdef __CFA_DEBUG__
10#define CFA_DEBUG( stmt ) stmt
11#else
12#define CFA_DEBUG( stmt )
13#endif // CFA_DEBUG
14
15#define DEBUG_ABORT( cond, string ) CFA_DEBUG( if ( cond ) abort( string ) )
16
17// Define the default number of processors created in the executor. Must be greater than 0.
18#define __DEFAULT_EXECUTOR_PROCESSORS__ 2
19
20// Define the default number of threads created in the executor. Must be greater than 0.
21#define __DEFAULT_EXECUTOR_WORKERS__ 2
22
23// Define the default number of executor request-queues (mailboxes) written to by actors and serviced by the
24// actor-executor threads. Must be greater than 0.
25#define __DEFAULT_EXECUTOR_RQUEUES__ 4
26
27// Define if executor is created in a separate cluster
28#define __DEFAULT_EXECUTOR_SEPCLUS__ false
29
30#define __DEFAULT_EXECUTOR_BUFSIZE__ 10
31
32#define __STEAL 1 // workstealing toggle. Disjoint from toggles above
33
34// workstealing heuristic selection (only set one to be 1)
35// #define RAND 0
36#define SEARCH 1
37
38// show stats
39// #define ACTOR_STATS
40
41// forward decls
42struct actor;
43struct message;
44struct executor;
45
46enum allocation { Nodelete, Delete, Destroy, Finished }; // allocation status
47
48typedef allocation (*__receive_fn)(actor &, message &, actor **, message **);
49struct request {
50    actor * receiver;
51    message * msg;
52    __receive_fn fn;
53};
54
55struct a_msg {
56    int m;
57};
58static inline void ?{}( request & this ) {}
59static inline void ?{}( request & this, actor * receiver, message * msg, __receive_fn fn ) {
60    this.receiver = receiver;
61    this.msg = msg;
62    this.fn = fn;
63}
64static inline void ?{}( request & this, request & copy ) {
65    this.receiver = copy.receiver;
66    this.msg = copy.msg;
67    this.fn = copy.fn;
68}
69
70// Vector-like data structure that supports O(1) queue operations with no bound on size
71// assumes gulping behaviour (once a remove occurs, removes happen until empty beforw next insert)
72struct copy_queue {
73    request * buffer;
74    size_t count, buffer_size, index, utilized, last_size;
75};
76static inline void ?{}( copy_queue & this ) {}
77static inline void ?{}( copy_queue & this, size_t buf_size ) with(this) {
78    buffer_size = buf_size;
79    buffer = aalloc( buffer_size );
80    count = 0;
81    utilized = 0;
82    index = 0;
83    last_size = 0;
84}
85static inline void ^?{}( copy_queue & this ) with(this) {
86    DEBUG_ABORT( count != 0, "Actor system terminated with messages sent but not received\n" );
87    adelete(buffer);
88}
89
90static inline void insert( copy_queue & this, request & elem ) with(this) {
91    if ( count >= buffer_size ) { // increase arr size
92        last_size = buffer_size;
93        buffer_size = 2 * buffer_size;
94        buffer = realloc( buffer, sizeof( request ) * buffer_size );
95        /* paranoid */ verify( buffer );
96    }
97    memcpy( &buffer[count], &elem, sizeof(request) );
98    count++;
99}
100
101// once you start removing you need to remove all elements
102// it is not supported to call insert() before the array is fully empty
103static inline request & remove( copy_queue & this ) with(this) {
104    if ( count > 0 ) {
105        count--;
106        size_t old_idx = index;
107        index = count == 0 ? 0 : index + 1;
108        return buffer[old_idx];
109    }
110    request * ret = 0p;
111    return *0p;
112}
113
114// try to reclaim some memory if less than half of buffer is utilized
115static inline void reclaim( copy_queue & this ) with(this) {
116    if ( utilized >= last_size || buffer_size <= 4 ) { utilized = 0; return; }
117    utilized = 0;
118    buffer_size--;
119    buffer = realloc( buffer, sizeof( request ) * buffer_size ); // try to reclaim some memory
120}
121
122static inline bool is_empty( copy_queue & this ) with(this) { return count == 0; }
123
124struct work_queue {
125    __spinlock_t mutex_lock;
126    copy_queue * owned_queue;       // copy queue allocated and cleaned up by this work_queue
127    copy_queue * c_queue;           // current queue
128    volatile bool being_processed;  // flag to prevent concurrent processing
129    #ifdef ACTOR_STATS
130    unsigned int id;
131    size_t missed;                  // transfers skipped due to being_processed flag being up
132    #endif
133}; // work_queue
134static inline void ?{}( work_queue & this, size_t buf_size, unsigned int i ) with(this) {
135    owned_queue = alloc();      // allocated separately to avoid false sharing
136    (*owned_queue){ buf_size };
137    c_queue = owned_queue;
138    being_processed = false;
139    #ifdef ACTOR_STATS
140    id = i;
141    missed = 0;
142    #endif
143}
144
145// clean up copy_queue owned by this work_queue
146static inline void ^?{}( work_queue & this ) with(this) { delete( owned_queue ); }
147
148static inline void insert( work_queue & this, request & elem ) with(this) {
149    lock( mutex_lock __cfaabi_dbg_ctx2 );
150    insert( *c_queue, elem );
151    unlock( mutex_lock );
152} // insert
153
154static inline void transfer( work_queue & this, copy_queue ** transfer_to ) with(this) {
155    lock( mutex_lock __cfaabi_dbg_ctx2 );
156    #ifdef __STEAL
157
158    // check if queue is being processed elsewhere
159    if ( unlikely( being_processed ) ) {
160        #ifdef ACTOR_STATS
161        missed++;
162        #endif
163        unlock( mutex_lock );
164        return;
165    }
166
167    being_processed = c_queue->count != 0;
168    #endif // __STEAL
169
170    c_queue->utilized = c_queue->count;
171
172    // swap copy queue ptrs
173    copy_queue * temp = *transfer_to;
174    *transfer_to = c_queue;
175    c_queue = temp;
176    unlock( mutex_lock );
177} // transfer
178
179// needed since some info needs to persist past worker lifetimes
180struct worker_info {
181    volatile unsigned long long stamp;
182    #ifdef ACTOR_STATS
183    size_t stolen_from, try_steal, stolen, empty_stolen, failed_swaps, msgs_stolen;
184    unsigned long long processed;
185    size_t gulps;
186    #endif
187};
188static inline void ?{}( worker_info & this ) {
189    #ifdef ACTOR_STATS
190    this.stolen_from = 0;
191    this.try_steal = 0;                             // attempts to steal
192    this.stolen = 0;                                // successful steals
193    this.processed = 0;                             // requests processed
194    this.gulps = 0;                                 // number of gulps
195    this.failed_swaps = 0;                          // steal swap failures
196    this.empty_stolen = 0;                          // queues empty after steal
197    this.msgs_stolen = 0;                           // number of messages stolen
198    #endif
199    this.stamp = rdtscl();
200}
201
202// #ifdef ACTOR_STATS
203// unsigned int * stolen_arr;
204// unsigned int * replaced_queue;
205// #endif
206thread worker {
207    work_queue ** request_queues;
208    copy_queue * current_queue;
209    executor * executor_;
210    unsigned int start, range;
211    int id;
212};
213
214#ifdef ACTOR_STATS
215// aggregate counters for statistics
216size_t __total_tries = 0, __total_stolen = 0, __total_workers, __all_gulps = 0, __total_empty_stolen = 0,
217    __total_failed_swaps = 0, __all_processed = 0, __num_actors_stats = 0, __all_msgs_stolen = 0;
218#endif
219static inline void ?{}( worker & this, cluster & clu, work_queue ** request_queues, copy_queue * current_queue, executor * executor_,
220    unsigned int start, unsigned int range, int id ) {
221    ((thread &)this){ clu };
222    this.request_queues = request_queues;           // array of all queues
223    this.current_queue = current_queue;             // currently gulped queue (start with empty queue to use in swap later)
224    this.executor_ = executor_;                     // pointer to current executor
225    this.start = start;                             // start of worker's subrange of request_queues
226    this.range = range;                             // size of worker's subrange of request_queues
227    this.id = id;                                   // worker's id and index in array of workers
228}
229
230static bool no_steal = false;
231struct executor {
232    cluster * cluster;                                                      // if workers execute on separate cluster
233        processor ** processors;                                            // array of virtual processors adding parallelism for workers
234        work_queue * request_queues;                                // master array of work request queues
235    copy_queue * local_queues;                      // array of all worker local queues to avoid deletion race
236        work_queue ** worker_req_queues;                // secondary array of work queues to allow for swapping
237    worker ** workers;                                                          // array of workers executing work requests
238    worker_info * w_infos;                          // array of info about each worker
239        unsigned int nprocessors, nworkers, nrqueues;   // number of processors/threads/request queues
240        bool seperate_clus;                                                             // use same or separate cluster for executor
241    volatile bool is_shutdown;                      // flag to communicate shutdown to worker threads
242}; // executor
243
244// #ifdef ACTOR_STATS
245// __spinlock_t out_lock;
246// #endif
247static inline void ^?{}( worker & mutex this ) with(this) {
248    #ifdef ACTOR_STATS
249    __atomic_add_fetch(&__all_gulps, executor_->w_infos[id].gulps,__ATOMIC_SEQ_CST);
250    __atomic_add_fetch(&__all_processed, executor_->w_infos[id].processed,__ATOMIC_SEQ_CST);
251    __atomic_add_fetch(&__all_msgs_stolen, executor_->w_infos[id].msgs_stolen,__ATOMIC_SEQ_CST);
252    __atomic_add_fetch(&__total_tries, executor_->w_infos[id].try_steal, __ATOMIC_SEQ_CST);
253    __atomic_add_fetch(&__total_stolen, executor_->w_infos[id].stolen, __ATOMIC_SEQ_CST);
254    __atomic_add_fetch(&__total_failed_swaps, executor_->w_infos[id].failed_swaps, __ATOMIC_SEQ_CST);
255    __atomic_add_fetch(&__total_empty_stolen, executor_->w_infos[id].empty_stolen, __ATOMIC_SEQ_CST);
256
257    // per worker steal stats (uncomment alongside the lock above this routine to print)
258    // lock( out_lock __cfaabi_dbg_ctx2 );
259    // printf("Worker id: %d, processed: %llu messages, attempted %lu, stole: %lu, stolen from: %lu\n", id, processed, try_steal, stolen, __atomic_add_fetch(&executor_->w_infos[id].stolen_from, 0, __ATOMIC_SEQ_CST) );
260    // int count = 0;
261    // int count2 = 0;
262    // for ( i; range ) {
263    //     if ( replaced_queue[start + i] > 0 ){
264    //         count++;
265    //         // printf("%d: %u, ",i, replaced_queue[i]);
266    //     }
267    //     if (__atomic_add_fetch(&stolen_arr[start + i],0,__ATOMIC_SEQ_CST) > 0)
268    //         count2++;
269    // }
270    // printf("swapped with: %d of %u indices\n", count, executor_->nrqueues / executor_->nworkers );
271    // printf("%d of %u indices were stolen\n", count2, executor_->nrqueues / executor_->nworkers );
272    // unlock( out_lock );
273    #endif
274}
275
276static inline void ?{}( executor & this, unsigned int nprocessors, unsigned int nworkers, unsigned int nrqueues, bool seperate_clus, size_t buf_size ) with(this) {
277    if ( nrqueues < nworkers ) abort( "nrqueues needs to be >= nworkers\n" );
278    this.nprocessors = nprocessors;
279    this.nworkers = nworkers;
280    this.nrqueues = nrqueues;
281    this.seperate_clus = seperate_clus;
282    this.is_shutdown = false;
283
284    if ( nworkers == nrqueues )
285        no_steal = true;
286   
287    #ifdef ACTOR_STATS
288    // stolen_arr = aalloc( nrqueues );
289    // replaced_queue = aalloc( nrqueues );
290    __total_workers = nworkers;
291    #endif
292
293    if ( seperate_clus ) {
294        cluster = alloc();
295        (*cluster){};
296    } else cluster = active_cluster();
297
298    request_queues = aalloc( nrqueues );
299    worker_req_queues = aalloc( nrqueues );
300    for ( i; nrqueues ) {
301        request_queues[i]{ buf_size, i };
302        worker_req_queues[i] = &request_queues[i];
303    }
304   
305    processors = aalloc( nprocessors );
306    for ( i; nprocessors )
307        (*(processors[i] = alloc())){ *cluster };
308
309    local_queues = aalloc( nworkers );
310    workers = aalloc( nworkers );
311    w_infos = aalloc( nworkers );
312    unsigned int reqPerWorker = nrqueues / nworkers, extras = nrqueues % nworkers;
313
314    for ( i; nworkers ) {
315        w_infos[i]{};
316        local_queues[i]{ buf_size };
317    }
318
319    for ( unsigned int i = 0, start = 0, range; i < nworkers; i += 1, start += range ) {
320        range = reqPerWorker + ( i < extras ? 1 : 0 );
321        (*(workers[i] = alloc())){ *cluster, worker_req_queues, &local_queues[i], &this, start, range, i };
322    } // for
323}
324static inline void ?{}( executor & this, unsigned int nprocessors, unsigned int nworkers, unsigned int nrqueues, bool seperate_clus ) { this{ nprocessors, nworkers, nrqueues, seperate_clus, __DEFAULT_EXECUTOR_BUFSIZE__ }; }
325static inline void ?{}( executor & this, unsigned int nprocessors, unsigned int nworkers, unsigned int nrqueues ) { this{ nprocessors, nworkers, nrqueues, __DEFAULT_EXECUTOR_SEPCLUS__ }; }
326static inline void ?{}( executor & this, unsigned int nprocessors, unsigned int nworkers ) { this{ nprocessors, nworkers, __DEFAULT_EXECUTOR_RQUEUES__ }; }
327static inline void ?{}( executor & this, unsigned int nprocessors ) { this{ nprocessors, __DEFAULT_EXECUTOR_WORKERS__ }; }
328static inline void ?{}( executor & this ) { this{ __DEFAULT_EXECUTOR_PROCESSORS__ }; }
329
330static inline void ^?{}( executor & this ) with(this) {
331    is_shutdown = true;
332
333    for ( i; nworkers )
334        delete( workers[i] );
335
336    for ( i; nprocessors ) {
337        delete( processors[i] );
338    } // for
339
340    #ifdef ACTOR_STATS
341    size_t misses = 0;
342    for ( i; nrqueues ) {
343        misses += worker_req_queues[i]->missed;
344    }
345    // adelete( stolen_arr );
346    // adelete( replaced_queue );
347    #endif
348
349    adelete( workers );
350    adelete( w_infos );
351    adelete( local_queues );
352    adelete( request_queues );
353    adelete( worker_req_queues );
354    adelete( processors );
355    if ( seperate_clus ) delete( cluster );
356
357    #ifdef ACTOR_STATS // print formatted stats
358    printf("    Actor System Stats:\n");
359    printf("\tActors Created:\t\t\t\t%lu\n\tMessages Sent:\t\t\t\t%lu\n", __num_actors_stats, __all_processed);
360    size_t avg_gulps = __all_gulps == 0 ? 0 : __all_processed / __all_gulps;
361    printf("\tGulps:\t\t\t\t\t%lu\n\tAverage Gulp Size:\t\t\t%lu\n\tMissed gulps:\t\t\t\t%lu\n", __all_gulps, avg_gulps, misses);
362    printf("\tSteal attempts:\t\t\t\t%lu\n\tSteals:\t\t\t\t\t%lu\n\tSteal failures (no candidates):\t\t%lu\n\tSteal failures (failed swaps):\t\t%lu\t Empty steals:\t\t%lu\n",
363        __total_tries, __total_stolen, __total_tries - __total_stolen - __total_failed_swaps, __total_failed_swaps, __total_empty_stolen);
364    size_t avg_steal = __total_stolen == 0 ? 0 : __all_msgs_stolen / __total_stolen;
365    printf("\tMessages stolen:\t\t\t%lu\n\tAverage steal size:\t\t\t%lu\n", __all_msgs_stolen, avg_steal);
366    #endif
367       
368}
369
370// this is a static field of executor but have to forward decl for get_next_ticket
371static size_t __next_ticket = 0;
372
373static inline size_t __get_next_ticket( executor & this ) with(this) {
374    #ifdef __CFA_DEBUG__
375    size_t temp = __atomic_fetch_add( &__next_ticket, 1, __ATOMIC_SEQ_CST) % nrqueues;
376
377    // reserve MAX for dead actors
378    if ( unlikely( temp == MAX ) ) temp = __atomic_fetch_add( &__next_ticket, 1, __ATOMIC_SEQ_CST) % nrqueues;
379    return temp;
380    #else
381    return __atomic_fetch_add( &__next_ticket, 1, __ATOMIC_RELAXED) % nrqueues;
382    #endif
383} // tickets
384
385// TODO: update globals in this file to be static fields once the static fields project is done
386static executor * __actor_executor_ = 0p;
387static bool __actor_executor_passed = false;            // was an executor passed to start_actor_system
388static size_t __num_actors_ = 0;                                        // number of actor objects in system
389static struct thread$ * __actor_executor_thd = 0p;              // used to wake executor after actors finish
390struct actor {
391    size_t ticket;                                          // executor-queue handle
392    allocation alloc;                                       // allocation action
393    inline virtual_dtor;
394};
395
396static inline void ?{}( actor & this ) with(this) {
397    // Once an actor is allocated it must be sent a message or the actor system cannot stop. Hence, its receive
398    // member must be called to end it
399    DEBUG_ABORT( __actor_executor_ == 0p, "Creating actor before calling start_actor_system() can cause undefined behaviour.\n" );
400    alloc = Nodelete;
401    ticket = __get_next_ticket( *__actor_executor_ );
402    __atomic_fetch_add( &__num_actors_, 1, __ATOMIC_RELAXED );
403    #ifdef ACTOR_STATS
404    __atomic_fetch_add( &__num_actors_stats, 1, __ATOMIC_SEQ_CST );
405    #endif
406}
407
408static inline void check_actor( actor & this ) {
409    if ( this.alloc != Nodelete ) {
410        switch( this.alloc ) {
411            case Delete: delete( &this ); break;
412            case Destroy:
413                CFA_DEBUG( this.ticket = MAX; );        // mark as terminated
414                ^?{}(this);
415                break;
416            case Finished:
417                CFA_DEBUG( this.ticket = MAX; );        // mark as terminated
418                break;
419            default: ;                                                          // stop warning
420        }
421
422        if ( unlikely( __atomic_add_fetch( &__num_actors_, -1, __ATOMIC_RELAXED ) == 0 ) ) { // all actors have terminated
423            unpark( __actor_executor_thd );
424        }
425    }
426}
427
428struct message {
429    allocation alloc;                   // allocation action
430    inline virtual_dtor;
431};
432
433static inline void ?{}( message & this ) {
434    this.alloc = Nodelete;
435}
436static inline void ?{}( message & this, allocation alloc ) {
437    memcpy( &this.alloc, &alloc, sizeof(allocation) ); // optimization to elide ctor
438    CFA_DEBUG( if( this.alloc == Finished ) this.alloc = Nodelete; )
439}
440static inline void ^?{}( message & this ) with(this) {
441    CFA_DEBUG(
442                if ( alloc == Nodelete ) {
443                        printf( "CFA warning (UNIX pid:%ld) : program terminating with message %p allocated but never sent.\n",
444                                        (long int)getpid(), &this );
445                }
446        )
447}
448
449static inline void check_message( message & this ) {
450    switch ( this.alloc ) {                                             // analyze message status
451        case Nodelete: CFA_DEBUG( this.alloc = Finished ); break;
452        case Delete: delete( &this ); break;
453        case Destroy: ^?{}( this ); break;
454        case Finished: break;
455    } // switch
456}
457static inline void set_allocation( message & this, allocation state ) {
458    CFA_DEBUG( if ( state == Nodelete ) state = Finished; )
459    this.alloc = state;
460}
461
462static inline void deliver_request( request & this ) {
463    DEBUG_ABORT( this.receiver->ticket == (unsigned long int)MAX, "Attempted to send message to deleted/dead actor\n" );
464    actor * base_actor;
465    message * base_msg;
466    allocation temp = this.fn( *this.receiver, *this.msg, &base_actor, &base_msg );
467    memcpy( &base_actor->alloc, &temp, sizeof(allocation) ); // optimization to elide ctor
468    check_message( *base_msg );
469    check_actor( *base_actor );
470}
471
472// tries to atomically swap two queues and returns 0p if the swap failed
473// returns ptr to newly owned queue if swap succeeds
474static inline work_queue * try_swap_queues( worker & this, unsigned int victim_idx, unsigned int my_idx ) with(this) {
475    work_queue * my_queue = request_queues[my_idx];
476    work_queue * other_queue = request_queues[victim_idx];
477
478    // if either queue is 0p then they are in the process of being stolen
479    if ( other_queue == 0p ) return 0p;
480
481    // try to set our queue ptr to be 0p. If it fails someone moved our queue so return false
482    if ( !__atomic_compare_exchange_n( &request_queues[my_idx], &my_queue, 0p, false, __ATOMIC_SEQ_CST, __ATOMIC_SEQ_CST ) )
483        return 0p;
484
485    // try to set other queue ptr to be our queue ptr. If it fails someone moved the other queue so fix up then return false
486    if ( !__atomic_compare_exchange_n( &request_queues[victim_idx], &other_queue, my_queue, false, __ATOMIC_SEQ_CST, __ATOMIC_SEQ_CST ) ) {
487        /* paranoid */ verify( request_queues[my_idx] == 0p );
488        request_queues[my_idx] = my_queue; // reset my queue ptr back to appropriate val
489        return 0p;
490    }
491
492    // we have successfully swapped and since our queue is 0p no one will touch it so write back new queue ptr non atomically
493    request_queues[my_idx] = other_queue; // last write does not need to be atomic
494    return other_queue;
495}
496
497// once a worker to steal from has been chosen, choose queue to steal from
498static inline void choose_queue( worker & this, unsigned int victim_id, unsigned int swap_idx ) with(this) {
499    // have to calculate victim start and range since victim may be deleted before us in shutdown
500    const unsigned int queues_per_worker = executor_->nrqueues / executor_->nworkers;
501    const unsigned int extras = executor_->nrqueues % executor_->nworkers;
502    unsigned int vic_start, vic_range;
503    if ( extras > victim_id  ) {
504        vic_range = queues_per_worker + 1;
505        vic_start = vic_range * victim_id;
506    } else {
507        vic_start = extras + victim_id * queues_per_worker;
508        vic_range = queues_per_worker;
509    }
510    unsigned int start_idx = prng( vic_range );
511
512    unsigned int tries = 0;
513    work_queue * curr_steal_queue;
514
515    for ( unsigned int i = start_idx; tries < vic_range; i = (i + 1) % vic_range ) {
516        tries++;
517        curr_steal_queue = request_queues[ i + vic_start ];
518        // avoid empty queues and queues that are being operated on
519        if ( curr_steal_queue == 0p || curr_steal_queue->being_processed || is_empty( *curr_steal_queue->c_queue ) )
520            continue;
521
522        #ifdef ACTOR_STATS
523        curr_steal_queue = try_swap_queues( this, i + vic_start, swap_idx );
524        if ( curr_steal_queue ) {
525            executor_->w_infos[id].msgs_stolen += curr_steal_queue->c_queue->count;
526            executor_->w_infos[id].stolen++;
527            if ( is_empty( *curr_steal_queue->c_queue ) ) executor_->w_infos[id].empty_stolen++;
528            // __atomic_add_fetch(&executor_->w_infos[victim_id].stolen_from, 1, __ATOMIC_RELAXED);
529            // replaced_queue[swap_idx]++;
530            // __atomic_add_fetch(&stolen_arr[ i + vic_start ], 1, __ATOMIC_RELAXED);
531        } else {
532            executor_->w_infos[id].failed_swaps++;
533        }
534        #else
535        curr_steal_queue = try_swap_queues( this, i + vic_start, swap_idx );
536        #endif // ACTOR_STATS
537
538        return;
539    }
540
541    return;
542}
543
544// choose a worker to steal from
545static inline void steal_work( worker & this, unsigned int swap_idx ) with(this) {
546    #if RAND
547    unsigned int victim = prng( executor_->nworkers );
548    if ( victim == id ) victim = ( victim + 1 ) % executor_->nworkers;
549    choose_queue( this, victim, swap_idx );
550    #elif SEARCH
551    unsigned long long min = MAX; // smaller timestamp means longer since service
552    int min_id = 0; // use ints not uints to avoid integer underflow without hacky math
553    int n_workers = executor_->nworkers;
554    unsigned long long curr_stamp;
555    int scount = 1;
556    for ( int i = (id + 1) % n_workers; scount < n_workers; i = (i + 1) % n_workers, scount++ ) {
557        curr_stamp = executor_->w_infos[i].stamp;
558        if ( curr_stamp < min ) {
559            min = curr_stamp;
560            min_id = i;
561        }
562    }
563    choose_queue( this, min_id, swap_idx );
564    #endif
565}
566
567#define CHECK_TERMINATION if ( unlikely( executor_->is_shutdown ) ) break Exit
568void main( worker & this ) with(this) {
569    // #ifdef ACTOR_STATS
570    // for ( i; executor_->nrqueues ) {
571    //     replaced_queue[i] = 0;
572    //     __atomic_store_n( &stolen_arr[i], 0, __ATOMIC_SEQ_CST );
573    // }
574    // #endif
575
576    // threshold of empty queues we see before we go stealing
577    const unsigned int steal_threshold = 2 * range;
578
579    // Store variable data here instead of worker struct to avoid any potential false sharing
580    unsigned int empty_count = 0;
581    request & req;
582    work_queue * curr_work_queue;
583
584    Exit:
585    for ( unsigned int i = 0;; i = (i + 1) % range ) { // cycle through set of request buffers
586        curr_work_queue = request_queues[i + start];
587
588        #ifndef __STEAL
589        CHECK_TERMINATION;
590        #endif
591
592        // check if queue is empty before trying to gulp it
593        if ( is_empty( *curr_work_queue->c_queue ) ) {
594            #ifdef __STEAL
595            empty_count++;
596            if ( empty_count < steal_threshold ) continue;
597            #else
598            continue;
599            #endif
600        }
601        transfer( *curr_work_queue, &current_queue );
602        #ifdef ACTOR_STATS
603        executor_->w_infos[id].gulps++;
604        #endif // ACTOR_STATS
605        #ifdef __STEAL
606        if ( is_empty( *current_queue ) ) {
607            if ( unlikely( no_steal ) ) { CHECK_TERMINATION; continue; }
608            empty_count++;
609            if ( empty_count < steal_threshold ) continue;
610            empty_count = 0;
611
612            CHECK_TERMINATION; // check for termination
613
614            __atomic_store_n( &executor_->w_infos[id].stamp, rdtscl(), __ATOMIC_RELAXED );
615           
616            #ifdef ACTOR_STATS
617            executor_->w_infos[id].try_steal++;
618            #endif // ACTOR_STATS
619           
620            steal_work( this, start + prng( range ) );
621            continue;
622        }
623        #endif // __STEAL
624        while ( ! is_empty( *current_queue ) ) {
625            #ifdef ACTOR_STATS
626            executor_->w_infos[id].processed++;
627            #endif
628            &req = &remove( *current_queue );
629            if ( !&req ) continue;
630            deliver_request( req );
631        }
632        #ifdef __STEAL
633        curr_work_queue->being_processed = false; // set done processing
634        empty_count = 0; // we found work so reset empty counter
635        #endif
636
637        CHECK_TERMINATION;
638       
639        // potentially reclaim some of the current queue's vector space if it is unused
640        reclaim( *current_queue );
641    } // for
642}
643
644static inline void send( executor & this, request & req, unsigned long int ticket ) with(this) {
645    insert( request_queues[ticket], req);
646}
647
648static inline void send( actor & this, request & req ) {
649    DEBUG_ABORT( this.ticket == (unsigned long int)MAX, "Attempted to send message to deleted/dead actor\n" );
650    send( *__actor_executor_, req, this.ticket );
651}
652
653static inline void __reset_stats() {
654    #ifdef ACTOR_STATS
655    __total_tries = 0;
656    __total_stolen = 0;
657    __all_gulps = 0;
658    __total_failed_swaps = 0;
659    __total_empty_stolen = 0;
660    __all_processed = 0;
661    __num_actors_stats = 0;
662    __all_msgs_stolen = 0;
663    #endif
664}
665
666static inline void start_actor_system( size_t num_thds ) {
667    __reset_stats();
668    __actor_executor_thd = active_thread();
669    __actor_executor_ = alloc();
670    (*__actor_executor_){ 0, num_thds, num_thds == 1 ? 1 : num_thds * 16 };
671}
672
673static inline void start_actor_system() { start_actor_system( get_proc_count( *active_cluster() ) ); }
674
675static inline void start_actor_system( executor & this ) {
676    __reset_stats();
677    __actor_executor_thd = active_thread();
678    __actor_executor_ = &this;
679    __actor_executor_passed = true;
680}
681
682static inline void stop_actor_system() {
683    park( ); // unparked when actor system is finished
684
685    if ( !__actor_executor_passed ) delete( __actor_executor_ );
686    __actor_executor_ = 0p;
687    __actor_executor_thd = 0p;
688    __next_ticket = 0;
689    __actor_executor_passed = false;
690}
691
692// Default messages to send to any actor to change status
693// assigned at creation to __base_msg_finished to avoid unused message warning
694message __base_msg_finished @= { .alloc : Finished };
695struct delete_message_t { inline message; } delete_msg = __base_msg_finished;
696struct destroy_msg_t { inline message; } destroy_msg = __base_msg_finished;
697struct finished_msg_t { inline message; } finished_msg = __base_msg_finished;
698
699allocation receive( actor & this, delete_message_t & msg ) { return Delete; }
700allocation receive( actor & this, destroy_msg_t & msg ) { return Destroy; }
701allocation receive( actor & this, finished_msg_t & msg ) { return Finished; }
702
Note: See TracBrowser for help on using the repository browser.