ARGOBOTS  1.1
prio.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 
8 /* Priority Scheduler Implementation */
9 
10 static int sched_init(ABT_sched sched, ABT_sched_config config);
11 static void sched_run(ABT_sched sched);
12 static int sched_free(ABT_sched);
13 
15  .init = sched_init,
16  .run = sched_run,
17  .free = sched_free,
18  .get_migr_pool = NULL };
19 
20 typedef struct {
21  uint32_t event_freq;
22  int num_pools;
23  ABT_pool *pools;
24 #ifdef ABT_CONFIG_USE_SCHED_SLEEP
25  struct timespec sleep_time;
26 #endif
27 } sched_data;
28 
30 {
31  return &sched_prio_def;
32 }
33 
34 static inline sched_data *sched_data_get_ptr(void *data)
35 {
36  return (sched_data *)data;
37 }
38 
39 static int sched_init(ABT_sched sched, ABT_sched_config config)
40 {
41  int abt_errno;
42  int num_pools;
43  ABTI_global *p_global = ABTI_global_get_global();
44 
45  ABTI_sched *p_sched = ABTI_sched_get_ptr(sched);
47  ABTI_sched_config *p_config = ABTI_sched_config_get_ptr(config);
48 
49  /* Default settings */
50  sched_data *p_data;
51  abt_errno = ABTU_malloc(sizeof(sched_data), (void **)&p_data);
52  ABTI_CHECK_ERROR(abt_errno);
53 #ifdef ABT_CONFIG_USE_SCHED_SLEEP
54  p_data->sleep_time.tv_sec = 0;
55  p_data->sleep_time.tv_nsec = p_global->sched_sleep_nsec;
56 #endif
57 
58  /* Set the default value by default. */
59  p_data->event_freq = p_global->sched_event_freq;
60  if (p_config) {
61  int event_freq;
62  /* Set the variables from config */
63  abt_errno = ABTI_sched_config_read(p_config, ABT_sched_basic_freq.idx,
64  &event_freq);
65  if (abt_errno == ABT_SUCCESS) {
66  p_data->event_freq = event_freq;
67  }
68  }
69 
70  /* Save the list of pools */
71  num_pools = p_sched->num_pools;
72  p_data->num_pools = num_pools;
73  abt_errno =
74  ABTU_malloc(num_pools * sizeof(ABT_pool), (void **)&p_data->pools);
75  if (ABTI_IS_ERROR_CHECK_ENABLED && abt_errno != ABT_SUCCESS) {
76  ABTU_free(p_data);
77  ABTI_CHECK_ERROR(abt_errno);
78  }
79  memcpy(p_data->pools, p_sched->pools, sizeof(ABT_pool) * num_pools);
80 
81  p_sched->data = p_data;
82  return ABT_SUCCESS;
83 }
84 
85 static void sched_run(ABT_sched sched)
86 {
87  ABTI_global *p_global = ABTI_global_get_global();
88  ABTI_xstream *p_local_xstream =
90  uint32_t work_count = 0;
91  sched_data *p_data;
92  uint32_t event_freq;
93  int num_pools;
94  ABT_pool *pools;
95  int i;
96  CNT_DECL(run_cnt);
97 
98  ABTI_sched *p_sched = ABTI_sched_get_ptr(sched);
99  ABTI_ASSERT(p_sched);
100 
101  p_data = sched_data_get_ptr(p_sched->data);
102  event_freq = p_data->event_freq;
103  num_pools = p_sched->num_pools;
104  pools = p_data->pools;
105 
106  while (1) {
107  CNT_INIT(run_cnt, 0);
108 
109  /* Execute one work unit from the scheduler's pool */
110  /* The pool with lower index has higher priority. */
111  for (i = 0; i < num_pools; i++) {
112  ABT_pool pool = pools[i];
113  ABTI_pool *p_pool = ABTI_pool_get_ptr(pool);
114  ABT_unit unit = ABTI_pool_pop(p_pool);
115  if (unit != ABT_UNIT_NULL) {
116  ABTI_thread *p_thread = ABTI_unit_get_thread(p_global, unit);
117  ABTI_xstream_run_thread(p_global, &p_local_xstream, p_thread);
118  CNT_INC(run_cnt);
119  break;
120  }
121  }
122 
123  if (++work_count >= event_freq) {
124  ABTI_xstream_check_events(p_local_xstream, p_sched);
125  ABTI_local *p_local = ABTI_xstream_get_local(p_local_xstream);
126  if (ABTI_sched_has_to_stop(&p_local, p_sched) == ABT_TRUE)
127  break;
128  p_local_xstream = ABTI_local_get_xstream(p_local);
129  work_count = 0;
130  SCHED_SLEEP(run_cnt, p_data->sleep_time);
131  }
132  }
133 }
134 
135 static int sched_free(ABT_sched sched)
136 {
137  ABTI_sched *p_sched = ABTI_sched_get_ptr(sched);
138  ABTI_ASSERT(p_sched);
139 
140  sched_data *p_data = sched_data_get_ptr(p_sched->data);
141  ABTU_free(p_data->pools);
142  ABTU_free(p_data);
143  return ABT_SUCCESS;
144 }
ABTI_CHECK_NULL_SCHED_PTR
#define ABTI_CHECK_NULL_SCHED_PTR(p)
Definition: abti_error.h:177
ABTI_sched::data
void * data
Definition: abti.h:301
ABTI_sched_get_ptr
static ABTI_sched * ABTI_sched_get_ptr(ABT_sched sched)
Definition: abti_sched.h:11
ABTI_global::sched_sleep_nsec
uint64_t sched_sleep_nsec
Definition: abti.h:212
sched_data::event_freq
uint32_t event_freq
Definition: basic.c:22
sched_run
static void sched_run(ABT_sched sched)
Definition: prio.c:85
ABTI_sched_get_prio_def
ABT_sched_def * ABTI_sched_get_prio_def(void)
Definition: prio.c:29
ABT_sched_def::type
ABT_sched_type type
Unused value.
Definition: abt.h:1309
ABTI_global_get_global
static ABTI_global * ABTI_global_get_global(void)
Definition: abti_global.h:9
ABTI_CHECK_ERROR
#define ABTI_CHECK_ERROR(abt_errno)
Definition: abti_error.h:120
data
Definition: fifo.c:29
ABTI_sched::num_pools
size_t num_pools
Definition: abti.h:299
sched_data
Definition: basic.c:21
ABT_sched_config
struct ABT_sched_config_opaque * ABT_sched_config
Scheduler configuration handle type.
Definition: abt.h:815
sched_init
static int sched_init(ABT_sched sched, ABT_sched_config config)
Definition: prio.c:39
ABTI_thread
Definition: abti.h:371
ABTI_IS_ERROR_CHECK_ENABLED
#define ABTI_IS_ERROR_CHECK_ENABLED
Definition: abti.h:20
ABTI_xstream
Definition: abti.h:264
ABTI_sched_config
Definition: abti.h:323
ABT_pool
struct ABT_pool_opaque * ABT_pool
Pool handle type.
Definition: abt.h:841
ABTI_sched_has_to_stop
ABT_bool ABTI_sched_has_to_stop(ABTI_local **pp_local, ABTI_sched *p_sched)
Definition: sched.c:871
ABT_sched
struct ABT_sched_opaque * ABT_sched
Scheduler handle type.
Definition: abt.h:808
ABT_SCHED_TYPE_ULT
@ ABT_SCHED_TYPE_ULT
Definition: abt.h:492
ABTI_pool
Definition: abti.h:327
abti.h
ABTU_malloc
static ABTU_ret_err int ABTU_malloc(size_t size, void **p_ptr)
Definition: abtu.h:262
ABTI_global::sched_event_freq
uint32_t sched_event_freq
Definition: abti.h:211
sched_data::pools
ABT_pool * pools
Definition: basic.c:24
ABTI_sched_config_read
ABTU_ret_err int ABTI_sched_config_read(const ABTI_sched_config *p_config, int idx, void *p_val)
Definition: config.c:259
ABT_unit
struct ABT_unit_opaque * ABT_unit
Work unit handle type for scheduling.
Definition: abt.h:869
ABTI_ASSERT
#define ABTI_ASSERT(cond)
Definition: abti_error.h:12
ABTI_sched::pools
ABT_pool * pools
Definition: abti.h:298
ABTI_local_get_local
static ABTI_local * ABTI_local_get_local(void)
Definition: abti_local.h:41
ABT_SUCCESS
#define ABT_SUCCESS
Error code: the routine returns successfully.
Definition: abt.h:92
ABTI_xstream_check_events
void ABTI_xstream_check_events(ABTI_xstream *p_xstream, ABTI_sched *p_sched)
Definition: stream.c:1568
ABTI_xstream_get_local
static ABTI_local * ABTI_xstream_get_local(ABTI_xstream *p_xstream)
Definition: abti_stream.h:68
ABT_TRUE
#define ABT_TRUE
True constant for ABT_bool.
Definition: abt.h:748
ABTI_pool_get_ptr
static ABTI_pool * ABTI_pool_get_ptr(ABT_pool pool)
Definition: abti_pool.h:11
ABTI_pool_pop
static ABT_unit ABTI_pool_pop(ABTI_pool *p_pool)
Definition: abti_pool.h:96
ABTI_sched
Definition: abti.h:289
sched_data_get_ptr
static sched_data * sched_data_get_ptr(void *data)
Definition: prio.c:34
CNT_INIT
#define CNT_INIT(c, v)
Definition: abti_sched.h:79
ABTU_free
static void ABTU_free(void *ptr)
Definition: abtu.h:217
ABT_sched_def
A struct that defines a scheduler.
Definition: abt.h:1302
ABTI_xstream_run_thread
void ABTI_xstream_run_thread(ABTI_global *p_global, ABTI_xstream **pp_local_xstream, ABTI_thread *p_thread)
Definition: stream.c:1554
ABTI_unit_get_thread
static ABTI_thread * ABTI_unit_get_thread(ABTI_global *p_global, ABT_unit unit)
Definition: abti_unit.h:43
CNT_DECL
#define CNT_DECL(c)
Definition: abti_sched.h:78
ABTI_local
struct ABTI_local ABTI_local
Definition: abti.h:110
ABTI_sched_config_get_ptr
static ABTI_sched_config * ABTI_sched_config_get_ptr(ABT_sched_config config)
Definition: abti_config.h:12
ABTI_global
Definition: abti.h:196
SCHED_SLEEP
#define SCHED_SLEEP(c, t)
Definition: abti_sched.h:81
sched_prio_def
static ABT_sched_def sched_prio_def
Definition: prio.c:14
ABT_sched_basic_freq
ABT_sched_config_var ABT_sched_basic_freq
Predefined ABT_sched_config_var to configure the frequency for checking events of the basic scheduler...
Definition: config.c:34
sched_free
static int sched_free(ABT_sched)
Definition: prio.c:135
ABT_UNIT_NULL
#define ABT_UNIT_NULL
Definition: abt.h:1061
CNT_INC
#define CNT_INC(c)
Definition: abti_sched.h:80
ABTI_local_get_xstream
static ABTI_xstream * ABTI_local_get_xstream(ABTI_local *p_local)
Definition: abti_local.h:86
ABT_sched_config_var::idx
int idx
Definition: abt.h:1242