18#ifndef __SCHED_POLICY_H__
19#define __SCHED_POLICY_H__
25#include <core/workers.h>
27#include <starpu_scheduler.h>
31#pragma GCC visibility push(hidden)
33#define _STARPU_SCHED_BEGIN \
34 _STARPU_TRACE_WORKER_SCHEDULING_PUSH; \
35 _SIMGRID_TIMER_BEGIN(_starpu_simgrid_sched_cost())
36#define _STARPU_SCHED_END \
38 _STARPU_TRACE_WORKER_SCHEDULING_POP
40void _starpu_sched_init(
void);
42struct starpu_machine_config;
43struct starpu_sched_policy *_starpu_get_sched_policy(
struct _starpu_sched_ctx *sched_ctx);
50struct starpu_sched_policy *_starpu_select_sched_policy(
struct _starpu_machine_config *config,
const char *required_policy);
52void _starpu_sched_task_submit(
struct starpu_task *task);
53void _starpu_sched_do_schedule(
unsigned sched_ctx_id);
63void _starpu_sched_post_exec_hook(
struct starpu_task *task);
64int _starpu_pop_task_end(
struct starpu_task *task);
66struct starpu_task *_starpu_create_conversion_task(starpu_data_handle_t handle,
67 unsigned int node) STARPU_ATTRIBUTE_MALLOC;
69struct starpu_task *_starpu_create_conversion_task_for_arch(starpu_data_handle_t handle,
72void _starpu_sched_pre_exec_hook(
struct starpu_task *task);
74void _starpu_print_idle_time();
78extern struct starpu_sched_policy _starpu_sched_lws_policy;
79extern struct starpu_sched_policy _starpu_sched_ws_policy;
80extern struct starpu_sched_policy _starpu_sched_prio_policy;
81extern struct starpu_sched_policy _starpu_sched_random_policy;
82extern struct starpu_sched_policy _starpu_sched_dm_policy;
83extern struct starpu_sched_policy _starpu_sched_dmda_policy STARPU_ATTRIBUTE_VISIBILITY_DEFAULT;
84extern struct starpu_sched_policy _starpu_sched_dmda_prio_policy;
85extern struct starpu_sched_policy _starpu_sched_dmda_ready_policy;
86extern struct starpu_sched_policy _starpu_sched_dmda_sorted_policy;
87extern struct starpu_sched_policy _starpu_sched_dmda_sorted_decision_policy;
88extern struct starpu_sched_policy _starpu_sched_eager_policy;
89extern struct starpu_sched_policy _starpu_sched_parallel_heft_policy STARPU_ATTRIBUTE_VISIBILITY_DEFAULT;
90extern struct starpu_sched_policy _starpu_sched_peager_policy;
91extern struct starpu_sched_policy _starpu_sched_heteroprio_policy;
92extern struct starpu_sched_policy _starpu_sched_modular_eager_policy;
93extern struct starpu_sched_policy _starpu_sched_modular_eager_prefetching_policy;
94extern struct starpu_sched_policy _starpu_sched_modular_eager_prio_policy;
95extern struct starpu_sched_policy _starpu_sched_modular_gemm_policy;
96extern struct starpu_sched_policy _starpu_sched_modular_prio_policy;
97extern struct starpu_sched_policy _starpu_sched_modular_prio_prefetching_policy;
98extern struct starpu_sched_policy _starpu_sched_modular_random_policy;
99extern struct starpu_sched_policy _starpu_sched_modular_random_prio_policy;
100extern struct starpu_sched_policy _starpu_sched_modular_random_prefetching_policy;
101extern struct starpu_sched_policy _starpu_sched_modular_random_prio_prefetching_policy;
102extern struct starpu_sched_policy _starpu_sched_modular_parallel_random_policy;
103extern struct starpu_sched_policy _starpu_sched_modular_parallel_random_prio_policy;
104extern struct starpu_sched_policy _starpu_sched_modular_ws_policy;
105extern struct starpu_sched_policy _starpu_sched_modular_dmda_policy;
106extern struct starpu_sched_policy _starpu_sched_modular_dmdap_policy;
107extern struct starpu_sched_policy _starpu_sched_modular_dmdar_policy;
108extern struct starpu_sched_policy _starpu_sched_modular_dmdas_policy;
109extern struct starpu_sched_policy _starpu_sched_modular_heft_policy;
110extern struct starpu_sched_policy _starpu_sched_modular_heft_prio_policy;
111extern struct starpu_sched_policy _starpu_sched_modular_heft2_policy;
112extern struct starpu_sched_policy _starpu_sched_modular_heteroprio_policy;
113extern struct starpu_sched_policy _starpu_sched_modular_heteroprio_heft_policy;
114extern struct starpu_sched_policy _starpu_sched_modular_parallel_heft_policy;
115extern struct starpu_sched_policy _starpu_sched_graph_test_policy;
116extern struct starpu_sched_policy _starpu_sched_tree_heft_hierarchical_policy;
118extern long _starpu_task_break_on_push;
119extern long _starpu_task_break_on_sched;
120extern long _starpu_task_break_on_pop;
121extern long _starpu_task_break_on_exec;
124#define _STARPU_TASK_BREAK_ON(task, what) do { \
125 if (_starpu_get_job_associated_to_task(task)->job_id == (unsigned long) _starpu_task_break_on_##what) \
129#define _STARPU_TASK_BREAK_ON(task, what) ((void) 0)
132#pragma GCC visibility pop
starpu_node_kind
Definition starpu_worker.h:44
int _starpu_push_task_to_workers(struct starpu_task *task)
struct starpu_task * _starpu_pop_task(struct _starpu_worker *worker)
Definition sched_ctx.h:48