13 #ifndef KMP_WAIT_RELEASE_H
14 #define KMP_WAIT_RELEASE_H
18 #include "kmp_stats.h"
20 #include "ompt-specific.h"
36 struct flag_properties {
37 unsigned int type : 16;
38 unsigned int reserved : 16;
41 template <enum flag_type FlagType>
struct flag_traits {};
43 template <>
struct flag_traits<flag32> {
44 typedef kmp_uint32 flag_t;
45 static const flag_type t = flag32;
46 static inline flag_t tcr(flag_t f) {
return TCR_4(f); }
47 static inline flag_t test_then_add4(
volatile flag_t *f) {
48 return KMP_TEST_THEN_ADD4_32(RCAST(
volatile kmp_int32 *, f));
50 static inline flag_t test_then_or(
volatile flag_t *f, flag_t v) {
51 return KMP_TEST_THEN_OR32(f, v);
53 static inline flag_t test_then_and(
volatile flag_t *f, flag_t v) {
54 return KMP_TEST_THEN_AND32(f, v);
58 template <>
struct flag_traits<atomic_flag64> {
59 typedef kmp_uint64 flag_t;
60 static const flag_type t = atomic_flag64;
61 static inline flag_t tcr(flag_t f) {
return TCR_8(f); }
62 static inline flag_t test_then_add4(
volatile flag_t *f) {
63 return KMP_TEST_THEN_ADD4_64(RCAST(
volatile kmp_int64 *, f));
65 static inline flag_t test_then_or(
volatile flag_t *f, flag_t v) {
66 return KMP_TEST_THEN_OR64(f, v);
68 static inline flag_t test_then_and(
volatile flag_t *f, flag_t v) {
69 return KMP_TEST_THEN_AND64(f, v);
73 template <>
struct flag_traits<flag64> {
74 typedef kmp_uint64 flag_t;
75 static const flag_type t = flag64;
76 static inline flag_t tcr(flag_t f) {
return TCR_8(f); }
77 static inline flag_t test_then_add4(
volatile flag_t *f) {
78 return KMP_TEST_THEN_ADD4_64(RCAST(
volatile kmp_int64 *, f));
80 static inline flag_t test_then_or(
volatile flag_t *f, flag_t v) {
81 return KMP_TEST_THEN_OR64(f, v);
83 static inline flag_t test_then_and(
volatile flag_t *f, flag_t v) {
84 return KMP_TEST_THEN_AND64(f, v);
88 template <>
struct flag_traits<flag_oncore> {
89 typedef kmp_uint64 flag_t;
90 static const flag_type t = flag_oncore;
91 static inline flag_t tcr(flag_t f) {
return TCR_8(f); }
92 static inline flag_t test_then_add4(
volatile flag_t *f) {
93 return KMP_TEST_THEN_ADD4_64(RCAST(
volatile kmp_int64 *, f));
95 static inline flag_t test_then_or(
volatile flag_t *f, flag_t v) {
96 return KMP_TEST_THEN_OR64(f, v);
98 static inline flag_t test_then_and(
volatile flag_t *f, flag_t v) {
99 return KMP_TEST_THEN_AND64(f, v);
109 std::atomic<bool> *sleepLoc;
112 typedef flag_traits<FlagType> traits_type;
135 enum barrier_type get_bt() {
return bs_last_barrier; }
139 template <
typename PtrType, flag_type FlagType,
bool Sleepable>
142 volatile PtrType *loc;
144 typedef flag_traits<FlagType> traits_type;
147 typedef PtrType flag_t;
155 kmp_flag_native(
volatile PtrType *p, PtrType c, std::atomic<bool> *sloc)
157 volatile PtrType *get() {
return loc; }
158 void *get_void_p() {
return RCAST(
void *, CCAST(PtrType *, loc)); }
159 void set(
volatile PtrType *new_loc) { loc = new_loc; }
160 PtrType load() {
return *loc; }
161 void store(PtrType val) { *loc = val; }
164 if (Sleepable && !(this->sleepLoc))
165 return (traits_type::tcr(*(this->get())) & ~KMP_BARRIER_SLEEP_STATE) ==
168 return traits_type::tcr(*(this->get())) ==
checker;
179 return traits_type::tcr(*(this->get())) !=
checker;
184 (void)traits_type::test_then_add4((
volatile PtrType *)this->get());
190 if (this->sleepLoc) {
191 this->sleepLoc->store(
true);
192 return *(this->get());
194 return traits_type::test_then_or((
volatile PtrType *)this->get(),
195 KMP_BARRIER_SLEEP_STATE);
201 if (this->sleepLoc) {
202 this->sleepLoc->store(
false);
205 traits_type::test_then_and((
volatile PtrType *)this->get(),
206 ~KMP_BARRIER_SLEEP_STATE);
212 return this->sleepLoc->load();
213 return old_loc & KMP_BARRIER_SLEEP_STATE;
218 return this->sleepLoc->load();
221 bool is_any_sleeping() {
223 return this->sleepLoc->load();
226 kmp_uint8 *get_stolen() {
return NULL; }
230 template <
typename PtrType, flag_type FlagType,
bool Sleepable>
233 std::atomic<PtrType> *
loc;
236 typedef flag_traits<FlagType> traits_type;
237 typedef PtrType flag_t;
245 kmp_flag_atomic(std::atomic<PtrType> *p, PtrType c, std::atomic<bool> *sloc)
248 std::atomic<PtrType> *
get() {
return loc; }
252 void set(std::atomic<PtrType> *new_loc) {
loc = new_loc; }
254 PtrType
load() {
return loc->load(std::memory_order_acquire); }
256 void store(PtrType val) {
loc->store(val, std::memory_order_release); }
259 if (Sleepable && !(this->sleepLoc))
260 return (this->
load() & ~KMP_BARRIER_SLEEP_STATE) ==
checker;
280 if (this->sleepLoc) {
281 this->sleepLoc->store(
true);
282 return *(this->
get());
284 return KMP_ATOMIC_OR(this->
get(), KMP_BARRIER_SLEEP_STATE);
290 if (this->sleepLoc) {
291 this->sleepLoc->store(
false);
294 KMP_ATOMIC_AND(this->
get(), ~KMP_BARRIER_SLEEP_STATE);
300 return this->sleepLoc->load();
301 return old_loc & KMP_BARRIER_SLEEP_STATE;
306 return this->sleepLoc->load();
309 bool is_any_sleeping() {
311 return this->sleepLoc->load();
314 kmp_uint8 *get_stolen() {
return NULL; }
319 static void __ompt_implicit_task_end(kmp_info_t *this_thr,
320 ompt_state_t ompt_state,
322 int ds_tid = this_thr->th.th_info.ds.ds_tid;
323 if (ompt_state == ompt_state_wait_barrier_implicit) {
324 this_thr->th.ompt_thread_info.state = ompt_state_overhead;
326 void *codeptr = NULL;
327 if (ompt_enabled.ompt_callback_sync_region_wait) {
328 ompt_callbacks.ompt_callback(ompt_callback_sync_region_wait)(
329 ompt_sync_region_barrier_implicit, ompt_scope_end, NULL, tId,
332 if (ompt_enabled.ompt_callback_sync_region) {
333 ompt_callbacks.ompt_callback(ompt_callback_sync_region)(
334 ompt_sync_region_barrier_implicit, ompt_scope_end, NULL, tId,
338 if (!KMP_MASTER_TID(ds_tid)) {
339 if (ompt_enabled.ompt_callback_implicit_task) {
340 int flags = this_thr->th.ompt_thread_info.parallel_flags;
341 flags = (flags & ompt_parallel_league) ? ompt_task_initial
342 : ompt_task_implicit;
343 ompt_callbacks.ompt_callback(ompt_callback_implicit_task)(
344 ompt_scope_end, NULL, tId, 0, ds_tid, flags);
347 this_thr->th.ompt_thread_info.state = ompt_state_idle;
349 this_thr->th.ompt_thread_info.state = ompt_state_overhead;
360 template <
class C,
bool final_spin,
bool Cancellable =
false,
361 bool Sleepable =
true>
363 __kmp_wait_template(kmp_info_t *this_thr,
364 C *flag USE_ITT_BUILD_ARG(
void *itt_sync_obj)) {
365 #if USE_ITT_BUILD && USE_ITT_NOTIFY
366 volatile void *spin = flag->get();
370 int tasks_completed = FALSE;
372 kmp_uint64 poll_count;
373 kmp_uint64 hibernate_goal;
375 kmp_uint32 hibernate;
378 KMP_FSYNC_SPIN_INIT(spin, NULL);
379 if (flag->done_check()) {
380 KMP_FSYNC_SPIN_ACQUIRED(CCAST(
void *, spin));
383 th_gtid = this_thr->th.th_info.ds.ds_gtid;
385 kmp_team_t *team = this_thr->th.th_team;
386 if (team && team->t.t_cancel_request == cancel_parallel)
391 KMP_ATOMIC_ST_REL(&this_thr->th.th_blocking,
true);
394 (
"__kmp_wait_sleep: T#%d waiting for flag(%p)\n", th_gtid, flag));
395 #if KMP_STATS_ENABLED
450 ompt_state_t ompt_entry_state;
452 if (ompt_enabled.enabled) {
453 ompt_entry_state = this_thr->th.ompt_thread_info.state;
454 if (!final_spin || ompt_entry_state != ompt_state_wait_barrier_implicit ||
455 KMP_MASTER_TID(this_thr->th.th_info.ds.ds_tid)) {
456 ompt_lw_taskteam_t *team = NULL;
457 if (this_thr->th.th_team)
458 team = this_thr->th.th_team->t.ompt_serialized_team_info;
460 tId = &(team->ompt_task_info.task_data);
462 tId = OMPT_CUR_TASK_DATA(this_thr);
465 tId = &(this_thr->th.ompt_thread_info.task_data);
467 if (final_spin && (__kmp_tasking_mode == tskm_immediate_exec ||
468 this_thr->th.th_task_team == NULL)) {
470 __ompt_implicit_task_end(this_thr, ompt_entry_state, tId);
475 KMP_INIT_YIELD(spins);
477 if (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME ||
478 __kmp_pause_status == kmp_soft_paused) {
482 #ifdef KMP_ADJUST_BLOCKTIME
483 if (__kmp_pause_status == kmp_soft_paused ||
484 (__kmp_zero_bt && !this_thr->th.th_team_bt_set))
489 hibernate = this_thr->th.th_team_bt_intervals;
491 hibernate = this_thr->th.th_team_bt_intervals;
502 hibernate += TCR_4(__kmp_global.g.g_time.dt.t_value);
503 KF_TRACE(20, (
"__kmp_wait_sleep: T#%d now=%d, hibernate=%d, intervals=%d\n",
504 th_gtid, __kmp_global.g.g_time.dt.t_value, hibernate,
505 hibernate - __kmp_global.g.g_time.dt.t_value));
507 if (__kmp_pause_status == kmp_soft_paused) {
509 hibernate_goal = KMP_NOW();
511 hibernate_goal = KMP_NOW() + this_thr->th.th_team_bt_intervals;
520 while (flag->notdone_check()) {
521 kmp_task_team_t *task_team = NULL;
522 if (__kmp_tasking_mode != tskm_immediate_exec) {
523 task_team = this_thr->th.th_task_team;
531 if (task_team != NULL) {
532 if (TCR_SYNC_4(task_team->tt.tt_active)) {
533 if (KMP_TASKING_ENABLED(task_team)) {
535 this_thr, th_gtid, final_spin,
536 &tasks_completed USE_ITT_BUILD_ARG(itt_sync_obj), 0);
538 this_thr->th.th_reap_state = KMP_SAFE_TO_REAP;
540 KMP_DEBUG_ASSERT(!KMP_MASTER_TID(this_thr->th.th_info.ds.ds_tid));
543 if (final_spin && ompt_enabled.enabled)
544 __ompt_implicit_task_end(this_thr, ompt_entry_state, tId);
546 this_thr->th.th_task_team = NULL;
547 this_thr->th.th_reap_state = KMP_SAFE_TO_REAP;
550 this_thr->th.th_reap_state = KMP_SAFE_TO_REAP;
554 KMP_FSYNC_SPIN_PREPARE(CCAST(
void *, spin));
555 if (TCR_4(__kmp_global.g.g_done)) {
556 if (__kmp_global.g.g_abort)
557 __kmp_abort_thread();
563 KMP_YIELD_OVERSUB_ELSE_SPIN(spins);
565 #if KMP_STATS_ENABLED
568 if (this_thr->th.th_stats->isIdle() &&
569 KMP_GET_THREAD_STATE() == FORK_JOIN_BARRIER) {
570 KMP_SET_THREAD_STATE(IDLE);
571 KMP_PUSH_PARTITIONED_TIMER(OMP_idle);
576 kmp_team_t *team = this_thr->th.th_team;
577 if (team && team->t.t_cancel_request == cancel_parallel)
591 if (task_team && KMP_HIDDEN_HELPER_WORKER_THREAD(th_gtid) &&
592 !TCR_4(__kmp_hidden_helper_team_done)) {
595 if (KMP_ATOMIC_LD_ACQ(&__kmp_unexecuted_hidden_helper_tasks) == 0) {
596 __kmp_hidden_helper_worker_thread_wait();
602 if (__kmp_dflt_blocktime == KMP_MAX_BLOCKTIME &&
603 __kmp_pause_status != kmp_soft_paused)
607 if ((task_team != NULL) && TCR_4(task_team->tt.tt_found_tasks))
612 if (TCR_4(__kmp_global.g.g_time.dt.t_value) < hibernate)
615 if (KMP_BLOCKING(hibernate_goal, poll_count++))
623 if (__kmp_dflt_blocktime == KMP_MAX_BLOCKTIME &&
624 __kmp_pause_status != kmp_soft_paused)
627 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
628 if (__kmp_mwait_enabled || __kmp_umwait_enabled) {
629 KF_TRACE(50, (
"__kmp_wait_sleep: T#%d using monitor/mwait\n", th_gtid));
630 flag->mwait(th_gtid);
633 KF_TRACE(50, (
"__kmp_wait_sleep: T#%d suspend time reached\n", th_gtid));
636 KMP_ATOMIC_ST_REL(&this_thr->th.th_blocking,
false);
638 flag->suspend(th_gtid);
641 KMP_ATOMIC_ST_REL(&this_thr->th.th_blocking,
true);
643 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
647 if (TCR_4(__kmp_global.g.g_done)) {
648 if (__kmp_global.g.g_abort)
649 __kmp_abort_thread();
651 }
else if (__kmp_tasking_mode != tskm_immediate_exec &&
652 this_thr->th.th_reap_state == KMP_SAFE_TO_REAP) {
653 this_thr->th.th_reap_state = KMP_NOT_SAFE_TO_REAP;
659 ompt_state_t ompt_exit_state = this_thr->th.ompt_thread_info.state;
660 if (ompt_enabled.enabled && ompt_exit_state != ompt_state_undefined) {
663 __ompt_implicit_task_end(this_thr, ompt_exit_state, tId);
664 ompt_exit_state = this_thr->th.ompt_thread_info.state;
667 if (ompt_exit_state == ompt_state_idle) {
668 this_thr->th.ompt_thread_info.state = ompt_state_overhead;
672 #if KMP_STATS_ENABLED
674 if (KMP_GET_THREAD_STATE() == IDLE) {
675 KMP_POP_PARTITIONED_TIMER();
676 KMP_SET_THREAD_STATE(thread_state);
677 this_thr->th.th_stats->resetIdleFlag();
683 KMP_ATOMIC_ST_REL(&this_thr->th.th_blocking,
false);
685 KMP_FSYNC_SPIN_ACQUIRED(CCAST(
void *, spin));
687 kmp_team_t *team = this_thr->th.th_team;
688 if (team && team->t.t_cancel_request == cancel_parallel) {
689 if (tasks_completed) {
692 kmp_task_team_t *task_team = this_thr->th.th_task_team;
693 std::atomic<kmp_int32> *unfinished_threads =
694 &(task_team->tt.tt_unfinished_threads);
695 KMP_ATOMIC_INC(unfinished_threads);
703 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
707 static inline void __kmp_mwait_template(
int th_gtid, C *flag) {
708 KMP_TIME_DEVELOPER_PARTITIONED_BLOCK(USER_mwait);
709 kmp_info_t *th = __kmp_threads[th_gtid];
711 KF_TRACE(30, (
"__kmp_mwait_template: T#%d enter for flag = %p\n", th_gtid,
715 KMP_DEBUG_ASSERT(__kmp_mwait_enabled || __kmp_umwait_enabled);
717 __kmp_suspend_initialize_thread(th);
718 __kmp_lock_suspend_mx(th);
720 volatile void *spin = flag->get();
721 void *cacheline = (
void *)(kmp_uintptr_t(spin) & ~(CACHE_LINE - 1));
723 if (!flag->done_check()) {
725 th->th.th_active = FALSE;
726 if (th->th.th_active_in_pool) {
727 th->th.th_active_in_pool = FALSE;
728 KMP_ATOMIC_DEC(&__kmp_thread_pool_active_nth);
729 KMP_DEBUG_ASSERT(TCR_4(__kmp_thread_pool_active_nth) >= 0);
731 flag->set_sleeping();
732 KF_TRACE(50, (
"__kmp_mwait_template: T#%d calling monitor\n", th_gtid));
734 if (__kmp_umwait_enabled) {
735 __kmp_umonitor(cacheline);
738 if (__kmp_mwait_enabled) {
739 __kmp_mm_monitor(cacheline, 0, 0);
745 if (flag->done_check())
746 flag->unset_sleeping();
749 TCW_PTR(th->th.th_sleep_loc, (
void *)flag);
750 th->th.th_sleep_loc_type = flag->get_type();
751 __kmp_unlock_suspend_mx(th);
752 KF_TRACE(50, (
"__kmp_mwait_template: T#%d calling mwait\n", th_gtid));
754 if (__kmp_umwait_enabled) {
755 __kmp_umwait(1, 100);
758 if (__kmp_mwait_enabled) {
759 __kmp_mm_mwait(0, __kmp_mwait_hints);
762 KF_TRACE(50, (
"__kmp_mwait_template: T#%d mwait done\n", th_gtid));
763 __kmp_lock_suspend_mx(th);
765 if (flag->is_sleeping())
766 flag->unset_sleeping();
767 TCW_PTR(th->th.th_sleep_loc, NULL);
768 th->th.th_sleep_loc_type = flag_unset;
771 th->th.th_active = TRUE;
772 if (TCR_4(th->th.th_in_pool)) {
773 KMP_ATOMIC_INC(&__kmp_thread_pool_active_nth);
774 th->th.th_active_in_pool = TRUE;
777 __kmp_unlock_suspend_mx(th);
778 KF_TRACE(30, (
"__kmp_mwait_template: T#%d exit\n", th_gtid));
786 template <
class C>
static inline void __kmp_release_template(C *flag) {
788 int gtid = TCR_4(__kmp_init_gtid) ? __kmp_get_gtid() : -1;
790 KF_TRACE(20, (
"__kmp_release: T#%d releasing flag(%x)\n", gtid, flag->get()));
791 KMP_DEBUG_ASSERT(flag->get());
792 KMP_FSYNC_RELEASING(flag->get_void_p());
794 flag->internal_release();
796 KF_TRACE(100, (
"__kmp_release: T#%d set new spin=%d\n", gtid, flag->get(),
799 if (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME) {
802 if (flag->is_any_sleeping()) {
803 for (
unsigned int i = 0; i < flag->get_num_waiters(); ++i) {
805 kmp_info_t *waiter = flag->get_waiter(i);
807 int wait_gtid = waiter->th.th_info.ds.ds_gtid;
809 KF_TRACE(50, (
"__kmp_release: T#%d waking up thread T#%d since sleep "
811 gtid, wait_gtid, flag->get()));
812 flag->resume(wait_gtid);
819 template <
bool Cancellable,
bool Sleepable>
820 class kmp_flag_32 :
public kmp_flag_atomic<kmp_uint32, flag32, Sleepable> {
822 kmp_flag_32(std::atomic<kmp_uint32> *p)
824 kmp_flag_32(std::atomic<kmp_uint32> *p, kmp_info_t *thr)
826 kmp_flag_32(std::atomic<kmp_uint32> *p, kmp_uint32 c)
828 void suspend(
int th_gtid) { __kmp_suspend_32(th_gtid,
this); }
829 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
830 void mwait(
int th_gtid) { __kmp_mwait_32(th_gtid,
this); }
832 void resume(
int th_gtid) { __kmp_resume_32(th_gtid,
this); }
833 int execute_tasks(kmp_info_t *this_thr, kmp_int32 gtid,
int final_spin,
834 int *thread_finished USE_ITT_BUILD_ARG(
void *itt_sync_obj),
835 kmp_int32 is_constrained) {
836 return __kmp_execute_tasks_32(
837 this_thr, gtid,
this, final_spin,
838 thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
840 bool wait(kmp_info_t *this_thr,
841 int final_spin USE_ITT_BUILD_ARG(
void *itt_sync_obj)) {
843 return __kmp_wait_template<kmp_flag_32, TRUE, Cancellable, Sleepable>(
844 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
846 return __kmp_wait_template<kmp_flag_32, FALSE, Cancellable, Sleepable>(
847 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
849 void release() { __kmp_release_template(
this); }
850 flag_type get_ptr_type() {
return flag32; }
853 template <
bool Cancellable,
bool Sleepable>
854 class kmp_flag_64 :
public kmp_flag_native<kmp_uint64, flag64, Sleepable> {
856 kmp_flag_64(
volatile kmp_uint64 *p)
858 kmp_flag_64(
volatile kmp_uint64 *p, kmp_info_t *thr)
860 kmp_flag_64(
volatile kmp_uint64 *p, kmp_uint64 c)
862 kmp_flag_64(
volatile kmp_uint64 *p, kmp_uint64 c, std::atomic<bool> *loc)
864 void suspend(
int th_gtid) { __kmp_suspend_64(th_gtid,
this); }
865 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
866 void mwait(
int th_gtid) { __kmp_mwait_64(th_gtid,
this); }
868 void resume(
int th_gtid) { __kmp_resume_64(th_gtid,
this); }
869 int execute_tasks(kmp_info_t *this_thr, kmp_int32 gtid,
int final_spin,
870 int *thread_finished USE_ITT_BUILD_ARG(
void *itt_sync_obj),
871 kmp_int32 is_constrained) {
872 return __kmp_execute_tasks_64(
873 this_thr, gtid,
this, final_spin,
874 thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
876 bool wait(kmp_info_t *this_thr,
877 int final_spin USE_ITT_BUILD_ARG(
void *itt_sync_obj)) {
879 return __kmp_wait_template<kmp_flag_64, TRUE, Cancellable, Sleepable>(
880 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
882 return __kmp_wait_template<kmp_flag_64, FALSE, Cancellable, Sleepable>(
883 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
885 void release() { __kmp_release_template(
this); }
886 flag_type get_ptr_type() {
return flag64; }
889 template <
bool Cancellable,
bool Sleepable>
890 class kmp_atomic_flag_64
893 kmp_atomic_flag_64(std::atomic<kmp_uint64> *p)
895 kmp_atomic_flag_64(std::atomic<kmp_uint64> *p, kmp_info_t *thr)
897 kmp_atomic_flag_64(std::atomic<kmp_uint64> *p, kmp_uint64 c)
899 kmp_atomic_flag_64(std::atomic<kmp_uint64> *p, kmp_uint64 c,
900 std::atomic<bool> *loc)
902 void suspend(
int th_gtid) { __kmp_atomic_suspend_64(th_gtid,
this); }
903 void mwait(
int th_gtid) { __kmp_atomic_mwait_64(th_gtid,
this); }
904 void resume(
int th_gtid) { __kmp_atomic_resume_64(th_gtid,
this); }
905 int execute_tasks(kmp_info_t *this_thr, kmp_int32 gtid,
int final_spin,
906 int *thread_finished USE_ITT_BUILD_ARG(
void *itt_sync_obj),
907 kmp_int32 is_constrained) {
908 return __kmp_atomic_execute_tasks_64(
909 this_thr, gtid,
this, final_spin,
910 thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
912 bool wait(kmp_info_t *this_thr,
913 int final_spin USE_ITT_BUILD_ARG(
void *itt_sync_obj)) {
915 return __kmp_wait_template<kmp_atomic_flag_64, TRUE, Cancellable,
917 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
919 return __kmp_wait_template<kmp_atomic_flag_64, FALSE, Cancellable,
921 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
923 void release() { __kmp_release_template(
this); }
924 flag_type get_ptr_type() {
return atomic_flag64; }
928 class kmp_flag_oncore :
public kmp_flag_native<kmp_uint64, flag_oncore, false> {
931 enum barrier_type bt;
932 kmp_info_t *this_thr;
936 unsigned char &byteref(
volatile kmp_uint64 *loc,
size_t offset) {
937 return (RCAST(
unsigned char *, CCAST(kmp_uint64 *, loc)))[offset];
941 kmp_flag_oncore(
volatile kmp_uint64 *p)
942 :
kmp_flag_native<kmp_uint64, flag_oncore, false>(p), flag_switch(false) {
944 kmp_flag_oncore(
volatile kmp_uint64 *p, kmp_uint32 idx)
946 flag_switch(false), bt(bs_last_barrier), itt_sync_obj(nullptr) {}
947 kmp_flag_oncore(
volatile kmp_uint64 *p, kmp_uint64 c, kmp_uint32 idx,
948 enum barrier_type bar_t,
949 kmp_info_t *thr USE_ITT_BUILD_ARG(
void *itt))
951 flag_switch(false), bt(bar_t),
952 this_thr(thr) USE_ITT_BUILD_ARG(itt_sync_obj(itt)) {}
954 return byteref(&old_loc, offset) ==
checker;
959 if (this_thr->th.th_bar[bt].bb.wait_flag == KMP_BARRIER_SWITCH_TO_OWN_FLAG)
961 if (byteref(get(), offset) != 1 && !flag_switch)
963 else if (flag_switch) {
964 this_thr->th.th_bar[bt].bb.wait_flag = KMP_BARRIER_SWITCHING;
965 kmp_flag_64<> flag(&this_thr->th.th_bar[bt].bb.b_go,
966 (kmp_uint64)KMP_BARRIER_STATE_BUMP);
967 __kmp_wait_64(this_thr, &flag, TRUE USE_ITT_BUILD_ARG(itt_sync_obj));
973 if (__kmp_dflt_blocktime == KMP_MAX_BLOCKTIME) {
974 byteref(get(), offset) = 1;
977 byteref(&mask, offset) = 1;
978 KMP_TEST_THEN_OR64(get(), mask);
981 void wait(kmp_info_t *this_thr,
int final_spin) {
983 __kmp_wait_template<kmp_flag_oncore, TRUE>(
984 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
986 __kmp_wait_template<kmp_flag_oncore, FALSE>(
987 this_thr,
this USE_ITT_BUILD_ARG(itt_sync_obj));
989 void release() { __kmp_release_template(
this); }
990 void suspend(
int th_gtid) { __kmp_suspend_oncore(th_gtid,
this); }
991 #if KMP_HAVE_MWAIT || KMP_HAVE_UMWAIT
992 void mwait(
int th_gtid) { __kmp_mwait_oncore(th_gtid,
this); }
994 void resume(
int th_gtid) { __kmp_resume_oncore(th_gtid,
this); }
995 int execute_tasks(kmp_info_t *this_thr, kmp_int32 gtid,
int final_spin,
996 int *thread_finished USE_ITT_BUILD_ARG(
void *itt_sync_obj),
997 kmp_int32 is_constrained) {
999 int ret = __kmp_execute_tasks_oncore(
1000 this_thr, gtid,
this, final_spin,
1001 thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
1002 if (ompd_state & OMPD_ENABLE_BP)
1006 return __kmp_execute_tasks_oncore(
1007 this_thr, gtid,
this, final_spin,
1008 thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
1011 enum barrier_type get_bt() {
return bt; }
1012 flag_type get_ptr_type() {
return flag_oncore; }
1015 static inline void __kmp_null_resume_wrapper(kmp_info_t *thr) {
1016 int gtid = __kmp_gtid_from_thread(thr);
1017 void *flag = CCAST(
void *, thr->th.th_sleep_loc);
1018 flag_type type = thr->th.th_sleep_loc_type;
1024 __kmp_resume_32(gtid, RCAST(kmp_flag_32<> *, flag));
1027 __kmp_resume_64(gtid, RCAST(kmp_flag_64<> *, flag));
1030 __kmp_atomic_resume_64(gtid, RCAST(kmp_atomic_flag_64<> *, flag));
1033 __kmp_resume_oncore(gtid, RCAST(kmp_flag_oncore *, flag));
1037 KF_TRACE(100, (
"__kmp_null_resume_wrapper: flag type %d is unset\n", type));
1040 KF_TRACE(100, (
"__kmp_null_resume_wrapper: flag type %d does not match any "
1041 "known flag type\n",
std::atomic< PtrType > * loc
bool is_sleeping_val(PtrType old_loc)
bool done_check_val(PtrType old_loc)
void set(std::atomic< PtrType > *new_loc)
std::atomic< PtrType > * get()
bool is_sleeping_val(PtrType old_loc)
virtual bool notdone_check()
virtual bool done_check_val(PtrType old_loc)
virtual bool done_check()
kmp_uint32 num_waiting_threads
kmp_info_t * waiting_threads[1]
kmp_uint32 get_num_waiters()
kmp_info_t * get_waiter(kmp_uint32 i)
void set_waiter(kmp_info_t *thr)
stats_state_e
the states which a thread can be in