starpu_thread.h 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010, 2012-2014 Université de Bordeaux 1
  4. * Copyright (C) 2010, 2011, 2012, 2013 Centre National de la Recherche Scientifique
  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 __STARPU_THREAD_H__
  18. #define __STARPU_THREAD_H__
  19. #ifdef __cplusplus
  20. extern "C"
  21. {
  22. #endif
  23. #include <starpu_config.h>
  24. #ifdef STARPU_SIMGRID
  25. #include <xbt/synchro_core.h>
  26. #include <msg/msg.h>
  27. #elif !defined(_MSC_VER)
  28. #include <pthread.h>
  29. #endif
  30. /*
  31. * Encapsulation of the pthread_create function.
  32. */
  33. #ifdef STARPU_SIMGRID
  34. typedef int starpu_pthread_t;
  35. typedef int starpu_pthread_attr_t;
  36. int starpu_pthread_create_on(char *name, starpu_pthread_t *thread, const starpu_pthread_attr_t *attr, void *(*start_routine) (void *), void *arg, int where);
  37. int starpu_pthread_create(starpu_pthread_t *thread, const starpu_pthread_attr_t *attr, void *(*start_routine) (void *), void *arg);
  38. int starpu_pthread_join(starpu_pthread_t thread, void **retval);
  39. int starpu_pthread_exit(void *retval);
  40. int starpu_pthread_attr_init(starpu_pthread_attr_t *attr);
  41. int starpu_pthread_attr_destroy(starpu_pthread_attr_t *attr);
  42. int starpu_pthread_attr_setdetachstate(starpu_pthread_attr_t *attr, int detachstate);
  43. #elif !defined(_MSC_VER) /* STARPU_SIMGRID */
  44. typedef pthread_t starpu_pthread_t;
  45. typedef pthread_attr_t starpu_pthread_attr_t;
  46. #define starpu_pthread_create pthread_create
  47. #define starpu_pthread_create_on(name, thread, attr, routine, arg, where) starpu_pthread_create(thread, attr, routine, arg)
  48. #define starpu_pthread_join pthread_join
  49. #define starpu_pthread_exit pthread_exit
  50. #define starpu_pthread_attr_init pthread_attr_init
  51. #define starpu_pthread_attr_destroy pthread_attr_destroy
  52. #define starpu_pthread_attr_setdetachstate pthread_attr_setdetachstate
  53. #endif /* STARPU_SIMGRID, _MSC_VER */
  54. /*
  55. * Encapsulation of the pthread_mutex_* functions.
  56. */
  57. #ifdef STARPU_SIMGRID
  58. typedef xbt_mutex_t starpu_pthread_mutex_t;
  59. typedef int starpu_pthread_mutexattr_t;
  60. #define STARPU_PTHREAD_MUTEX_INITIALIZER NULL
  61. int starpu_pthread_mutex_init(starpu_pthread_mutex_t *mutex, const starpu_pthread_mutexattr_t *mutexattr);
  62. int starpu_pthread_mutex_destroy(starpu_pthread_mutex_t *mutex);
  63. int starpu_pthread_mutex_lock(starpu_pthread_mutex_t *mutex);
  64. int starpu_pthread_mutex_unlock(starpu_pthread_mutex_t *mutex);
  65. int starpu_pthread_mutex_trylock(starpu_pthread_mutex_t *mutex);
  66. #elif !defined(_MSC_VER) /* !STARPU_SIMGRID */
  67. typedef pthread_mutex_t starpu_pthread_mutex_t;
  68. typedef pthread_mutexattr_t starpu_pthread_mutexattr_t;
  69. #define starpu_pthread_mutex_init pthread_mutex_init
  70. #define starpu_pthread_mutex_destroy pthread_mutex_destroy
  71. int starpu_pthread_mutex_lock(starpu_pthread_mutex_t *mutex);
  72. int starpu_pthread_mutex_unlock(starpu_pthread_mutex_t *mutex);
  73. int starpu_pthread_mutex_trylock(starpu_pthread_mutex_t *mutex);
  74. #define STARPU_PTHREAD_MUTEX_INITIALIZER PTHREAD_MUTEX_INITIALIZER
  75. #endif /* STARPU_SIMGRID, _MSC_VER */
  76. /*
  77. * Encapsulation of the pthread_key_* functions.
  78. */
  79. #ifdef STARPU_SIMGRID
  80. typedef int starpu_pthread_key_t;
  81. int starpu_pthread_key_create(starpu_pthread_key_t *key, void (*destr_function) (void *));
  82. int starpu_pthread_key_delete(starpu_pthread_key_t key);
  83. int starpu_pthread_setspecific(starpu_pthread_key_t key, const void *pointer);
  84. void *starpu_pthread_getspecific(starpu_pthread_key_t key);
  85. #elif !defined(_MSC_VER) /* STARPU_SIMGRID */
  86. typedef pthread_key_t starpu_pthread_key_t;
  87. #define starpu_pthread_key_create pthread_key_create
  88. #define starpu_pthread_key_delete pthread_key_delete
  89. #define starpu_pthread_setspecific pthread_setspecific
  90. #define starpu_pthread_getspecific pthread_getspecific
  91. #endif /* STARPU_SIMGRID, _MSC_VER */
  92. /*
  93. * Encapsulation of the pthread_cond_* functions.
  94. */
  95. #ifdef STARPU_SIMGRID
  96. typedef xbt_cond_t starpu_pthread_cond_t;
  97. typedef int starpu_pthread_condattr_t;
  98. #define STARPU_PTHREAD_COND_INITIALIZER NULL
  99. int starpu_pthread_cond_init(starpu_pthread_cond_t *cond, starpu_pthread_condattr_t *cond_attr);
  100. int starpu_pthread_cond_signal(starpu_pthread_cond_t *cond);
  101. int starpu_pthread_cond_broadcast(starpu_pthread_cond_t *cond);
  102. int starpu_pthread_cond_wait(starpu_pthread_cond_t *cond, starpu_pthread_mutex_t *mutex);
  103. int starpu_pthread_cond_timedwait(starpu_pthread_cond_t *cond, starpu_pthread_mutex_t *mutex, const struct timespec *abstime);
  104. int starpu_pthread_cond_destroy(starpu_pthread_cond_t *cond);
  105. #elif !defined(_MSC_VER) /* STARPU_SIMGRID */
  106. typedef pthread_cond_t starpu_pthread_cond_t;
  107. typedef pthread_condattr_t starpu_pthread_condattr_t;
  108. #define STARPU_PTHREAD_COND_INITIALIZER PTHREAD_COND_INITIALIZER
  109. #define starpu_pthread_cond_init pthread_cond_init
  110. #define starpu_pthread_cond_signal pthread_cond_signal
  111. #define starpu_pthread_cond_broadcast pthread_cond_broadcast
  112. int starpu_pthread_cond_wait(starpu_pthread_cond_t *cond, starpu_pthread_mutex_t *mutex);
  113. #define starpu_pthread_cond_timedwait pthread_cond_timedwait
  114. #define starpu_pthread_cond_destroy pthread_cond_destroy
  115. #endif /* STARPU_SIMGRID, _MSC_VER */
  116. /*
  117. * Encapsulation of the pthread_rwlock_* functions.
  118. */
  119. #ifdef STARPU_SIMGRID
  120. typedef xbt_mutex_t starpu_pthread_rwlock_t;
  121. typedef int starpu_pthread_rwlockattr_t;
  122. int starpu_pthread_rwlock_init(starpu_pthread_rwlock_t *rwlock, const starpu_pthread_rwlockattr_t *attr);
  123. int starpu_pthread_rwlock_destroy(starpu_pthread_rwlock_t *rwlock);
  124. int starpu_pthread_rwlock_rdlock(starpu_pthread_rwlock_t *rwlock);
  125. int starpu_pthread_rwlock_tryrdlock(starpu_pthread_rwlock_t *rwlock);
  126. int starpu_pthread_rwlock_wrlock(starpu_pthread_rwlock_t *rwlock);
  127. int starpu_pthread_rwlock_trywrlock(starpu_pthread_rwlock_t *rwlock);
  128. int starpu_pthread_rwlock_unlock(starpu_pthread_rwlock_t *rwlock);
  129. #elif !defined(_MSC_VER) /* STARPU_SIMGRID */
  130. typedef pthread_rwlock_t starpu_pthread_rwlock_t;
  131. typedef pthread_rwlockattr_t starpu_pthread_rwlockattr_t;
  132. #define starpu_pthread_rwlock_init pthread_rwlock_init
  133. #define starpu_pthread_rwlock_destroy pthread_rwlock_destroy
  134. int starpu_pthread_rwlock_rdlock(starpu_pthread_rwlock_t *rwlock);
  135. int starpu_pthread_rwlock_tryrdlock(starpu_pthread_rwlock_t *rwlock);
  136. int starpu_pthread_rwlock_wrlock(starpu_pthread_rwlock_t *rwlock);
  137. int starpu_pthread_rwlock_trywrlock(starpu_pthread_rwlock_t *rwlock);
  138. int starpu_pthread_rwlock_unlock(starpu_pthread_rwlock_t *rwlock);
  139. #endif /* STARPU_SIMGRID, _MSC_VER */
  140. /*
  141. * Encapsulation of the pthread_barrier_* functions.
  142. */
  143. #ifdef STARPU_SIMGRID
  144. typedef struct {
  145. starpu_pthread_mutex_t mutex;
  146. starpu_pthread_cond_t cond;
  147. unsigned count;
  148. unsigned done;
  149. } starpu_pthread_barrier_t;
  150. typedef int starpu_pthread_barrierattr_t;
  151. #define STARPU_PTHREAD_BARRIER_SERIAL_THREAD -1
  152. int starpu_pthread_barrier_init(starpu_pthread_barrier_t *barrier, const starpu_pthread_barrierattr_t *attr, unsigned count);
  153. int starpu_pthread_barrier_destroy(starpu_pthread_barrier_t *barrier);
  154. int starpu_pthread_barrier_wait(starpu_pthread_barrier_t *barrier);
  155. #elif !defined(_MSC_VER) /* STARPU_SIMGRID */
  156. typedef pthread_barrier_t starpu_pthread_barrier_t;
  157. typedef pthread_barrierattr_t starpu_pthread_barrierattr_t;
  158. #define starpu_pthread_barrier_init pthread_barrier_init
  159. #define starpu_pthread_barrier_destroy pthread_barrier_destroy
  160. int starpu_pthread_barrier_wait(starpu_pthread_barrier_t *barrier);
  161. #define STARPU_PTHREAD_BARRIER_SERIAL_THREAD PTHREAD_BARRIER_SERIAL_THREAD
  162. #endif /* STARPU_SIMGRID, _MSC_VER */
  163. #ifdef _MSC_VER
  164. typedef void* starpu_pthread_rwlock_t;
  165. typedef void* starpu_pthread_mutex_t;
  166. typedef void* starpu_pthread_cond_t;
  167. typedef void* starpu_pthread_barrier_t;
  168. #endif /* _MSC_VER */
  169. #ifdef __cplusplus
  170. }
  171. #endif
  172. #endif /* __STARPU_THREAD_H__ */