19 #ifndef __STARPU_SCHED_CTX_H__
20 #define __STARPU_SCHED_CTX_H__
51 #define STARPU_SCHED_CTX_POLICY_NAME (1<<16)
57 #define STARPU_SCHED_CTX_POLICY_STRUCT (2<<16)
63 #define STARPU_SCHED_CTX_POLICY_MIN_PRIO (3<<16)
69 #define STARPU_SCHED_CTX_POLICY_MAX_PRIO (4<<16)
71 #define STARPU_SCHED_CTX_HIERARCHY_LEVEL (5<<16)
72 #define STARPU_SCHED_CTX_NESTED (6<<16)
77 #define STARPU_SCHED_CTX_AWAKE_WORKERS (7<<16)
83 #define STARPU_SCHED_CTX_POLICY_INIT (8<<16)
89 #define STARPU_SCHED_CTX_USER_DATA (9<<16)
96 #define STARPU_SCHED_CTX_CUDA_NSMS (10<<16)
102 #define STARPU_SCHED_CTX_SUB_CTXS (11<<16)
185 unsigned starpu_sched_ctx_get_inheritor(
unsigned sched_ctx_id);
187 unsigned starpu_sched_ctx_get_hierarchy_level(
unsigned sched_ctx_id);
249 unsigned starpu_sched_ctx_contains_type_of_worker(
enum starpu_worker_archtype arch,
unsigned sched_ctx_id);
258 unsigned starpu_sched_ctx_get_ctx_for_task(
struct starpu_task *task);
270 void starpu_sched_ctx_set_user_data(
unsigned sched_ctx_id,
void* user_data);
293 int starpu_sched_ctx_get_nready_tasks(
unsigned sched_ctx_id);
295 double starpu_sched_ctx_get_nready_flops(
unsigned sched_ctx_id);
297 void starpu_sched_ctx_list_task_counters_increment(
unsigned sched_ctx_id,
int workerid);
299 void starpu_sched_ctx_list_task_counters_decrement(
unsigned sched_ctx_id,
int workerid);
301 void starpu_sched_ctx_list_task_counters_reset(
unsigned sched_ctx_id,
int workerid);
303 void starpu_sched_ctx_list_task_counters_increment_all_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx_id);
305 void starpu_sched_ctx_list_task_counters_decrement_all_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx_id);
307 void starpu_sched_ctx_list_task_counters_reset_all(
struct starpu_task *task,
unsigned sched_ctx_id);
309 void starpu_sched_ctx_set_priority(
int *workers,
int nworkers,
unsigned sched_ctx_id,
unsigned priority);
311 unsigned starpu_sched_ctx_get_priority(
int worker,
unsigned sched_ctx_id);
313 void starpu_sched_ctx_get_available_cpuids(
unsigned sched_ctx_id,
int **cpuids,
int *ncpuids);
315 void starpu_sched_ctx_bind_current_thread_to_cpuid(
unsigned cpuid);
317 int starpu_sched_ctx_book_workers_for_task(
unsigned sched_ctx_id,
int *workerids,
int nworkers);
319 void starpu_sched_ctx_unbook_workers_for_task(
unsigned sched_ctx_id,
int master);
333 void starpu_sched_ctx_revert_task_counters_ctx_locked(
unsigned sched_ctx_id,
double flops);
335 void starpu_sched_ctx_move_task_to_ctx_locked(
struct starpu_task *task,
unsigned sched_ctx,
unsigned with_repush);
337 int starpu_sched_ctx_get_worker_rank(
unsigned sched_ctx_id);
339 void (*starpu_sched_ctx_get_sched_policy_init(
unsigned sched_ctx_id))(unsigned);
341 unsigned starpu_sched_ctx_has_starpu_scheduler(
unsigned sched_ctx_id,
unsigned *awake_workers);
343 int starpu_sched_ctx_get_stream_worker(
unsigned sub_ctx);
344 int starpu_sched_ctx_get_nsms(
unsigned sched_ctx);
345 void starpu_sched_ctx_get_sms_interval(
int stream_workerid,
int *start,
int *end);
387 int starpu_sched_ctx_min_priority_is_set(
unsigned sched_ctx_id);
389 int starpu_sched_ctx_max_priority_is_set(
unsigned sched_ctx_id);
394 #define STARPU_MIN_PRIO (starpu_sched_get_min_priority())
399 #define STARPU_MAX_PRIO (starpu_sched_get_max_priority())
405 #define STARPU_DEFAULT_PRIO 0
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)
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_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_get_policy_data(unsigned sched_ctx_id)
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_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_exec_parallel_code(void *(*func)(void *), void *param, 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)
unsigned starpu_sched_ctx_get_context(void)
void * starpu_sched_ctx_get_user_data(unsigned sched_ctx_id)
void starpu_sched_ctx_stop_task_submission(void)
Definition: starpu_scheduler.h:83
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