ARGOBOTS  66b1c39742507d8df30e8d28c54839b961a14814
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Macros Groups
ythread_htable.c
Go to the documentation of this file.
1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
2 /*
3  * See COPYRIGHT in top-level directory.
4  */
5 
6 #include "abti.h"
7 #include "abti_ythread_htable.h"
8 
10  ABTI_ythread_htable **pp_htable)
11 {
12  ABTI_STATIC_ASSERT(sizeof(ABTI_ythread_queue) == 192);
13 
14  int abt_errno;
15  ABTI_ythread_htable *p_htable;
16  size_t q_size = num_rows * sizeof(ABTI_ythread_queue);
17 
18  abt_errno = ABTU_malloc(sizeof(ABTI_ythread_htable), (void **)&p_htable);
19  ABTI_CHECK_ERROR(abt_errno);
20 
21  abt_errno = ABTU_memalign(64, q_size, (void **)&p_htable->queue);
22  if (ABTI_IS_ERROR_CHECK_ENABLED && abt_errno != ABT_SUCCESS) {
23  ABTU_free(p_htable);
24  return abt_errno;
25  }
26  memset(p_htable->queue, 0, q_size);
27 
28 #if defined(HAVE_LH_LOCK_H)
29  lh_lock_init(&p_htable->mutex);
30 #elif defined(HAVE_CLH_H)
31  clh_init(&p_htable->mutex);
32 #elif defined(USE_PTHREAD_MUTEX)
33  int ret = pthread_mutex_init(&p_htable->mutex, NULL);
34  if (ret) {
35  ABTU_free(p_htable->queue);
36  ABTU_free(p_htable);
37  return ABT_ERR_OTHER;
38  }
39 #else
40  ABTI_spinlock_clear(&p_htable->mutex);
41 #endif
43  p_htable->num_rows = num_rows;
44  p_htable->h_list = NULL;
45  p_htable->l_list = NULL;
46  *pp_htable = p_htable;
47  return ABT_SUCCESS;
48 }
49 
51 {
53 
54 #if defined(HAVE_LH_LOCK_H)
55  lh_lock_destroy(&p_htable->mutex);
56 #elif defined(HAVE_CLH_H)
57  clh_destroy(&p_htable->mutex);
58 #elif defined(USE_PTHREAD_MUTEX)
59  int ret = pthread_mutex_destroy(&p_htable->mutex);
60  assert(!ret);
61 #else
62  /* ABTI_spinlock needs no finalization. */
63 #endif
64  ABTU_free(p_htable->queue);
65  ABTU_free(p_htable);
66 }
67 
69  ABTI_ythread *p_ythread)
70 {
71  ABTI_ythread_queue *p_queue;
72 
73  if (idx >= p_htable->num_rows) {
74  ABTI_ASSERT(0);
76  }
77 
78  /* Add p_ythread to the end of the idx-th row */
79  p_queue = &p_htable->queue[idx];
81  if (p_queue->head == NULL) {
82  p_queue->head = p_ythread;
83  p_queue->tail = p_ythread;
84  } else {
85  p_queue->tail->thread.p_next = &p_ythread->thread;
86  p_queue->tail = p_ythread;
87  }
88  p_queue->num_threads++;
91 }
92 
94  ABTI_ythread *p_ythread)
95 {
96  ABTI_ythread_queue *p_queue;
97 
98  if (idx >= p_htable->num_rows) {
99  ABTI_ASSERT(0);
101  }
102 
103  /* Add p_ythread to the end of the idx-th row */
104  p_queue = &p_htable->queue[idx];
106  if (p_queue->low_head == NULL) {
107  p_queue->low_head = p_ythread;
108  p_queue->low_tail = p_ythread;
109  } else {
110  p_queue->low_tail->thread.p_next = &p_ythread->thread;
111  p_queue->low_tail = p_ythread;
112  }
113  p_queue->low_num_threads++;
116 }
117 
119  ABTI_ythread_queue *p_queue)
120 {
121  ABTI_ythread *p_ythread = NULL;
122 
124  if (p_queue->head) {
126  p_ythread = p_queue->head;
127  if (p_queue->head == p_queue->tail) {
128  p_queue->head = NULL;
129  p_queue->tail = NULL;
130  } else {
131  p_queue->head = ABTI_thread_get_ythread(p_ythread->thread.p_next);
132  }
133 
134  p_queue->num_threads--;
135  }
137 
138  return p_ythread;
139 }
140 
142  ABTI_ythread_queue *p_queue)
143 {
144  ABTI_ythread *p_ythread = NULL;
145 
147  if (p_queue->low_head) {
149  p_ythread = p_queue->low_head;
150  if (p_queue->low_head == p_queue->low_tail) {
151  p_queue->low_head = NULL;
152  p_queue->low_tail = NULL;
153  } else {
154  p_queue->low_head =
156  }
157 
158  p_queue->low_num_threads--;
159  }
161 
162  return p_ythread;
163 }
164 
166  ABTI_ythread_queue *p_queue,
167  ABTI_ythread *p_ythread,
168  ABTI_ythread_htable *p_htable,
169  ABT_sync_event_type sync_event_type,
170  void *p_sync)
171 {
172  ABTI_ythread *p_target = NULL;
173  ABTI_xstream *p_local_xstream = *pp_local_xstream;
174 
176  if (p_queue->low_head) {
177  p_target = p_queue->low_head;
178 
179  /* Push p_ythread to the queue */
182  ABTI_tool_event_ythread_suspend(p_local_xstream, p_ythread,
183  p_ythread->thread.p_parent,
184  sync_event_type, p_sync);
185  if (p_queue->low_head == p_queue->low_tail) {
186  p_queue->low_head = p_ythread;
187  p_queue->low_tail = p_ythread;
188  } else {
189  p_queue->low_head =
191  p_queue->low_tail->thread.p_next = &p_ythread->thread;
192  p_queue->low_tail = p_ythread;
193  }
194  }
196 
197  if (p_target) {
198  LOG_DEBUG("switch -> U%" PRIu64 "\n",
199  ABTI_thread_get_id(&p_target->thread));
200 
201  /* Context-switch to p_target */
205  p_target,
206  p_local_xstream
207  ? p_local_xstream->p_thread
208  : NULL);
209  ABTI_ythread *p_prev =
210  ABTI_ythread_context_switch_to_sibling(pp_local_xstream, p_ythread,
211  p_target);
212  ABTI_tool_event_thread_run(*pp_local_xstream, &p_ythread->thread,
213  &p_prev->thread, p_ythread->thread.p_parent);
214  return ABT_TRUE;
215  } else {
216  return ABT_FALSE;
217  }
218 }
static void ABTI_ythread_queue_acquire_mutex(ABTI_ythread_queue *p_queue)
void ABTI_ythread_htable_push(ABTI_ythread_htable *p_htable, int idx, ABTI_ythread *p_ythread)
ABTU_ret_err int ABTI_ythread_htable_create(uint32_t num_rows, ABTI_ythread_htable **pp_htable)
Definition: ythread_htable.c:9
#define ABTU_unreachable()
Definition: abtu.h:25
static void ABTD_atomic_release_store_int(ABTD_atomic_int *ptr, int val)
Definition: abtd_atomic.h:924
static void ABTI_ythread_queue_acquire_low_mutex(ABTI_ythread_queue *p_queue)
ABTI_thread * p_thread
Definition: abti.h:251
int ABT_bool
Definition: abt.h:373
static void ABTI_spinlock_clear(ABTI_spinlock *p_lock)
Definition: abti_spinlock.h:18
struct ABTI_ythread_queue ABTI_ythread_queue
Definition: abti.h:117
static void ABTI_ythread_queue_release_mutex(ABTI_ythread_queue *p_queue)
#define ABTI_tool_event_ythread_suspend(p_local_xstream, p_ythread, p_parent,sync_event_type, p_sync)
Definition: abti_tool.h:373
#define ABT_ERR_OTHER
Definition: abt.h:67
ABTI_ythread * low_head
ABTD_atomic_uint32 num_elems
ABT_unit_id ABTI_thread_get_id(ABTI_thread *p_thread)
Definition: thread.c:1600
static ABTU_ret_err int ABTU_malloc(size_t size, void **p_ptr)
Definition: abtu.h:142
static uint32_t ABTD_atomic_fetch_add_uint32(ABTD_atomic_uint32 *ptr, uint32_t v)
Definition: abtd_atomic.h:391
#define ABT_FALSE
Definition: abt.h:285
ABTI_ythread_queue * h_list
ABTI_ythread * ABTI_ythread_htable_pop(ABTI_ythread_htable *p_htable, ABTI_ythread_queue *p_queue)
static ABTU_ret_err int ABTU_memalign(size_t alignment, size_t size, void **p_ptr)
Definition: abtu.h:123
ABTI_ythread * low_tail
ABTD_atomic_int state
Definition: abti.h:322
#define ABTI_tool_event_ythread_resume(p_local, p_ythread, p_caller)
Definition: abti_tool.h:383
ABTI_ythread_queue * l_list
ABTI_thread thread
Definition: abti.h:348
#define ABT_SUCCESS
Definition: abt.h:64
#define ABT_TRUE
Definition: abt.h:284
static void ABTD_atomic_relaxed_store_uint32(ABTD_atomic_uint32 *ptr, uint32_t val)
Definition: abtd_atomic.h:884
ABTI_ythread_queue * queue
ABT_bool ABTI_ythread_htable_switch_low(ABTI_xstream **pp_local_xstream, ABTI_ythread_queue *p_queue, ABTI_ythread *p_ythread, ABTI_ythread_htable *p_htable, ABT_sync_event_type sync_event_type, void *p_sync)
static uint32_t ABTD_atomic_fetch_sub_uint32(ABTD_atomic_uint32 *ptr, uint32_t v)
Definition: abtd_atomic.h:440
static uint32_t ABTD_atomic_relaxed_load_uint32(const ABTD_atomic_uint32 *ptr)
Definition: abtd_atomic.h:689
#define ABTI_ASSERT(cond)
Definition: abti_error.h:12
#define LOG_DEBUG(fmt,...)
Definition: abti_log.h:26
#define ABTI_CHECK_ERROR(abt_errno)
Definition: abti_error.h:127
void ABTI_ythread_htable_push_low(ABTI_ythread_htable *p_htable, int idx, ABTI_ythread *p_ythread)
static void ABTI_ythread_queue_release_low_mutex(ABTI_ythread_queue *p_queue)
void ABTI_ythread_htable_free(ABTI_ythread_htable *p_htable)
ABTI_thread * p_next
Definition: abti.h:314
static ABTI_ythread * ABTI_thread_get_ythread(ABTI_thread *p_thread)
Definition: abti_thread.h:52
ABTI_ythread * ABTI_ythread_htable_pop_low(ABTI_ythread_htable *p_htable, ABTI_ythread_queue *p_queue)
#define ABTI_tool_event_thread_run(p_local_xstream, p_thread, p_prev,p_parent)
Definition: abti_tool.h:339
static ABTI_local * ABTI_xstream_get_local(ABTI_xstream *p_xstream)
Definition: abti_stream.h:67
ABT_sync_event_type
Definition: abt.h:244
#define ABTI_IS_ERROR_CHECK_ENABLED
Definition: abti.h:20
static void ABTU_free(void *ptr)
Definition: abtu.h:135
static ABTI_ythread * ABTI_ythread_context_switch_to_sibling(ABTI_xstream **pp_local_xstream, ABTI_ythread *p_old, ABTI_ythread *p_new)
Definition: abti_ythread.h:304
#define ABTI_STATIC_ASSERT(cond)
Definition: abti_error.h:19
ABTI_thread * p_parent
Definition: abti.h:319
#define ABTU_ret_err
Definition: abtu.h:49