StarPU Handbook
Loading...
Searching...
No Matches
starpu_sched_ctx.h
Go to the documentation of this file.
1/* StarPU --- Runtime system for heterogeneous multicore architectures.
2 *
3 * Copyright (C) 2010-2021 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
4 * Copyright (C) 2016 Uppsala University
5 * Copyright (C) 2017 Arthur Chevalier
6 *
7 * StarPU is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU Lesser General Public License as published by
9 * the Free Software Foundation; either version 2.1 of the License, or (at
10 * your option) any later version.
11 *
12 * StarPU is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
15 *
16 * See the GNU Lesser General Public License in COPYING.LGPL for more details.
17 */
18
19#ifndef __STARPU_SCHED_CTX_H__
20#define __STARPU_SCHED_CTX_H__
21
22#include <starpu.h>
23
24#ifdef __cplusplus
25extern "C"
26{
27#endif
28
51#define STARPU_SCHED_CTX_POLICY_NAME (1<<16)
52
57#define STARPU_SCHED_CTX_POLICY_STRUCT (2<<16)
58
63#define STARPU_SCHED_CTX_POLICY_MIN_PRIO (3<<16)
64
69#define STARPU_SCHED_CTX_POLICY_MAX_PRIO (4<<16)
70
71#define STARPU_SCHED_CTX_HIERARCHY_LEVEL (5<<16)
72#define STARPU_SCHED_CTX_NESTED (6<<16)
73
77#define STARPU_SCHED_CTX_AWAKE_WORKERS (7<<16)
78
83#define STARPU_SCHED_CTX_POLICY_INIT (8<<16)
84
89#define STARPU_SCHED_CTX_USER_DATA (9<<16)
90
96#define STARPU_SCHED_CTX_CUDA_NSMS (10<<16)
97
102#define STARPU_SCHED_CTX_SUB_CTXS (11<<16)
103
138unsigned starpu_sched_ctx_create(int *workerids_ctx, int nworkers_ctx, const char *sched_ctx_name, ...);
139
143unsigned starpu_sched_ctx_create_inside_interval(const char *policy_name, const char *sched_ctx_name, int min_ncpus, int max_ncpus, int min_ngpus, int max_ngpus, unsigned allow_overlap);
144
150void starpu_sched_ctx_register_close_callback(unsigned sched_ctx_id, void (*close_callback)(unsigned sched_ctx_id, void* args), void *args);
151
157void starpu_sched_ctx_add_workers(int *workerids_ctx, unsigned nworkers_ctx, unsigned sched_ctx_id);
158
164void starpu_sched_ctx_remove_workers(int *workerids_ctx, unsigned nworkers_ctx, unsigned sched_ctx_id);
165
170void starpu_sched_ctx_display_workers(unsigned sched_ctx_id, FILE *f);
171
176void starpu_sched_ctx_delete(unsigned sched_ctx_id);
177
183void starpu_sched_ctx_set_inheritor(unsigned sched_ctx_id, unsigned inheritor);
184
185unsigned starpu_sched_ctx_get_inheritor(unsigned sched_ctx_id);
186
187unsigned starpu_sched_ctx_get_hierarchy_level(unsigned sched_ctx_id);
188
193void starpu_sched_ctx_set_context(unsigned *sched_ctx_id);
194
201
207
213void starpu_sched_ctx_finished_submit(unsigned sched_ctx_id);
214
221unsigned starpu_sched_ctx_get_workers_list(unsigned sched_ctx_id, int **workerids);
222
230unsigned starpu_sched_ctx_get_workers_list_raw(unsigned sched_ctx_id, int **workerids);
231
237unsigned starpu_sched_ctx_get_nworkers(unsigned sched_ctx_id);
238
242unsigned starpu_sched_ctx_get_nshared_workers(unsigned sched_ctx_id, unsigned sched_ctx_id2);
243
247unsigned starpu_sched_ctx_contains_worker(int workerid, unsigned sched_ctx_id);
248
249unsigned starpu_sched_ctx_contains_type_of_worker(enum starpu_worker_archtype arch, unsigned sched_ctx_id);
250
256unsigned starpu_sched_ctx_worker_get_id(unsigned sched_ctx_id);
257
258unsigned starpu_sched_ctx_get_ctx_for_task(struct starpu_task *task);
259
264
268void *starpu_sched_ctx_get_user_data(unsigned sched_ctx_id);
269
270void starpu_sched_ctx_set_user_data(unsigned sched_ctx_id, void* user_data);
271
277void starpu_sched_ctx_set_policy_data(unsigned sched_ctx_id, void *policy_data);
278
283void *starpu_sched_ctx_get_policy_data(unsigned sched_ctx_id);
284
285struct starpu_sched_policy *starpu_sched_ctx_get_sched_policy(unsigned sched_ctx_id);
286
291void *starpu_sched_ctx_exec_parallel_code(void* (*func)(void*), void *param, unsigned sched_ctx_id);
292
293int starpu_sched_ctx_get_nready_tasks(unsigned sched_ctx_id);
294
295double starpu_sched_ctx_get_nready_flops(unsigned sched_ctx_id);
296
297void starpu_sched_ctx_list_task_counters_increment(unsigned sched_ctx_id, int workerid);
298
299void starpu_sched_ctx_list_task_counters_decrement(unsigned sched_ctx_id, int workerid);
300
301void starpu_sched_ctx_list_task_counters_reset(unsigned sched_ctx_id, int workerid);
302
303void starpu_sched_ctx_list_task_counters_increment_all_ctx_locked(struct starpu_task *task, unsigned sched_ctx_id);
304
305void starpu_sched_ctx_list_task_counters_decrement_all_ctx_locked(struct starpu_task *task, unsigned sched_ctx_id);
306
307void starpu_sched_ctx_list_task_counters_reset_all(struct starpu_task *task, unsigned sched_ctx_id);
308
309void starpu_sched_ctx_set_priority(int *workers, int nworkers, unsigned sched_ctx_id, unsigned priority);
310
311unsigned starpu_sched_ctx_get_priority(int worker, unsigned sched_ctx_id);
312
313void starpu_sched_ctx_get_available_cpuids(unsigned sched_ctx_id, int **cpuids, int *ncpuids);
314
315void starpu_sched_ctx_bind_current_thread_to_cpuid(unsigned cpuid);
316
317int starpu_sched_ctx_book_workers_for_task(unsigned sched_ctx_id, int *workerids, int nworkers);
318
319void starpu_sched_ctx_unbook_workers_for_task(unsigned sched_ctx_id, int master);
320
325unsigned starpu_sched_ctx_worker_is_master_for_child_ctx(int workerid, unsigned sched_ctx_id);
326
332
333void starpu_sched_ctx_revert_task_counters_ctx_locked(unsigned sched_ctx_id, double flops);
334
335void starpu_sched_ctx_move_task_to_ctx_locked(struct starpu_task *task, unsigned sched_ctx, unsigned with_repush);
336
337int starpu_sched_ctx_get_worker_rank(unsigned sched_ctx_id);
338
339void (*starpu_sched_ctx_get_sched_policy_init(unsigned sched_ctx_id))(unsigned);
340
341unsigned starpu_sched_ctx_has_starpu_scheduler(unsigned sched_ctx_id, unsigned *awake_workers);
342
343int starpu_sched_ctx_get_stream_worker(unsigned sub_ctx);
344int starpu_sched_ctx_get_nsms(unsigned sched_ctx);
345void starpu_sched_ctx_get_sms_interval(int stream_workerid, int *start, int *end);
346
358int starpu_sched_ctx_get_min_priority(unsigned sched_ctx_id);
359
364int starpu_sched_ctx_get_max_priority(unsigned sched_ctx_id);
365
375int starpu_sched_ctx_set_min_priority(unsigned sched_ctx_id, int min_prio);
376
385int starpu_sched_ctx_set_max_priority(unsigned sched_ctx_id, int max_prio);
386
387int starpu_sched_ctx_min_priority_is_set(unsigned sched_ctx_id);
388
389int starpu_sched_ctx_max_priority_is_set(unsigned sched_ctx_id);
390
394#define STARPU_MIN_PRIO (starpu_sched_get_min_priority())
395
399#define STARPU_MAX_PRIO (starpu_sched_get_max_priority())
400
405#define STARPU_DEFAULT_PRIO 0
406
419
424
429
434#ifdef __cplusplus
435}
436#endif
437
438#endif /* __STARPU_SCHED_CTX_H__ */
Definition: starpu_task.h:567
void starpu_sched_ctx_display_workers(unsigned sched_ctx_id, FILE *f)
unsigned starpu_sched_ctx_contains_worker(int workerid, unsigned sched_ctx_id)
void starpu_sched_ctx_delete(unsigned sched_ctx_id)
void starpu_sched_ctx_finished_submit(unsigned sched_ctx_id)
unsigned starpu_sched_ctx_get_workers_list(unsigned sched_ctx_id, int **workerids)
unsigned starpu_sched_ctx_get_nworkers(unsigned sched_ctx_id)
int starpu_sched_ctx_set_max_priority(unsigned sched_ctx_id, int max_prio)
int starpu_sched_ctx_get_max_priority(unsigned sched_ctx_id)
unsigned starpu_sched_ctx_create_inside_interval(const char *policy_name, const char *sched_ctx_name, int min_ncpus, int max_ncpus, int min_ngpus, int max_ngpus, unsigned allow_overlap)
void starpu_sched_ctx_set_policy_data(unsigned sched_ctx_id, void *policy_data)
unsigned starpu_sched_ctx_worker_is_master_for_child_ctx(int workerid, unsigned sched_ctx_id)
void starpu_sched_ctx_remove_workers(int *workerids_ctx, unsigned nworkers_ctx, unsigned sched_ctx_id)
unsigned starpu_sched_ctx_master_get_context(int masterid)
void starpu_sched_ctx_set_context(unsigned *sched_ctx_id)
unsigned starpu_sched_ctx_create(int *workerids_ctx, int nworkers_ctx, const char *sched_ctx_name,...)
void starpu_sched_ctx_delete_worker_collection(unsigned sched_ctx_id)
unsigned starpu_sched_ctx_overlapping_ctxs_on_worker(int workerid)
int starpu_sched_ctx_get_min_priority(unsigned sched_ctx_id)
int starpu_sched_ctx_set_min_priority(unsigned sched_ctx_id, int min_prio)
struct starpu_worker_collection * starpu_sched_ctx_get_worker_collection(unsigned sched_ctx_id)
unsigned starpu_sched_ctx_worker_get_id(unsigned sched_ctx_id)
void * starpu_sched_ctx_get_policy_data(unsigned sched_ctx_id)
void starpu_sched_ctx_register_close_callback(unsigned sched_ctx_id, void(*close_callback)(unsigned sched_ctx_id, void *args), void *args)
void starpu_sched_ctx_add_workers(int *workerids_ctx, unsigned nworkers_ctx, unsigned sched_ctx_id)
void * starpu_sched_ctx_get_user_data(unsigned sched_ctx_id)
void starpu_sched_ctx_set_inheritor(unsigned sched_ctx_id, unsigned inheritor)
unsigned starpu_sched_ctx_get_nshared_workers(unsigned sched_ctx_id, unsigned sched_ctx_id2)
unsigned starpu_sched_ctx_get_workers_list_raw(unsigned sched_ctx_id, int **workerids)
struct starpu_worker_collection * starpu_sched_ctx_create_worker_collection(unsigned sched_ctx_id, enum starpu_worker_collection_type type) STARPU_ATTRIBUTE_MALLOC
unsigned starpu_sched_ctx_get_context(void)
void * starpu_sched_ctx_exec_parallel_code(void *(*func)(void *), void *param, unsigned sched_ctx_id)
void starpu_sched_ctx_stop_task_submission(void)
Definition: starpu_scheduler.h:83
#define STARPU_ATTRIBUTE_MALLOC
Definition: starpu_util.h:112
enum starpu_worker_collection_type type
Definition: starpu_worker.h:125
starpu_worker_archtype
Definition: starpu_worker.h:63
starpu_worker_collection_type
Definition: starpu_worker.h:92
Definition: starpu_worker.h:105