fstarpu_mpi_mod.f90 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754
  1. ! StarPU --- Runtime system for heterogeneous multicore architectures.
  2. !
  3. ! Copyright (C) 2017 CNRS
  4. ! Copyright (C) 2016 Inria
  5. ! Copyright (C) 2016,2017 Université de Bordeaux
  6. !
  7. ! StarPU is free software; you can redistribute it and/or modify
  8. ! it under the terms of the GNU Lesser General Public License as published by
  9. ! the Free Software Foundation; either version 2.1 of the License, or (at
  10. ! your option) any later version.
  11. !
  12. ! StarPU is distributed in the hope that it will be useful, but
  13. ! WITHOUT ANY WARRANTY; without even the implied warranty of
  14. ! MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  15. !
  16. ! See the GNU Lesser General Public License in COPYING.LGPL for more details.
  17. !
  18. module fstarpu_mpi_mod
  19. use iso_c_binding
  20. use fstarpu_mod
  21. implicit none
  22. interface
  23. ! == mpi/include/starpu_mpi.h ==
  24. ! int starpu_mpi_isend(starpu_data_handle_t data_handle, starpu_mpi_req *req, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm);
  25. function fstarpu_mpi_isend (dh, mpi_req, dst, data_tag, mpi_comm) bind(C)
  26. use iso_c_binding
  27. implicit none
  28. integer(c_int) :: fstarpu_mpi_isend
  29. type(c_ptr), value, intent(in) :: dh
  30. type(c_ptr), value, intent(in) :: mpi_req
  31. integer(c_int), value, intent(in) :: dst
  32. integer(c_int), value, intent(in) :: data_tag
  33. integer(c_int), value, intent(in) :: mpi_comm
  34. end function fstarpu_mpi_isend
  35. ! == mpi/include/starpu_mpi.h ==
  36. ! int starpu_mpi_isend_prio(starpu_data_handle_t data_handle, starpu_mpi_req *req, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm);
  37. function fstarpu_mpi_isend_prio (dh, mpi_req, dst, data_tag, prio, mpi_comm) bind(C)
  38. use iso_c_binding
  39. implicit none
  40. integer(c_int) :: fstarpu_mpi_isend_prio
  41. type(c_ptr), value, intent(in) :: dh
  42. type(c_ptr), value, intent(in) :: mpi_req
  43. integer(c_int), value, intent(in) :: dst
  44. integer(c_int), value, intent(in) :: data_tag
  45. integer(c_int), value, intent(in) :: prio
  46. integer(c_int), value, intent(in) :: mpi_comm
  47. end function fstarpu_mpi_isend_prio
  48. ! int starpu_mpi_irecv(starpu_data_handle_t data_handle, starpu_mpi_req *req, int source, starpu_mpi_tag_t data_tag, MPI_Comm comm);
  49. function fstarpu_mpi_irecv (dh, mpi_req, src, data_tag, mpi_comm) bind(C)
  50. use iso_c_binding
  51. implicit none
  52. integer(c_int) :: fstarpu_mpi_irecv
  53. type(c_ptr), value, intent(in) :: dh
  54. type(c_ptr), value, intent(in) :: mpi_req
  55. integer(c_int), value, intent(in) :: src
  56. integer(c_int), value, intent(in) :: data_tag
  57. integer(c_int), value, intent(in) :: mpi_comm
  58. end function fstarpu_mpi_irecv
  59. ! int starpu_mpi_send(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm);
  60. function fstarpu_mpi_send (dh, dst, data_tag, mpi_comm) bind(C)
  61. use iso_c_binding
  62. implicit none
  63. integer(c_int) :: fstarpu_mpi_send
  64. type(c_ptr), value, intent(in) :: dh
  65. integer(c_int), value, intent(in) :: dst
  66. integer(c_int), value, intent(in) :: data_tag
  67. integer(c_int), value, intent(in) :: mpi_comm
  68. end function fstarpu_mpi_send
  69. ! int starpu_mpi_send_prio(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm);
  70. function fstarpu_mpi_send_prio (dh, dst, data_tag, prio, mpi_comm) bind(C)
  71. use iso_c_binding
  72. implicit none
  73. integer(c_int) :: fstarpu_mpi_send_prio
  74. type(c_ptr), value, intent(in) :: dh
  75. integer(c_int), value, intent(in) :: dst
  76. integer(c_int), value, intent(in) :: data_tag
  77. integer(c_int), value, intent(in) :: prio
  78. integer(c_int), value, intent(in) :: mpi_comm
  79. end function fstarpu_mpi_send_prio
  80. ! int starpu_mpi_recv(starpu_data_handle_t data_handle, int source, starpu_mpi_tag_t data_tag, MPI_Comm comm, MPI_Status *status);
  81. function fstarpu_mpi_recv (dh, src, data_tag, mpi_comm, mpi_status) bind(C)
  82. use iso_c_binding
  83. implicit none
  84. integer(c_int) :: fstarpu_mpi_recv
  85. type(c_ptr), value, intent(in) :: dh
  86. integer(c_int), value, intent(in) :: src
  87. integer(c_int), value, intent(in) :: data_tag
  88. integer(c_int), value, intent(in) :: mpi_comm
  89. type(c_ptr), value, intent(in) :: mpi_status
  90. end function fstarpu_mpi_recv
  91. ! int starpu_mpi_isend_detached(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm, void (*callback)(void *), void *arg);
  92. function fstarpu_mpi_isend_detached (dh, dst, data_tag, mpi_comm, callback, arg) bind(C)
  93. use iso_c_binding
  94. implicit none
  95. integer(c_int) :: fstarpu_mpi_isend_detached
  96. type(c_ptr), value, intent(in) :: dh
  97. integer(c_int), value, intent(in) :: dst
  98. integer(c_int), value, intent(in) :: data_tag
  99. integer(c_int), value, intent(in) :: mpi_comm
  100. type(c_funptr), value, intent(in) :: callback
  101. type(c_ptr), value, intent(in) :: arg
  102. end function fstarpu_mpi_isend_detached
  103. ! int starpu_mpi_isend_detached_prio(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm, void (*callback)(void *), void *arg);
  104. function fstarpu_mpi_isend_detached_prio (dh, dst, data_tag, prio, mpi_comm, callback, arg) bind(C)
  105. use iso_c_binding
  106. implicit none
  107. integer(c_int) :: fstarpu_mpi_isend_detached_prio
  108. type(c_ptr), value, intent(in) :: dh
  109. integer(c_int), value, intent(in) :: dst
  110. integer(c_int), value, intent(in) :: data_tag
  111. integer(c_int), value, intent(in) :: prio
  112. integer(c_int), value, intent(in) :: mpi_comm
  113. type(c_funptr), value, intent(in) :: callback
  114. type(c_ptr), value, intent(in) :: arg
  115. end function fstarpu_mpi_isend_detached_prio
  116. ! int starpu_mpi_irecv_detached(starpu_data_handle_t data_handle, int source, starpu_mpi_tag_t data_tag, MPI_Comm comm, void (*callback)(void *), void *arg);
  117. function fstarpu_mpi_recv_detached (dh, src, data_tag, mpi_comm, callback, arg) bind(C)
  118. use iso_c_binding
  119. implicit none
  120. integer(c_int) :: fstarpu_mpi_recv_detached
  121. type(c_ptr), value, intent(in) :: dh
  122. integer(c_int), value, intent(in) :: src
  123. integer(c_int), value, intent(in) :: data_tag
  124. integer(c_int), value, intent(in) :: mpi_comm
  125. type(c_funptr), value, intent(in) :: callback
  126. type(c_ptr), value, intent(in) :: arg
  127. end function fstarpu_mpi_recv_detached
  128. ! int starpu_mpi_issend(starpu_data_handle_t data_handle, starpu_mpi_req *req, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm);
  129. function fstarpu_mpi_issend (dh, mpi_req, dst, data_tag, mpi_comm) bind(C)
  130. use iso_c_binding
  131. implicit none
  132. integer(c_int) :: fstarpu_mpi_issend
  133. type(c_ptr), value, intent(in) :: dh
  134. type(c_ptr), value, intent(in) :: mpi_req
  135. integer(c_int), value, intent(in) :: dst
  136. integer(c_int), value, intent(in) :: data_tag
  137. integer(c_int), value, intent(in) :: mpi_comm
  138. end function fstarpu_mpi_issend
  139. ! int starpu_mpi_issend_prio(starpu_data_handle_t data_handle, starpu_mpi_req *req, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm);
  140. function fstarpu_mpi_issend_prio (dh, mpi_req, dst, data_tag, prio, mpi_comm) bind(C)
  141. use iso_c_binding
  142. implicit none
  143. integer(c_int) :: fstarpu_mpi_issend_prio
  144. type(c_ptr), value, intent(in) :: dh
  145. type(c_ptr), value, intent(in) :: mpi_req
  146. integer(c_int), value, intent(in) :: dst
  147. integer(c_int), value, intent(in) :: data_tag
  148. integer(c_int), value, intent(in) :: prio
  149. integer(c_int), value, intent(in) :: mpi_comm
  150. end function fstarpu_mpi_issend_prio
  151. ! int starpu_mpi_issend_detached(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm, void (*callback)(void *), void *arg);
  152. function fstarpu_mpi_issend_detached (dh, dst, data_tag, mpi_comm, callback, arg) bind(C)
  153. use iso_c_binding
  154. implicit none
  155. integer(c_int) :: fstarpu_mpi_issend_detached
  156. type(c_ptr), value, intent(in) :: dh
  157. integer(c_int), value, intent(in) :: dst
  158. integer(c_int), value, intent(in) :: data_tag
  159. integer(c_int), value, intent(in) :: mpi_comm
  160. type(c_funptr), value, intent(in) :: callback
  161. type(c_ptr), value, intent(in) :: arg
  162. end function fstarpu_mpi_issend_detached
  163. ! int starpu_mpi_issend_detached_prio(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm, void (*callback)(void *), void *arg);
  164. function fstarpu_mpi_issend_detached_prio (dh, dst, data_tag, prio, mpi_comm, callback, arg) bind(C)
  165. use iso_c_binding
  166. implicit none
  167. integer(c_int) :: fstarpu_mpi_issend_detached_prio
  168. type(c_ptr), value, intent(in) :: dh
  169. integer(c_int), value, intent(in) :: dst
  170. integer(c_int), value, intent(in) :: data_tag
  171. integer(c_int), value, intent(in) :: prio
  172. integer(c_int), value, intent(in) :: mpi_comm
  173. type(c_funptr), value, intent(in) :: callback
  174. type(c_ptr), value, intent(in) :: arg
  175. end function fstarpu_mpi_issend_detached_prio
  176. ! int starpu_mpi_wait(starpu_mpi_req *req, MPI_Status *status);
  177. function fstarpu_mpi_wait(req,st) bind(C,name="starpu_mpi_wait")
  178. use iso_c_binding
  179. implicit none
  180. integer(c_int) :: fstarpu_mpi_wait
  181. type(c_ptr), value, intent(in) :: req
  182. type(c_ptr), value, intent(in) :: st
  183. end function fstarpu_mpi_wait
  184. ! int starpu_mpi_test(starpu_mpi_req *req, int *flag, MPI_Status *status);
  185. function fstarpu_mpi_test(req,flag,st) bind(C,name="starpu_mpi_test")
  186. use iso_c_binding
  187. implicit none
  188. integer(c_int) :: fstarpu_mpi_test
  189. type(c_ptr), value, intent(in) :: req
  190. type(c_ptr), value, intent(in) :: flag
  191. type(c_ptr), value, intent(in) :: st
  192. end function fstarpu_mpi_test
  193. ! int starpu_mpi_barrier(MPI_Comm comm);
  194. function fstarpu_mpi_barrier (mpi_comm) bind(C)
  195. use iso_c_binding
  196. implicit none
  197. integer(c_int) :: fstarpu_mpi_barrier
  198. integer(c_int), value, intent(in) :: mpi_comm
  199. end function fstarpu_mpi_barrier
  200. ! int starpu_mpi_irecv_detached_sequential_consistency(starpu_data_handle_t data_handle, int source, starpu_mpi_tag_t data_tag, MPI_Comm comm, void (*callback)(void *), void *arg, int sequential_consistency);
  201. function fstarpu_mpi_recv_detached_sequential_consistency (dh, src, data_tag, mpi_comm, callback, arg, seq_const) &
  202. bind(C)
  203. use iso_c_binding
  204. implicit none
  205. integer(c_int) :: fstarpu_mpi_recv_detached_sequential_consistency
  206. type(c_ptr), value, intent(in) :: dh
  207. integer(c_int), value, intent(in) :: src
  208. integer(c_int), value, intent(in) :: data_tag
  209. integer(c_int), value, intent(in) :: mpi_comm
  210. type(c_funptr), value, intent(in) :: callback
  211. type(c_ptr), value, intent(in) :: arg
  212. integer(c_int), value, intent(in) :: seq_const
  213. end function fstarpu_mpi_recv_detached_sequential_consistency
  214. ! int starpu_mpi_init_comm(int *argc, char ***argv, int initialize_mpi, MPI_Comm comm);
  215. ! -> cf fstarpu_mpi_init
  216. ! int starpu_mpi_init(int *argc, char ***argv, int initialize_mpi);
  217. ! -> cf fstarpu_mpi_init
  218. ! int starpu_mpi_initialize(void) STARPU_DEPRECATED;
  219. ! -> cf fstarpu_mpi_init
  220. ! int starpu_mpi_initialize_extended(int *rank, int *world_size) STARPU_DEPRECATED;
  221. ! -> cf fstarpu_mpi_init
  222. ! int starpu_mpi_shutdown(void);
  223. function fstarpu_mpi_shutdown () bind(C,name="starpu_mpi_shutdown")
  224. use iso_c_binding
  225. implicit none
  226. integer(c_int) :: fstarpu_mpi_shutdown
  227. end function fstarpu_mpi_shutdown
  228. ! struct starpu_task *starpu_mpi_task_build(MPI_Comm comm, struct starpu_codelet *codelet, ...);
  229. function fstarpu_mpi_task_build(mpi_comm,arglist) bind(C)
  230. use iso_c_binding, only: c_ptr,c_int
  231. type(c_ptr) :: fstarpu_mpi_task_build
  232. integer(c_int), value, intent(in) :: mpi_comm
  233. type(c_ptr), dimension(:), intent(in) :: arglist
  234. end function fstarpu_mpi_task_build
  235. ! int starpu_mpi_task_post_build(MPI_Comm comm, struct starpu_codelet *codelet, ...);
  236. function fstarpu_mpi_task_post_build(mpi_comm,arglist) bind(C)
  237. use iso_c_binding, only: c_ptr,c_int
  238. integer(c_int) :: fstarpu_mpi_task_post_build
  239. integer(c_int), value, intent(in) :: mpi_comm
  240. type(c_ptr), dimension(:), intent(in) :: arglist
  241. end function fstarpu_mpi_task_post_build
  242. ! int starpu_mpi_task_insert(MPI_Comm comm, struct starpu_codelet *codelet, ...);
  243. function fstarpu_mpi_task_insert(mpi_comm,arglist) bind(C)
  244. use iso_c_binding, only: c_ptr,c_int
  245. integer(c_int) :: fstarpu_mpi_task_insert
  246. integer(c_int), value, intent(in) :: mpi_comm
  247. type(c_ptr), dimension(:), intent(in) :: arglist
  248. end function fstarpu_mpi_task_insert
  249. function fstarpu_mpi_insert_task(mpi_comm,arglist) bind(C,name="fstarpu_mpi_task_insert")
  250. use iso_c_binding, only: c_ptr,c_int
  251. integer(c_int) :: fstarpu_mpi_insert_task
  252. integer(c_int), value, intent(in) :: mpi_comm
  253. type(c_ptr), dimension(:), intent(in) :: arglist
  254. end function fstarpu_mpi_insert_task
  255. ! void starpu_mpi_get_data_on_node(MPI_Comm comm, starpu_data_handle_t data_handle, int node);
  256. subroutine fstarpu_mpi_get_data_on_node(mpi_comm,dh,node) bind(C)
  257. use iso_c_binding
  258. implicit none
  259. integer(c_int), value, intent(in) :: mpi_comm
  260. type(c_ptr), value, intent(in) :: dh
  261. integer(c_int), value, intent(in) :: node
  262. end subroutine fstarpu_mpi_get_data_on_node
  263. ! void starpu_mpi_get_data_on_node_detached(MPI_Comm comm, starpu_data_handle_t data_handle, int node, void (*callback)(void*), void *arg);
  264. subroutine fstarpu_mpi_get_data_on_node_detached(mpi_comm,dh,node,callback,arg) bind(C)
  265. use iso_c_binding
  266. implicit none
  267. integer(c_int), value, intent(in) :: mpi_comm
  268. type(c_ptr), value, intent(in) :: dh
  269. integer(c_int), value, intent(in) :: node
  270. type(c_funptr), value, intent(in) :: callback
  271. type(c_ptr), value, intent(in) :: arg
  272. end subroutine fstarpu_mpi_get_data_on_node_detached
  273. ! void starpu_mpi_redux_data(MPI_Comm comm, starpu_data_handle_t data_handle);
  274. subroutine fstarpu_mpi_redux_data(mpi_comm,dh) bind(C)
  275. use iso_c_binding
  276. implicit none
  277. integer(c_int), value, intent(in) :: mpi_comm
  278. type(c_ptr), value, intent(in) :: dh
  279. end subroutine fstarpu_mpi_redux_data
  280. ! void starpu_mpi_redux_data_prio(MPI_Comm comm, starpu_data_handle_t data_handle, int prio);
  281. subroutine fstarpu_mpi_redux_data_prio(mpi_comm,dh, prio) bind(C)
  282. use iso_c_binding
  283. implicit none
  284. integer(c_int), value, intent(in) :: mpi_comm
  285. type(c_ptr), value, intent(in) :: dh
  286. integer(c_int), value, intent(in) :: prio
  287. end subroutine fstarpu_mpi_redux_data_prio
  288. ! int starpu_mpi_scatter_detached(starpu_data_handle_t *data_handles, int count, int root, MPI_Comm comm, void (*scallback)(void *), void *sarg, void (*rcallback)(void *), void *rarg);
  289. function fstarpu_mpi_scatter_detached (dhs, cnt, root, mpi_comm, scallback, sarg, rcallback, rarg) bind(C)
  290. use iso_c_binding
  291. implicit none
  292. integer(c_int) :: fstarpu_mpi_scatter_detached
  293. type(c_ptr), intent(in) :: dhs(*)
  294. integer(c_int), value, intent(in) :: cnt
  295. integer(c_int), value, intent(in) :: root
  296. integer(c_int), value, intent(in) :: mpi_comm
  297. type(c_funptr), value, intent(in) :: scallback
  298. type(c_ptr), value, intent(in) :: sarg
  299. type(c_funptr), value, intent(in) :: rcallback
  300. type(c_ptr), value, intent(in) :: rarg
  301. end function fstarpu_mpi_scatter_detached
  302. ! int starpu_mpi_gather_detached(starpu_data_handle_t *data_handles, int count, int root, MPI_Comm comm, void (*scallback)(void *), void *sarg, void (*rcallback)(void *), void *rarg);
  303. function fstarpu_mpi_gather_detached (dhs, cnt, root, mpi_comm, scallback, sarg, rcallback, rarg) bind(C)
  304. use iso_c_binding
  305. implicit none
  306. integer(c_int) :: fstarpu_mpi_gather_detached
  307. type(c_ptr), intent(in) :: dhs(*)
  308. integer(c_int), value, intent(in) :: cnt
  309. integer(c_int), value, intent(in) :: root
  310. integer(c_int), value, intent(in) :: mpi_comm
  311. type(c_funptr), value, intent(in) :: scallback
  312. type(c_ptr), value, intent(in) :: sarg
  313. type(c_funptr), value, intent(in) :: rcallback
  314. type(c_ptr), value, intent(in) :: rarg
  315. end function fstarpu_mpi_gather_detached
  316. ! int starpu_mpi_isend_detached_unlock_tag(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, MPI_Comm comm, starpu_tag_t tag);
  317. function fstarpu_mpi_isend_detached_unlock_tag (dh, dst, data_tag, mpi_comm, starpu_tag) bind(C)
  318. use iso_c_binding
  319. implicit none
  320. integer(c_int) :: fstarpu_mpi_isend_detached_unlock_tag
  321. type(c_ptr), value, intent(in) :: dh
  322. integer(c_int), value, intent(in) :: dst
  323. integer(c_int), value, intent(in) :: data_tag
  324. integer(c_int), value, intent(in) :: mpi_comm
  325. type(c_ptr), value, intent(in) :: starpu_tag
  326. end function fstarpu_mpi_isend_detached_unlock_tag
  327. ! int starpu_mpi_isend_detached_unlock_tag_prio(starpu_data_handle_t data_handle, int dest, starpu_mpi_tag_t data_tag, int prio, MPI_Comm comm, starpu_tag_t tag);
  328. function fstarpu_mpi_isend_detached_unlock_tag_prio (dh, dst, data_tag, prio, mpi_comm, starpu_tag) bind(C)
  329. use iso_c_binding
  330. implicit none
  331. integer(c_int) :: fstarpu_mpi_isend_detached_unlock_tag_prio
  332. type(c_ptr), value, intent(in) :: dh
  333. integer(c_int), value, intent(in) :: dst
  334. integer(c_int), value, intent(in) :: data_tag
  335. integer(c_int), value, intent(in) :: prio
  336. integer(c_int), value, intent(in) :: mpi_comm
  337. type(c_ptr), value, intent(in) :: starpu_tag
  338. end function fstarpu_mpi_isend_detached_unlock_tag_prio
  339. ! int starpu_mpi_irecv_detached_unlock_tag(starpu_data_handle_t data_handle, int source, starpu_mpi_tag_t data_tag, MPI_Comm comm, starpu_tag_t tag);
  340. function fstarpu_mpi_recv_detached_unlock_tag (dh, src, data_tag, mpi_comm, starpu_tag) bind(C)
  341. use iso_c_binding
  342. implicit none
  343. integer(c_int) :: fstarpu_mpi_recv_detached_unlock_tag
  344. type(c_ptr), value, intent(in) :: dh
  345. integer(c_int), value, intent(in) :: src
  346. integer(c_int), value, intent(in) :: data_tag
  347. integer(c_int), value, intent(in) :: mpi_comm
  348. type(c_ptr), value, intent(in) :: starpu_tag
  349. end function fstarpu_mpi_recv_detached_unlock_tag
  350. ! int starpu_mpi_isend_array_detached_unlock_tag(unsigned array_size, starpu_data_handle_t *data_handle, int *dest, int *data_tag, MPI_Comm *comm, starpu_tag_t tag);
  351. function fstarpu_mpi_isend_array_detached_unlock_tag (array_size, dhs, dsts, data_tags, mpi_comms, starpu_tag) &
  352. bind(C)
  353. use iso_c_binding
  354. implicit none
  355. integer(c_int) :: fstarpu_mpi_isend_array_detached_unlock_tag
  356. integer(c_int), value, intent(in) :: array_size
  357. type(c_ptr), intent(in) :: dhs(*)
  358. integer(c_int), intent(in) :: dsts(*)
  359. integer(c_int), intent(in) :: data_tags(*)
  360. integer(c_int), intent(in) :: mpi_comms(*)
  361. type(c_ptr), value, intent(in) :: starpu_tag
  362. end function fstarpu_mpi_isend_array_detached_unlock_tag
  363. ! int starpu_mpi_isend_array_detached_unlock_tag_prio(unsigned array_size, starpu_data_handle_t *data_handle, int *dest, int *data_tag, int *prio, MPI_Comm *comm, starpu_tag_t tag);
  364. function fstarpu_mpi_isend_array_detached_unlock_tag_prio (array_size, dhs, dsts, data_tags, prio, mpi_comms, &
  365. starpu_tag) bind(C)
  366. use iso_c_binding
  367. implicit none
  368. integer(c_int) :: fstarpu_mpi_isend_array_detached_unlock_tag_prio
  369. integer(c_int), value, intent(in) :: array_size
  370. type(c_ptr), intent(in) :: dhs(*)
  371. integer(c_int), intent(in) :: dsts(*)
  372. integer(c_int), intent(in) :: data_tags(*)
  373. integer(c_int), intent(in) :: prio(*)
  374. integer(c_int), intent(in) :: mpi_comms(*)
  375. type(c_ptr), value, intent(in) :: starpu_tag
  376. end function fstarpu_mpi_isend_array_detached_unlock_tag_prio
  377. ! int starpu_mpi_irecv_array_detached_unlock_tag(unsigned array_size, starpu_data_handle_t *data_handle, int *source, int *data_tag, MPI_Comm *comm, starpu_tag_t tag);
  378. function fstarpu_mpi_recv_array_detached_unlock_tag (array_size, dhs, srcs, data_tags, mpi_comms, starpu_tag) &
  379. bind(C)
  380. use iso_c_binding
  381. implicit none
  382. integer(c_int) :: fstarpu_mpi_recv_array_detached_unlock_tag
  383. integer(c_int), value, intent(in) :: array_size
  384. type(c_ptr), intent(in) :: dhs(*)
  385. integer(c_int), intent(in) :: srcs(*)
  386. integer(c_int), intent(in) :: data_tags(*)
  387. integer(c_int), intent(in) :: mpi_comms(*)
  388. type(c_ptr), value, intent(in) :: starpu_tag
  389. end function fstarpu_mpi_recv_array_detached_unlock_tag
  390. ! void starpu_mpi_comm_amounts_retrieve(size_t *comm_amounts);
  391. subroutine fstarpu_mpi_comm_amounts_retrieve (comm_amounts) bind(C,name="starpu_mpi_comm_amounts_retrieve")
  392. use iso_c_binding
  393. implicit none
  394. integer(c_size_t), intent(in) :: comm_amounts(*)
  395. end subroutine fstarpu_mpi_comm_amounts_retrieve
  396. ! void starpu_mpi_cache_flush(MPI_Comm comm, starpu_data_handle_t data_handle);
  397. subroutine fstarpu_mpi_cache_flush(mpi_comm,dh) bind(C)
  398. use iso_c_binding
  399. implicit none
  400. integer(c_int), value, intent(in) :: mpi_comm
  401. type(c_ptr), value, intent(in) :: dh
  402. end subroutine fstarpu_mpi_cache_flush
  403. ! void starpu_mpi_cache_flush_all_data(MPI_Comm comm);
  404. subroutine fstarpu_mpi_cache_flush_all_data(mpi_comm) bind(C)
  405. use iso_c_binding
  406. implicit none
  407. integer(c_int), value, intent(in) :: mpi_comm
  408. end subroutine fstarpu_mpi_cache_flush_all_data
  409. ! int starpu_mpi_comm_size(MPI_Comm comm, int *size);
  410. function fstarpu_mpi_comm_size(mpi_comm,sz) bind(C)
  411. use iso_c_binding
  412. implicit none
  413. integer(c_int), value, intent(in) :: mpi_comm
  414. integer(c_int), intent(out) :: sz
  415. integer(c_int) :: fstarpu_mpi_comm_size
  416. end function fstarpu_mpi_comm_size
  417. ! int starpu_mpi_comm_rank(MPI_Comm comm, int *rank);
  418. function fstarpu_mpi_comm_rank(mpi_comm,rank) bind(C)
  419. use iso_c_binding
  420. implicit none
  421. integer(c_int), value, intent(in) :: mpi_comm
  422. integer(c_int), intent(out) :: rank
  423. integer(c_int) :: fstarpu_mpi_comm_rank
  424. end function fstarpu_mpi_comm_rank
  425. ! int starpu_mpi_world_rank(void);
  426. function fstarpu_mpi_world_rank() bind(C,name="starpu_mpi_world_rank")
  427. use iso_c_binding
  428. implicit none
  429. integer(c_int) :: fstarpu_mpi_world_rank
  430. end function fstarpu_mpi_world_rank
  431. ! int starpu_mpi_world_size(void);
  432. function fstarpu_mpi_world_size() bind(C,name="starpu_mpi_world_size")
  433. use iso_c_binding
  434. implicit none
  435. integer(c_int) :: fstarpu_mpi_world_size
  436. end function fstarpu_mpi_world_size
  437. ! int starpu_mpi_world_size(void);
  438. function fstarpu_mpi_world_comm() bind(C)
  439. use iso_c_binding
  440. implicit none
  441. integer(c_int) :: fstarpu_mpi_world_comm
  442. end function fstarpu_mpi_world_comm
  443. ! int starpu_mpi_get_communication_tag(void);
  444. function fstarpu_mpi_get_communication_tag() bind(C,name="starpu_mpi_get_communication_tag")
  445. use iso_c_binding
  446. implicit none
  447. integer(c_int) :: fstarpu_mpi_get_communication_tag
  448. end function fstarpu_mpi_get_communication_tag
  449. ! void starpu_mpi_set_communication_tag(int tag);
  450. subroutine fstarpu_mpi_set_communication_tag(tag) bind(C,name="starpu_mpi_set_communication_tag")
  451. use iso_c_binding
  452. implicit none
  453. integer(c_int), value, intent(in) :: tag
  454. end subroutine fstarpu_mpi_set_communication_tag
  455. ! void starpu_mpi_data_register_comm(starpu_data_handle_t data_handle, int tag, int rank, MPI_Comm comm);
  456. subroutine fstarpu_mpi_data_register_comm(dh,tag,rank,mpi_comm) bind(C)
  457. use iso_c_binding
  458. implicit none
  459. type(c_ptr), value, intent(in) :: dh
  460. integer(c_int), value, intent(in) :: tag
  461. integer(c_int), value, intent(in) :: rank
  462. integer(c_int), value, intent(in) :: mpi_comm
  463. end subroutine fstarpu_mpi_data_register_comm
  464. ! #define starpu_mpi_data_register(data_handle, tag, rank) starpu_mpi_data_register_comm(data_handle, tag, rank, MPI_COMM_WORLD)
  465. subroutine fstarpu_mpi_data_register(dh,tag,rank) bind(C)
  466. use iso_c_binding
  467. implicit none
  468. type(c_ptr), value, intent(in) :: dh
  469. integer(c_int), value, intent(in) :: tag
  470. integer(c_int), value, intent(in) :: rank
  471. end subroutine fstarpu_mpi_data_register
  472. ! void starpu_mpi_data_set_rank_comm(starpu_data_handle_t handle, int rank, MPI_Comm comm);
  473. subroutine fstarpu_mpi_data_set_rank_comm(dh,rank,mpi_comm) bind(C)
  474. use iso_c_binding
  475. implicit none
  476. type(c_ptr), value, intent(in) :: dh
  477. integer(c_int), value, intent(in) :: rank
  478. integer(c_int), value, intent(in) :: mpi_comm
  479. end subroutine fstarpu_mpi_data_set_rank_comm
  480. ! #define starpu_mpi_data_set_rank(handle, rank) starpu_mpi_data_set_rank_comm(handle, rank, MPI_COMM_WORLD)
  481. subroutine fstarpu_mpi_data_set_rank(dh,rank) bind(C)
  482. use iso_c_binding
  483. implicit none
  484. type(c_ptr), value, intent(in) :: dh
  485. integer(c_int), value, intent(in) :: rank
  486. end subroutine fstarpu_mpi_data_set_rank
  487. ! void starpu_mpi_data_set_tag(starpu_data_handle_t handle, int tag);
  488. subroutine fstarpu_mpi_data_set_tag(dh,tag) bind(C,name="starpu_mpi_data_set_tag")
  489. use iso_c_binding
  490. implicit none
  491. type(c_ptr), value, intent(in) :: dh
  492. integer(c_int), value, intent(in) :: tag
  493. end subroutine fstarpu_mpi_data_set_tag
  494. ! int starpu_mpi_data_get_rank(starpu_data_handle_t handle);
  495. function fstarpu_mpi_data_get_rank(dh) bind(C,name="starpu_mpi_data_get_rank")
  496. use iso_c_binding
  497. implicit none
  498. integer(c_int) :: fstarpu_mpi_data_get_rank
  499. type(c_ptr), value, intent(in) :: dh
  500. end function fstarpu_mpi_data_get_rank
  501. ! int starpu_mpi_data_get_tag(starpu_data_handle_t handle);
  502. function fstarpu_mpi_data_get_tag(dh) bind(C,name="starpu_mpi_data_get_tag")
  503. use iso_c_binding
  504. implicit none
  505. integer(c_int) :: fstarpu_mpi_data_get_tag
  506. type(c_ptr), value, intent(in) :: dh
  507. end function fstarpu_mpi_data_get_tag
  508. ! void starpu_mpi_data_migrate(MPI_Comm comm, starpu_data_handle_t handle, int rank);
  509. subroutine fstarpu_mpi_data_migrate(mpi_comm,dh,rank) bind(C)
  510. use iso_c_binding
  511. implicit none
  512. integer(c_int), value, intent(in) :: mpi_comm
  513. type(c_ptr), value, intent(in) :: dh
  514. integer(c_int), value, intent(in) :: rank
  515. end subroutine fstarpu_mpi_data_migrate
  516. ! #define STARPU_MPI_NODE_SELECTION_CURRENT_POLICY -1
  517. ! #define STARPU_MPI_NODE_SELECTION_MOST_R_DATA 0
  518. ! int starpu_mpi_node_selection_register_policy(starpu_mpi_select_node_policy_func_t policy_func);
  519. function fstarpu_mpi_node_selection_register_policy(policy_func) &
  520. bind(C,name="starpu_mpi_node_selection_register_policy")
  521. use iso_c_binding
  522. implicit none
  523. integer(c_int) :: fstarpu_mpi_node_selection_register_policy
  524. type(c_funptr), value, intent(in) :: policy_func
  525. end function fstarpu_mpi_node_selection_register_policy
  526. ! int starpu_mpi_node_selection_unregister_policy(int policy);
  527. function fstarpu_mpi_node_selection_unregister_policy(policy) &
  528. bind(C,name="starpu_mpi_node_selection_unregister_policy")
  529. use iso_c_binding
  530. implicit none
  531. integer(c_int) :: fstarpu_mpi_node_selection_unregister_policy
  532. type(c_ptr), value, intent(in) :: policy
  533. end function fstarpu_mpi_node_selection_unregister_policy
  534. ! int starpu_mpi_node_selection_get_current_policy();
  535. function fstarpu_mpi_data_selection_get_current_policy() &
  536. bind(C,name="starpu_mpi_data_selection_get_current_policy")
  537. use iso_c_binding
  538. implicit none
  539. integer(c_int) :: fstarpu_mpi_data_selection_get_current_policy
  540. end function fstarpu_mpi_data_selection_get_current_policy
  541. ! int starpu_mpi_node_selection_set_current_policy(int policy);
  542. function fstarpu_mpi_data_selection_set_current_policy(policy) &
  543. bind(C,name="starpu_mpi_data_selection_set_current_policy")
  544. use iso_c_binding
  545. implicit none
  546. integer(c_int) :: fstarpu_mpi_data_selection_set_current_policy
  547. type(c_ptr), value, intent(in) :: policy
  548. end function fstarpu_mpi_data_selection_set_current_policy
  549. ! int starpu_mpi_cache_is_enabled();
  550. function fstarpu_mpi_cache_is_enabled() bind(C,name="starpu_mpi_cache_is_enabled")
  551. use iso_c_binding
  552. implicit none
  553. integer(c_int) :: fstarpu_mpi_cache_is_enabled
  554. end function fstarpu_mpi_cache_is_enabled
  555. ! int starpu_mpi_cache_set(int enabled);
  556. function fstarpu_mpi_cache_set(enabled) bind(C,name="starpu_mpi_cache_set")
  557. use iso_c_binding
  558. implicit none
  559. integer(c_int) :: fstarpu_mpi_cache_set
  560. integer(c_int), value, intent(in) :: enabled
  561. end function fstarpu_mpi_cache_set
  562. ! int starpu_mpi_wait_for_all(MPI_Comm comm);
  563. function fstarpu_mpi_wait_for_all (mpi_comm) bind(C)
  564. use iso_c_binding
  565. implicit none
  566. integer(c_int) :: fstarpu_mpi_wait_for_all
  567. integer(c_int), value, intent(in) :: mpi_comm
  568. end function fstarpu_mpi_wait_for_all
  569. ! int starpu_mpi_datatype_register(starpu_data_handle_t handle, starpu_mpi_datatype_allocate_func_t allocate_datatype_func, starpu_mpi_datatype_free_func_t free_datatype_func);
  570. function fstarpu_mpi_datatype_register(dh, alloc_func, free_func) bind(C,name="starpu_mpi_datatype_register")
  571. use iso_c_binding
  572. implicit none
  573. integer(c_int) :: fstarpu_mpi_datatype_register
  574. type(c_ptr), value, intent(in) :: dh
  575. type(c_funptr), value, intent(in) :: alloc_func
  576. type(c_funptr), value, intent(in) :: free_func
  577. end function fstarpu_mpi_datatype_register
  578. ! int starpu_mpi_datatype_unregister(starpu_data_handle_t handle);
  579. function fstarpu_mpi_datatype_unregister(dh) bind(C,name="starpu_mpi_datatype_unregister")
  580. use iso_c_binding
  581. implicit none
  582. integer(c_int) :: fstarpu_mpi_datatype_unregister
  583. type(c_ptr), value, intent(in) :: dh
  584. end function fstarpu_mpi_datatype_unregister
  585. function fstarpu_mpi_req_alloc() bind(C)
  586. use iso_c_binding
  587. implicit none
  588. type(c_ptr) :: fstarpu_mpi_req_alloc
  589. end function fstarpu_mpi_req_alloc
  590. subroutine fstarpu_mpi_req_free(req) bind(C)
  591. use iso_c_binding
  592. implicit none
  593. type(c_ptr),value,intent(in) :: req
  594. end subroutine fstarpu_mpi_req_free
  595. function fstarpu_mpi_status_alloc() bind(C)
  596. use iso_c_binding
  597. implicit none
  598. type(c_ptr) :: fstarpu_mpi_status_alloc
  599. end function fstarpu_mpi_status_alloc
  600. subroutine fstarpu_mpi_status_free(st) bind(C)
  601. use iso_c_binding
  602. implicit none
  603. type(c_ptr),value,intent(in) :: st
  604. end subroutine fstarpu_mpi_status_free
  605. end interface
  606. contains
  607. function fstarpu_mpi_init (initialize_mpi,mpi_comm) bind(C)
  608. use iso_c_binding
  609. implicit none
  610. integer(c_int) :: fstarpu_mpi_init
  611. integer(c_int), intent(in) :: initialize_mpi
  612. integer(c_int), optional, intent(in) :: mpi_comm
  613. type(c_ptr) :: argcv
  614. integer(c_int) :: fargc,i,farg_len
  615. character(len=1) :: farg_1
  616. character(len=:), allocatable :: farg
  617. integer(c_int) :: mpi_comm_present, mpi_comm_or_0
  618. integer(c_int) :: ret
  619. interface
  620. function fstarpu_mpi_argcv_alloc(argc, initialize_mpi, comm_present, comm) bind(C)
  621. use iso_c_binding
  622. implicit none
  623. type(c_ptr) :: fstarpu_mpi_argcv_alloc
  624. integer(c_int),value,intent(in) :: argc
  625. integer(c_int),value,intent(in) :: initialize_mpi
  626. integer(c_int),value,intent(in) :: comm_present
  627. integer(c_int),value,intent(in) :: comm
  628. end function fstarpu_mpi_argcv_alloc
  629. subroutine fstarpu_mpi_argcv_set_arg(argcv, i, l, s) bind(C)
  630. use iso_c_binding
  631. implicit none
  632. type(c_ptr),value,intent(in) :: argcv
  633. integer(c_int),value,intent(in) :: i
  634. integer(c_int),value,intent(in) :: l
  635. character(c_char),intent(in) :: s
  636. end subroutine fstarpu_mpi_argcv_set_arg
  637. subroutine fstarpu_mpi_argcv_free(argcv) bind(C)
  638. use iso_c_binding
  639. implicit none
  640. type(c_ptr),value,intent(in) :: argcv
  641. end subroutine fstarpu_mpi_argcv_free
  642. function fstarpu_mpi_init_c(argcv) bind(C)
  643. use iso_c_binding
  644. implicit none
  645. integer(c_int) :: fstarpu_mpi_init_c
  646. type(c_ptr),value,intent(in) :: argcv
  647. end function fstarpu_mpi_init_c
  648. end interface
  649. fargc = command_argument_count()
  650. write(*,*) "fargc",fargc
  651. if (present(mpi_comm)) then
  652. mpi_comm_present = 1
  653. mpi_comm_or_0 = mpi_comm
  654. else
  655. mpi_comm_present = 0
  656. mpi_comm_or_0 = 0
  657. end if
  658. write(*,*) "initialize_mpi",initialize_mpi
  659. write(*,*) "mpi_comm_present",mpi_comm_present
  660. argcv = fstarpu_mpi_argcv_alloc(fargc, initialize_mpi, mpi_comm_present, mpi_comm_or_0)
  661. do i=0,fargc-1
  662. call get_command_argument(i, farg_1, farg_len)
  663. allocate (character(len=farg_len) :: farg)
  664. call get_command_argument(i, farg)
  665. call fstarpu_mpi_argcv_set_arg(argcv, i, farg_len, farg)
  666. deallocate (farg)
  667. end do
  668. ret = fstarpu_mpi_init_c(argcv)
  669. call fstarpu_mpi_argcv_free(argcv)
  670. fstarpu_mpi_init = ret
  671. end function fstarpu_mpi_init
  672. end module fstarpu_mpi_mod