fstarpu_mpi_mod.f90 45 KB

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