19#ifndef __STARPU_SCHED_CTX_H__
20#define __STARPU_SCHED_CTX_H__
50#define STARPU_SCHED_CTX_POLICY_NAME (1 << 16)
56#define STARPU_SCHED_CTX_POLICY_STRUCT (2 << 16)
62#define STARPU_SCHED_CTX_POLICY_MIN_PRIO (3 << 16)
68#define STARPU_SCHED_CTX_POLICY_MAX_PRIO (4 << 16)
70#define STARPU_SCHED_CTX_HIERARCHY_LEVEL (5 << 16)
71#define STARPU_SCHED_CTX_NESTED (6 << 16)
76#define STARPU_SCHED_CTX_AWAKE_WORKERS (7 << 16)
82#define STARPU_SCHED_CTX_POLICY_INIT (8 << 16)
88#define STARPU_SCHED_CTX_USER_DATA (9 << 16)
95#define STARPU_SCHED_CTX_CUDA_NSMS (10 << 16)
101#define STARPU_SCHED_CTX_SUB_CTXS (11 << 16)
189unsigned starpu_sched_ctx_get_inheritor(
unsigned sched_ctx_id);
191unsigned starpu_sched_ctx_get_hierarchy_level(
unsigned sched_ctx_id);
256unsigned starpu_sched_ctx_contains_type_of_worker(
enum starpu_worker_archtype arch,
unsigned sched_ctx_id);
265unsigned starpu_sched_ctx_get_ctx_for_task(
struct starpu_task *task);
267unsigned starpu_worker_get_sched_ctx_id_stream(
unsigned stream_workerid);
279void starpu_sched_ctx_set_user_data(
unsigned sched_ctx_id,
void *user_data);
304int starpu_sched_ctx_get_nready_tasks(
unsigned sched_ctx_id);
306double starpu_sched_ctx_get_nready_flops(
unsigned sched_ctx_id);
308void starpu_sched_ctx_list_task_counters_increment(
unsigned sched_ctx_id,
int workerid);
310void starpu_sched_ctx_list_task_counters_decrement(
unsigned sched_ctx_id,
int workerid);
312void starpu_sched_ctx_list_task_counters_reset(
unsigned sched_ctx_id,
int workerid);
314void starpu_sched_ctx_list_task_counters_increment_all_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx_id);
316void starpu_sched_ctx_list_task_counters_decrement_all_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx_id);
318void starpu_sched_ctx_list_task_counters_reset_all(
struct starpu_task *task,
unsigned sched_ctx_id);
320void starpu_sched_ctx_set_priority(
int *workers,
int nworkers,
unsigned sched_ctx_id,
unsigned priority);
322unsigned starpu_sched_ctx_get_priority(
int worker,
unsigned sched_ctx_id);
324void starpu_sched_ctx_get_available_cpuids(
unsigned sched_ctx_id,
int **cpuids,
int *ncpuids);
326void starpu_sched_ctx_bind_current_thread_to_cpuid(
unsigned cpuid);
328int starpu_sched_ctx_book_workers_for_task(
unsigned sched_ctx_id,
int *workerids,
int nworkers);
330void starpu_sched_ctx_unbook_workers_for_task(
unsigned sched_ctx_id,
int master);
344void starpu_sched_ctx_revert_task_counters_ctx_locked(
unsigned sched_ctx_id,
double flops);
346void starpu_sched_ctx_move_task_to_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx,
unsigned with_repush);
348int starpu_sched_ctx_get_worker_rank(
unsigned sched_ctx_id);
357unsigned starpu_sched_ctx_has_starpu_scheduler(
unsigned sched_ctx_id,
unsigned *awake_workers);
359int starpu_sched_ctx_get_stream_worker(
unsigned sub_ctx);
360int starpu_sched_ctx_get_nsms(
unsigned sched_ctx);
361void starpu_sched_ctx_get_sms_interval(
int stream_workerid,
int *start,
int *end);
403int starpu_sched_ctx_min_priority_is_set(
unsigned sched_ctx_id);
405int starpu_sched_ctx_max_priority_is_set(
unsigned sched_ctx_id);
410#define STARPU_MIN_PRIO (starpu_sched_get_min_priority())
415#define STARPU_MAX_PRIO (starpu_sched_get_max_priority())
421#define STARPU_DEFAULT_PRIO 0
Definition starpu_task.h:683
void starpu_sched_ctx_display_workers(unsigned sched_ctx_id, FILE *f)
void(*)(unsigned) starpu_sched_ctx_get_sched_policy_callback(unsigned sched_ctx_id)
Definition starpu_sched_ctx.h:355
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:82
enum starpu_worker_collection_type type
Definition starpu_worker.h:133
starpu_worker_archtype
Definition starpu_worker.h:66
starpu_worker_collection_type
Definition starpu_worker.h:100
Definition starpu_worker.h:113