sched_policy.h 3.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010, 2012-2013, 2015 Université de Bordeaux
  4. * Copyright (C) 2011 INRIA
  5. *
  6. * StarPU is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU Lesser General Public License as published by
  8. * the Free Software Foundation; either version 2.1 of the License, or (at
  9. * your option) any later version.
  10. *
  11. * StarPU is distributed in the hope that it will be useful, but
  12. * WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  14. *
  15. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  16. */
  17. #ifndef __SCHED_POLICY_H__
  18. #define __SCHED_POLICY_H__
  19. #include <starpu.h>
  20. #include <core/workers.h>
  21. #include <core/sched_ctx.h>
  22. #include <starpu_scheduler.h>
  23. struct starpu_machine_config;
  24. struct starpu_sched_policy *_starpu_get_sched_policy( struct _starpu_sched_ctx *sched_ctx);
  25. void _starpu_init_sched_policy(struct _starpu_machine_config *config,
  26. struct _starpu_sched_ctx *sched_ctx, struct starpu_sched_policy *policy);
  27. void _starpu_deinit_sched_policy(struct _starpu_sched_ctx *sched_ctx);
  28. struct starpu_sched_policy *_starpu_select_sched_policy(struct _starpu_machine_config *config, const char *required_policy);
  29. int _starpu_push_task(struct _starpu_job *task);
  30. int _starpu_repush_task(struct _starpu_job *task);
  31. /* actually pushes the tasks to the specific worker or to the scheduler */
  32. int _starpu_push_task_to_workers(struct starpu_task *task);
  33. /* pop a task that can be executed on the worker */
  34. struct starpu_task *_starpu_pop_task(struct _starpu_worker *worker);
  35. /* pop every task that can be executed on the worker */
  36. struct starpu_task *_starpu_pop_every_task(struct _starpu_sched_ctx *sched_ctx);
  37. void _starpu_sched_post_exec_hook(struct starpu_task *task);
  38. int _starpu_pop_task_end(struct starpu_task *task);
  39. void _starpu_wait_on_sched_event(void);
  40. struct starpu_task *_starpu_create_conversion_task(starpu_data_handle_t handle,
  41. unsigned int node);
  42. struct starpu_task *_starpu_create_conversion_task_for_arch(starpu_data_handle_t handle,
  43. enum starpu_node_kind node_kind);
  44. void _starpu_sched_pre_exec_hook(struct starpu_task *task);
  45. void _starpu_print_idle_time();
  46. /*
  47. * Predefined policies
  48. */
  49. extern struct starpu_sched_policy _starpu_sched_lws_policy;
  50. extern struct starpu_sched_policy _starpu_sched_ws_policy;
  51. extern struct starpu_sched_policy _starpu_sched_prio_policy;
  52. extern struct starpu_sched_policy _starpu_sched_random_policy;
  53. extern struct starpu_sched_policy _starpu_sched_dm_policy;
  54. extern struct starpu_sched_policy _starpu_sched_dmda_policy;
  55. extern struct starpu_sched_policy _starpu_sched_dmda_ready_policy;
  56. extern struct starpu_sched_policy _starpu_sched_dmda_sorted_policy;
  57. extern struct starpu_sched_policy _starpu_sched_dmda_sorted_decision_policy;
  58. extern struct starpu_sched_policy _starpu_sched_eager_policy;
  59. extern struct starpu_sched_policy _starpu_sched_parallel_heft_policy;
  60. extern struct starpu_sched_policy _starpu_sched_peager_policy;
  61. extern struct starpu_sched_policy _starpu_sched_heteroprio_policy;
  62. extern struct starpu_sched_policy _starpu_sched_modular_eager_policy;
  63. extern struct starpu_sched_policy _starpu_sched_modular_eager_prefetching_policy;
  64. extern struct starpu_sched_policy _starpu_sched_modular_prio_policy;
  65. extern struct starpu_sched_policy _starpu_sched_modular_prio_prefetching_policy;
  66. extern struct starpu_sched_policy _starpu_sched_modular_random_policy;
  67. extern struct starpu_sched_policy _starpu_sched_modular_random_prio_policy;
  68. extern struct starpu_sched_policy _starpu_sched_modular_random_prefetching_policy;
  69. extern struct starpu_sched_policy _starpu_sched_modular_random_prio_prefetching_policy;
  70. extern struct starpu_sched_policy _starpu_sched_modular_ws_policy;
  71. extern struct starpu_sched_policy _starpu_sched_modular_heft_policy;
  72. extern struct starpu_sched_policy _starpu_sched_modular_heft2_policy;
  73. #endif // __SCHED_POLICY_H__