21 #ifndef __STARPU_TASK_H__ 22 #define __STARPU_TASK_H__ 28 #if defined STARPU_USE_CUDA && !defined STARPU_DONT_INCLUDE_CUDA_HEADERS 50 #define STARPU_NOWHERE ((1ULL)<<0) 57 #define STARPU_CPU ((1ULL)<<1) 64 #define STARPU_CUDA ((1ULL)<<3) 71 #define STARPU_OPENCL ((1ULL)<<6) 78 #define STARPU_MIC ((1ULL)<<7) 85 #define STARPU_MPI_MS ((1ULL)<<9) 91 #define STARPU_CODELET_SIMGRID_EXECUTE (1<<0) 98 #define STARPU_CODELET_SIMGRID_EXECUTE_AND_INJECT (1<<1) 104 #define STARPU_CODELET_NOPLANS (1<<2) 110 #define STARPU_CUDA_ASYNC (1<<0) 116 #define STARPU_OPENCL_ASYNC (1<<0) 121 #define STARPU_MAIN_RAM 0 150 #define STARPU_TASK_INVALID 0 204 #define STARPU_MULTIPLE_CPU_IMPLEMENTATIONS ((starpu_cpu_func_t) -1) 213 #define STARPU_MULTIPLE_CUDA_IMPLEMENTATIONS ((starpu_cuda_func_t) -1) 222 #define STARPU_MULTIPLE_OPENCL_IMPLEMENTATIONS ((starpu_opencl_func_t) -1) 229 #define STARPU_VARIABLE_NBUFFERS (-1) 236 #define STARPU_SPECIFIC_NODE_LOCAL (-1) 237 #define STARPU_SPECIFIC_NODE_CPU (-2) 238 #define STARPU_SPECIFIC_NODE_SLOW (-3) 239 #define STARPU_SPECIFIC_NODE_FAST (-4) 726 void (*callback_func)(
void *);
754 void (*prologue_callback_func)(
void *);
767 void (*prologue_callback_pop_func)(
void *);
768 void *prologue_callback_pop_arg;
941 unsigned prefetched:1;
1123 double predicted_start;
1145 #ifdef STARPU_OPENMP 1170 #define STARPU_TASK_TYPE_NORMAL 0 1171 #define STARPU_TASK_TYPE_INTERNAL (1<<0) 1172 #define STARPU_TASK_TYPE_DATA_ACQUIRE (1<<1) 1180 #define STARPU_TASK_INITIALIZER \ 1186 .callback_func = NULL, \ 1187 .callback_arg = NULL, \ 1188 .priority = STARPU_DEFAULT_PRIO, \ 1190 .sequential_consistency = 1, \ 1192 .execute_on_a_specific_worker = 0, \ 1198 .status = STARPU_TASK_INVALID, \ 1199 .profiling_info = NULL, \ 1201 .predicted_transfer = NAN, \ 1202 .predicted_start = NAN, \ 1203 .starpu_private = NULL, \ 1207 .sched_ctx = STARPU_NMAX_SCHED_CTXS, \ 1208 .hypervisor_tag = 0, \ 1212 .dyn_handles = NULL, \ 1213 .dyn_interfaces = NULL, \ 1214 .dyn_modes = NULL, \ 1216 .possibly_parallel = 0 \ 1224 #define STARPU_TASK_GET_NBUFFERS(task) ((unsigned)((task)->cl->nbuffers == STARPU_VARIABLE_NBUFFERS ? ((task)->nbuffers) : ((task)->cl->nbuffers))) 1233 #define STARPU_TASK_GET_HANDLE(task, i) (((task)->dyn_handles) ? (task)->dyn_handles[i] : (task)->handles[i]) 1234 #define STARPU_TASK_GET_HANDLES(task) (((task)->dyn_handles) ? (task)->dyn_handles : (task)->handles) 1243 #define STARPU_TASK_SET_HANDLE(task, handle, i) \ 1244 do { if ((task)->dyn_handles) (task)->dyn_handles[i] = handle; else (task)->handles[i] = handle; } while(0) 1253 #define STARPU_CODELET_GET_MODE(codelet, i) \ 1254 (((codelet)->dyn_modes) ? (codelet)->dyn_modes[i] : (assert(i < STARPU_NMAXBUFS), (codelet)->modes[i])) 1263 #define STARPU_CODELET_SET_MODE(codelet, mode, i) \ 1264 do { if ((codelet)->dyn_modes) (codelet)->dyn_modes[i] = mode; else (codelet)->modes[i] = mode; } while(0) 1273 #define STARPU_TASK_GET_MODE(task, i) \ 1274 ((task)->cl->nbuffers == STARPU_VARIABLE_NBUFFERS || (task)->dyn_modes ? \ 1275 (((task)->dyn_modes) ? (task)->dyn_modes[i] : (task)->modes[i]) : \ 1276 STARPU_CODELET_GET_MODE((task)->cl, i) ) 1285 #define STARPU_TASK_SET_MODE(task, mode, i) \ 1287 if ((task)->cl->nbuffers == STARPU_VARIABLE_NBUFFERS || (task)->cl->nbuffers > STARPU_NMAXBUFS) \ 1288 if ((task)->dyn_modes) (task)->dyn_modes[i] = mode; else (task)->modes[i] = mode; \ 1290 STARPU_CODELET_SET_MODE((task)->cl, mode, i); \ 1300 #define STARPU_CODELET_GET_NODE(codelet, i) (((codelet)->dyn_nodes) ? (codelet)->dyn_nodes[i] : (codelet)->nodes[i]) 1309 #define STARPU_CODELET_SET_NODE(codelet, __node, i) \ 1310 do { if ((codelet)->dyn_nodes) (codelet)->dyn_nodes[i] = __node; else (codelet)->nodes[i] = __node; } while(0) 1382 int starpu_task_finished(
struct starpu_task *task) STARPU_WARN_UNUSED_RESULT;
1465 void starpu_do_schedule(
void);
void * prologue_callback_arg
Definition: starpu_task.h:765
unsigned color
Definition: starpu_task.h:522
unsigned use_tag
Definition: starpu_task.h:845
starpu_opencl_func_t opencl_func
Definition: starpu_task.h:310
char cuda_flags[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:349
Definition: starpu_task.h:250
starpu_opencl_func_t opencl_funcs[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:363
Definition: starpu_task.h:133
Definition: starpu_profiling.h:49
enum starpu_data_access_mode modes[STARPU_NMAXBUFS]
Definition: starpu_task.h:437
int nbuffers
Definition: starpu_task.h:584
void ** dyn_interfaces
Definition: starpu_task.h:612
void(* starpu_cpu_func_t)(void **, void *)
Definition: starpu_task.h:165
unsigned regenerate
Definition: starpu_task.h:917
int max_parallelism
Definition: starpu_task.h:289
int starpu_task_submit_to_ctx(struct starpu_task *task, unsigned sched_ctx_id)
void starpu_create_callback_task(void(*callback)(void *), void *callback_arg)
uint32_t * workerids
Definition: starpu_task.h:984
unsigned execute_on_a_specific_worker
Definition: starpu_task.h:878
void(* starpu_opencl_func_t)(void **, void *)
Definition: starpu_task.h:175
int starpu_task_nready(void)
unsigned possibly_parallel
Definition: starpu_task.h:1077
unsigned type
Definition: starpu_task.h:1045
struct starpu_omp_task * omp_task
Definition: starpu_task.h:1150
double predicted
Definition: starpu_task.h:1113
starpu_mic_func_t mic_funcs[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:386
int priority
Definition: starpu_task.h:1018
unsigned workerids_len
Definition: starpu_task.h:995
struct starpu_task * prev
Definition: starpu_task.h:1130
int starpu_task_submit(struct starpu_task *task) STARPU_WARN_UNUSED_RESULT
Definition: starpu_task.h:130
char opencl_flags[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:370
int starpu_task_nsubmitted(void)
struct _starpu_task_bundle * starpu_task_bundle_t
Definition: starpu_task_bundle.h:42
unsigned sequential_consistency
Definition: starpu_task.h:857
void starpu_iteration_pop(void)
unsigned workerid
Definition: starpu_task.h:954
const char * starpu_task_get_name(struct starpu_task *task)
size_t cl_arg_size
Definition: starpu_task.h:708
unsigned detach
Definition: starpu_task.h:889
int nodes[STARPU_NMAXBUFS]
Definition: starpu_task.h:470
const char * cpu_funcs_name[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:411
int magic
Definition: starpu_task.h:1034
struct starpu_profiling_task_info * profiling_info
Definition: starpu_task.h:1092
void starpu_task_clean(struct starpu_task *task)
unsigned sched_ctx
Definition: starpu_task.h:1061
void starpu_task_destroy(struct starpu_task *task)
unsigned nb_termination_call_required
Definition: starpu_task.h:1158
int starpu_task_wait_for_all(void)
starpu_cpu_func_t cpu_func
Definition: starpu_task.h:296
unsigned specific_nodes
Definition: starpu_task.h:461
void(* starpu_mpi_ms_kernel_t)(void **, void *)
Definition: starpu_task.h:190
void * cl_arg
Definition: starpu_task.h:691
starpu_data_handle_t * dyn_handles
Definition: starpu_task.h:604
uint32_t where
Definition: starpu_task.h:266
Definition: starpu_task.h:151
Definition: starpu_task.h:139
unsigned mf_skip
Definition: starpu_task.h:924
starpu_mpi_ms_func_t mpi_ms_funcs[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:402
Definition: starpu_task.h:154
unsigned no_submitorder
Definition: starpu_task.h:933
struct starpu_perfmodel * model
Definition: starpu_task.h:491
int starpu_task_wait_array(struct starpu_task **tasks, unsigned nb_tasks) STARPU_WARN_UNUSED_RESULT
struct starpu_task * next
Definition: starpu_task.h:1137
enum starpu_codelet_type type
Definition: starpu_task.h:282
starpu_cuda_func_t cuda_funcs[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:342
void * starpu_private
Definition: starpu_task.h:1143
starpu_mic_kernel_t(* starpu_mic_func_t)(void)
Definition: starpu_task.h:185
void(* starpu_mic_kernel_t)(void **, void *)
Definition: starpu_task.h:180
uint64_t starpu_tag_t
Definition: starpu_task_dep.h:129
struct starpu_task * starpu_task_dup(struct starpu_task *task)
unsigned color
Definition: starpu_task.h:1053
const char * name
Definition: starpu_task.h:516
Definition: starpu_task.h:153
int starpu_task_wait_for_all_in_ctx(unsigned sched_ctx_id)
Definition: starpu_task.h:548
starpu_task_status
Definition: starpu_task.h:147
starpu_mpi_ms_kernel_t(* starpu_mpi_ms_func_t)(void)
Definition: starpu_task.h:195
const char * name
Definition: starpu_task.h:557
void starpu_iteration_push(unsigned long iteration)
starpu_tag_t tag_id
Definition: starpu_task.h:778
unsigned synchronous
Definition: starpu_task.h:868
void starpu_task_set_implementation(struct starpu_task *task, unsigned impl)
int starpu_task_wait_for_n_submitted(unsigned n)
unsigned starpu_task_get_implementation(struct starpu_task *task)
void * sched_data
Definition: starpu_task.h:1167
int starpu_task_wait(struct starpu_task *task) STARPU_WARN_UNUSED_RESULT
enum starpu_data_access_mode * dyn_modes
Definition: starpu_task.h:629
int * dyn_nodes
Definition: starpu_task.h:483
struct _starpu_data_state * starpu_data_handle_t
Definition: starpu_data.h:46
#define STARPU_NMAXWORKERS
Definition: starpu_config.h:115
void starpu_task_init(struct starpu_task *task)
double predicted_transfer
Definition: starpu_task.h:1122
unsigned cl_arg_free
Definition: starpu_task.h:793
void starpu_codelet_init(struct starpu_codelet *cl)
Definition: starpu_task.h:157
Definition: starpu_task.h:149
Definition: starpu_task.h:158
unsigned prologue_callback_pop_arg_free
Definition: starpu_task.h:833
int32_t where
Definition: starpu_task.h:575
unsigned char * handles_sequential_consistency
Definition: starpu_task.h:675
struct starpu_task * starpu_task_create(void) STARPU_ATTRIBUTE_MALLOC
#define STARPU_MAXIMPLEMENTATIONS
Definition: starpu_config.h:117
const char * starpu_task_get_model_name(struct starpu_task *task)
unsigned long per_worker_stats[STARPU_NMAXWORKERS]
Definition: starpu_task.h:510
Definition: starpu_task.h:155
int nbuffers
Definition: starpu_task.h:426
int starpu_task_wait_for_n_submitted_in_ctx(unsigned sched_ctx_id, unsigned n)
struct starpu_perfmodel * energy_model
Definition: starpu_task.h:501
int(* can_execute)(unsigned workerid, struct starpu_task *task, unsigned nimpl)
Definition: starpu_task.h:273
enum starpu_data_access_mode * dyn_modes
Definition: starpu_task.h:450
starpu_data_access_mode
Definition: starpu_data.h:56
double flops
Definition: starpu_task.h:1105
unsigned destroy
Definition: starpu_task.h:906
void starpu_create_sync_task(starpu_tag_t sync_tag, unsigned ndeps, starpu_tag_t *deps, void(*callback)(void *), void *callback_arg)
void starpu_codelet_display_stats(struct starpu_codelet *cl)
starpu_task_bundle_t bundle
Definition: starpu_task.h:1083
unsigned callback_arg_free
Definition: starpu_task.h:806
starpu_codelet_type
Definition: starpu_task.h:128
Definition: starpu_task.h:156
#define STARPU_NMAXBUFS
Definition: starpu_config.h:109
void(* starpu_cuda_func_t)(void **, void *)
Definition: starpu_task.h:170
void * callback_arg
Definition: starpu_task.h:739
starpu_cuda_func_t cuda_func
Definition: starpu_task.h:303
Definition: starpu_task.h:533
Definition: starpu_task.h:159
unsigned workerorder
Definition: starpu_task.h:970
int flags
Definition: starpu_task.h:527
unsigned prologue_callback_arg_free
Definition: starpu_task.h:819
starpu_cpu_func_t cpu_funcs[STARPU_MAXIMPLEMENTATIONS]
Definition: starpu_task.h:327
struct starpu_task * starpu_task_get_current(void)
int starpu_task_get_current_data_node(unsigned i)
starpu_data_handle_t handle
Definition: starpu_task.h:535
struct starpu_codelet * cl
Definition: starpu_task.h:566
unsigned scheduled
Definition: starpu_task.h:940
int hypervisor_tag
Definition: starpu_task.h:1069
int starpu_task_wait_for_no_ready(void)