starpu_openmp.h 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2014 Inria
  4. *
  5. * StarPU is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * StarPU is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #ifndef __STARPU_OPENMP_H__
  17. #define __STARPU_OPENMP_H__
  18. #include <starpu_config.h>
  19. #if defined STARPU_OPENMP
  20. typedef void *starpu_omp_lock_t; /* TODO: select a proper type */
  21. typedef void *starpu_omp_nest_lock_t; /* TODO: select a proper type */
  22. typedef enum e_starpu_omp_sched
  23. {
  24. starpu_omp_sched_undefined = 0,
  25. starpu_omp_sched_static = 1,
  26. starpu_omp_sched_dynamic = 2,
  27. starpu_omp_sched_guided = 3,
  28. starpu_omp_sched_auto = 4,
  29. } starpu_omp_sched_t;
  30. typedef enum e_starpu_omp_proc_bind
  31. {
  32. starpu_omp_proc_bind_false = 0,
  33. starpu_omp_proc_bind_true = 1,
  34. starpu_omp_proc_bind_master = 2,
  35. starpu_omp_proc_bind_close = 3,
  36. starpu_omp_proc_bind_spread = 4,
  37. } starpu_omp_proc_bind_t;
  38. #ifdef __cplusplus
  39. extern "C"
  40. {
  41. #define __STARPU_OMP_NOTHROW throw ()
  42. #else
  43. #define __STARPU_OMP_NOTHROW __attribute__((__nothrow__))
  44. #endif
  45. extern int starpu_omp_init(void) __STARPU_OMP_NOTHROW;
  46. extern void starpu_omp_shutdown(void) __STARPU_OMP_NOTHROW;
  47. extern void starpu_omp_parallel_region(const struct starpu_codelet * const parallel_region_cl, void * const parallel_region_cl_arg) __STARPU_OMP_NOTHROW;
  48. extern void starpu_omp_task_region(const struct starpu_codelet * const _task_region_cl,
  49. void * const task_region_cl_arg,
  50. int if_clause, int final_clause, int untied_clause, int mergeable_clause) __STARPU_OMP_NOTHROW;
  51. extern void starpu_omp_barrier(void) __STARPU_OMP_NOTHROW;
  52. extern void starpu_omp_master(void (*f)(void *arg), void *arg, int nowait) __STARPU_OMP_NOTHROW;
  53. extern void starpu_omp_single(void (*f)(void *arg), void *arg, int nowait) __STARPU_OMP_NOTHROW;
  54. extern void starpu_omp_critical(void (*f)(void *arg), void *arg, const char *name) __STARPU_OMP_NOTHROW;
  55. extern void starpu_omp_taskwait(void) __STARPU_OMP_NOTHROW;
  56. extern void starpu_omp_taskgroup(void (*f)(void *arg), void *arg) __STARPU_OMP_NOTHROW;
  57. extern void starpu_omp_set_num_threads(int threads) __STARPU_OMP_NOTHROW;
  58. extern int starpu_omp_get_num_threads() __STARPU_OMP_NOTHROW;
  59. extern int starpu_omp_get_thread_num() __STARPU_OMP_NOTHROW;
  60. extern int starpu_omp_get_max_threads() __STARPU_OMP_NOTHROW;
  61. extern int starpu_omp_get_num_procs (void) __STARPU_OMP_NOTHROW;
  62. extern int starpu_omp_in_parallel (void) __STARPU_OMP_NOTHROW;
  63. extern void starpu_omp_set_dynamic (int dynamic_threads) __STARPU_OMP_NOTHROW;
  64. extern int starpu_omp_get_dynamic (void) __STARPU_OMP_NOTHROW;
  65. extern void starpu_omp_set_nested (int nested) __STARPU_OMP_NOTHROW;
  66. extern int starpu_omp_get_nested (void) __STARPU_OMP_NOTHROW;
  67. extern int starpu_omp_get_cancellation(void) __STARPU_OMP_NOTHROW;
  68. extern void starpu_omp_set_schedule (starpu_omp_sched_t kind, int modifier) __STARPU_OMP_NOTHROW;
  69. extern void starpu_omp_get_schedule (starpu_omp_sched_t *kind, int *modifier) __STARPU_OMP_NOTHROW;
  70. extern int starpu_omp_get_thread_limit (void) __STARPU_OMP_NOTHROW;
  71. extern void starpu_omp_set_max_active_levels (int max_levels) __STARPU_OMP_NOTHROW;
  72. extern int starpu_omp_get_max_active_levels (void) __STARPU_OMP_NOTHROW;
  73. extern int starpu_omp_get_level (void) __STARPU_OMP_NOTHROW;
  74. extern int starpu_omp_get_ancestor_thread_num (int level) __STARPU_OMP_NOTHROW;
  75. extern int starpu_omp_get_team_size (int level) __STARPU_OMP_NOTHROW;
  76. extern int starpu_omp_get_active_level (void) __STARPU_OMP_NOTHROW;
  77. extern int starpu_omp_in_final(void) __STARPU_OMP_NOTHROW;
  78. extern starpu_omp_proc_bind_t starpu_omp_get_proc_bind(void) __STARPU_OMP_NOTHROW;
  79. extern void starpu_omp_set_default_device(int device_num) __STARPU_OMP_NOTHROW;
  80. extern int starpu_omp_get_default_device(void) __STARPU_OMP_NOTHROW;
  81. extern int starpu_omp_get_num_devices(void) __STARPU_OMP_NOTHROW;
  82. extern int starpu_omp_get_num_teams(void) __STARPU_OMP_NOTHROW;
  83. extern int starpu_omp_get_team_num(void) __STARPU_OMP_NOTHROW;
  84. extern int starpu_omp_is_initial_device(void) __STARPU_OMP_NOTHROW;
  85. extern void starpu_omp_init_lock (starpu_omp_lock_t *lock) __STARPU_OMP_NOTHROW;
  86. extern void starpu_omp_destroy_lock (starpu_omp_lock_t *lock) __STARPU_OMP_NOTHROW;
  87. extern void starpu_omp_set_lock (starpu_omp_lock_t *lock) __STARPU_OMP_NOTHROW;
  88. extern void starpu_omp_unset_lock (starpu_omp_lock_t *lock) __STARPU_OMP_NOTHROW;
  89. extern int starpu_omp_test_lock (starpu_omp_lock_t *lock) __STARPU_OMP_NOTHROW;
  90. extern void starpu_omp_init_nest_lock (starpu_omp_nest_lock_t *lock) __STARPU_OMP_NOTHROW;
  91. extern void starpu_omp_destroy_nest_lock (starpu_omp_nest_lock_t *lock) __STARPU_OMP_NOTHROW;
  92. extern void starpu_omp_set_nest_lock (starpu_omp_nest_lock_t *lock) __STARPU_OMP_NOTHROW;
  93. extern void starpu_omp_unset_nest_lock (starpu_omp_nest_lock_t *lock) __STARPU_OMP_NOTHROW;
  94. extern int starpu_omp_test_nest_lock (starpu_omp_nest_lock_t *lock) __STARPU_OMP_NOTHROW;
  95. extern double starpu_omp_get_wtime (void) __STARPU_OMP_NOTHROW;
  96. extern double starpu_omp_get_wtick (void) __STARPU_OMP_NOTHROW;
  97. #ifdef __cplusplus
  98. }
  99. #endif
  100. #endif /* STARPU_USE_OPENMP && !STARPU_DONT_INCLUDE_OPENMP_HEADERS */
  101. #endif /* __STARPU_OPENMP_H__ */