starpu_mpi_insert_task.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2011, 2012 Centre National de la Recherche Scientifique
  4. * Copyright (C) 2011-2012 Université de Bordeaux 1
  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. #include <stdarg.h>
  18. #include <mpi.h>
  19. #include <starpu.h>
  20. #include <starpu_data.h>
  21. #include <common/utils.h>
  22. #include <common/uthash.h>
  23. #include <util/starpu_insert_task_utils.h>
  24. #include <datawizard/coherency.h>
  25. //#define STARPU_MPI_VERBOSE 1
  26. #include <starpu_mpi_private.h>
  27. #ifdef STARPU_MPI_CACHE
  28. /* Whether we are allowed to keep copies of remote data. */
  29. struct _starpu_data_entry
  30. {
  31. UT_hash_handle hh;
  32. void *data;
  33. };
  34. struct _starpu_data_entry **sent_data = NULL;
  35. struct _starpu_data_entry **received_data = NULL;
  36. #endif /* STARPU_MPI_CACHE */
  37. void _starpu_mpi_tables_init(MPI_Comm comm)
  38. {
  39. #ifdef STARPU_MPI_CACHE
  40. int nb_nodes;
  41. int i;
  42. MPI_Comm_size(comm, &nb_nodes);
  43. _STARPU_MPI_DEBUG("Initialising htable for cache\n");
  44. sent_data = malloc(nb_nodes * sizeof(struct _starpu_data_entry *));
  45. for(i=0 ; i<nb_nodes ; i++) sent_data[i] = NULL;
  46. received_data = malloc(nb_nodes * sizeof(struct _starpu_data_entry *));
  47. for(i=0 ; i<nb_nodes ; i++) received_data[i] = NULL;
  48. #else
  49. if (!getenv("STARPU_SILENT")) fprintf(stderr,"Warning: StarPU was configured with --disable-mpi-cache\n");
  50. #endif /* STARPU_MPI_CACHE */
  51. }
  52. void _starpu_mpi_tables_free(int world_size)
  53. {
  54. #ifdef STARPU_MPI_CACHE
  55. int i;
  56. _STARPU_MPI_DEBUG("Clearing htable for cache\n");
  57. for(i=0 ; i<world_size ; i++)
  58. {
  59. struct _starpu_data_entry *entry, *tmp;
  60. HASH_ITER(hh, sent_data[i], entry, tmp)
  61. {
  62. HASH_DEL(sent_data[i], entry);
  63. free(entry);
  64. }
  65. HASH_ITER(hh, received_data[i], entry, tmp)
  66. {
  67. HASH_DEL(received_data[i], entry);
  68. free(entry);
  69. }
  70. }
  71. free(sent_data);
  72. free(received_data);
  73. #endif
  74. }
  75. static
  76. int _starpu_mpi_find_executee_node(starpu_data_handle_t data, enum starpu_access_mode mode, int me, int *do_execute, int *inconsistent_execute, int *dest, size_t *size_on_nodes)
  77. {
  78. if (data && mode & STARPU_R) {
  79. struct starpu_data_interface_ops *ops;
  80. int rank = starpu_data_get_rank(data);
  81. ops = data->ops;
  82. size_on_nodes[rank] += ops->get_size(data);
  83. }
  84. if (mode & STARPU_W) {
  85. if (!data) {
  86. /* We don't have anything allocated for this.
  87. * The application knows we won't do anything
  88. * about this task */
  89. /* Yes, the app could actually not call
  90. * insert_task at all itself, this is just a
  91. * safeguard. */
  92. _STARPU_MPI_DEBUG("oh oh\n");
  93. _STARPU_MPI_LOG_OUT();
  94. return -EINVAL;
  95. }
  96. int mpi_rank = starpu_data_get_rank(data);
  97. if (mpi_rank == me) {
  98. if (*do_execute == 0) {
  99. *inconsistent_execute = 1;
  100. }
  101. else {
  102. *do_execute = 1;
  103. }
  104. }
  105. else if (mpi_rank != -1) {
  106. if (*do_execute == 1) {
  107. *inconsistent_execute = 1;
  108. }
  109. else {
  110. *do_execute = 0;
  111. *dest = mpi_rank;
  112. /* That's the rank which needs the data to be sent to */
  113. }
  114. }
  115. else {
  116. _STARPU_ERROR("rank invalid\n");
  117. }
  118. }
  119. return 0;
  120. }
  121. static
  122. void *_starpu_mpi_already_received(starpu_data_handle_t data, int mpi_rank)
  123. {
  124. #ifdef STARPU_MPI_CACHE
  125. struct _starpu_data_entry *already_received;
  126. HASH_FIND_PTR(received_data[mpi_rank], &data, already_received);
  127. if (already_received == NULL) {
  128. struct _starpu_data_entry *entry = (struct _starpu_data_entry *)malloc(sizeof(*entry));
  129. entry->data = data;
  130. HASH_ADD_PTR(received_data[mpi_rank], data, entry);
  131. }
  132. else {
  133. _STARPU_MPI_DEBUG("Do not receive data %p from node %d as it is already available\n", data, mpi_rank);
  134. }
  135. return already_received;
  136. #else
  137. return NULL;
  138. #endif
  139. }
  140. static
  141. void *_starpu_mpi_already_sent(starpu_data_handle_t data, int dest)
  142. {
  143. #ifdef STARPU_MPI_CACHE
  144. struct _starpu_data_entry *already_sent;
  145. HASH_FIND_PTR(sent_data[dest], &data, already_sent);
  146. if (already_sent == NULL) {
  147. struct _starpu_data_entry *entry = (struct _starpu_data_entry *)malloc(sizeof(*entry));
  148. entry->data = data;
  149. HASH_ADD_PTR(sent_data[dest], data, entry);
  150. _STARPU_MPI_DEBUG("Noting that data %p has already been sent to %d\n", data, dest);
  151. }
  152. else {
  153. _STARPU_MPI_DEBUG("Do not send data %p to node %d as it has already been sent\n", data, dest);
  154. }
  155. return already_sent;
  156. #else
  157. return NULL;
  158. #endif
  159. }
  160. static
  161. void _starpu_mpi_exchange_data_before_execution(starpu_data_handle_t data, enum starpu_access_mode mode, int me, int dest, int do_execute, MPI_Comm comm)
  162. {
  163. if (data && mode & STARPU_R) {
  164. int mpi_rank = starpu_data_get_rank(data);
  165. int mpi_tag = starpu_data_get_tag(data);
  166. if(mpi_rank == -1) {
  167. fprintf(stderr,"StarPU needs to be told the MPI rank of this data, using starpu_data_set_rank\n");
  168. STARPU_ABORT();
  169. }
  170. if(mpi_tag == -1) {
  171. fprintf(stderr,"StarPU needs to be told the MPI tag of this data, using starpu_data_set_tag\n");
  172. STARPU_ABORT();
  173. }
  174. /* The task needs to read this data */
  175. if (do_execute && mpi_rank != me && mpi_rank != -1) {
  176. /* I will have to execute but I don't have the data, receive */
  177. void *already_received = _starpu_mpi_already_received(data, mpi_rank);
  178. if (already_received == NULL)
  179. {
  180. _STARPU_MPI_DEBUG("Receive data %p from %d\n", data, mpi_rank);
  181. starpu_mpi_irecv_detached(data, mpi_rank, mpi_tag, comm, NULL, NULL);
  182. }
  183. }
  184. if (!do_execute && mpi_rank == me) {
  185. /* Somebody else will execute it, and I have the data, send it. */
  186. void *already_sent = _starpu_mpi_already_sent(data, dest);
  187. if (already_sent == NULL)
  188. {
  189. _STARPU_MPI_DEBUG("Send data %p to %d\n", data, dest);
  190. starpu_mpi_isend_detached(data, dest, mpi_tag, comm, NULL, NULL);
  191. }
  192. }
  193. }
  194. }
  195. static
  196. void _starpu_mpi_exchange_data_after_execution(starpu_data_handle_t data, enum starpu_access_mode mode, int me, int xrank, int dest, int do_execute, MPI_Comm comm)
  197. {
  198. if (mode & STARPU_W) {
  199. int mpi_rank = starpu_data_get_rank(data);
  200. int mpi_tag = starpu_data_get_tag(data);
  201. if(mpi_rank == -1) {
  202. fprintf(stderr,"StarPU needs to be told the MPI rank of this data, using starpu_data_set_rank\n");
  203. STARPU_ABORT();
  204. }
  205. if(mpi_tag == -1) {
  206. fprintf(stderr,"StarPU needs to be told the MPI tag of this data, using starpu_data_set_tag\n");
  207. STARPU_ABORT();
  208. }
  209. if (mpi_rank == me) {
  210. if (xrank != -1 && me != xrank) {
  211. _STARPU_MPI_DEBUG("Receive data %p back from the task %d which executed the codelet ...\n", data, dest);
  212. starpu_mpi_irecv_detached(data, dest, mpi_tag, comm, NULL, NULL);
  213. }
  214. }
  215. else if (do_execute) {
  216. _STARPU_MPI_DEBUG("Send data %p back to its owner %d...\n", data, mpi_rank);
  217. starpu_mpi_isend_detached(data, mpi_rank, mpi_tag, comm, NULL, NULL);
  218. }
  219. }
  220. }
  221. void _starpu_mpi_clear_data_after_execution(starpu_data_handle_t data, enum starpu_access_mode mode, int me, int do_execute, MPI_Comm comm)
  222. {
  223. #ifdef STARPU_MPI_CACHE
  224. if (mode & STARPU_W) {
  225. if (do_execute) {
  226. /* Note that all copies I've sent to neighbours are now invalid */
  227. int n, size;
  228. MPI_Comm_size(comm, &size);
  229. for(n=0 ; n<size ; n++) {
  230. struct _starpu_data_entry *already_sent;
  231. HASH_FIND_PTR(sent_data[n], &data, already_sent);
  232. if (already_sent) {
  233. _STARPU_MPI_DEBUG("Clearing send cache for data %p\n", data);
  234. HASH_DEL(sent_data[n], already_sent);
  235. }
  236. }
  237. }
  238. else {
  239. int mpi_rank = starpu_data_get_rank(data);
  240. struct _starpu_data_entry *already_received;
  241. HASH_FIND_PTR(received_data[mpi_rank], &data, already_received);
  242. if (already_received) {
  243. /* Somebody else will write to the data, so discard our cached copy if any */
  244. /* TODO: starpu_mpi could just remember itself. */
  245. _STARPU_MPI_DEBUG("Clearing receive cache for data %p\n", data);
  246. HASH_DEL(received_data[mpi_rank], already_received);
  247. starpu_data_invalidate_submit(data);
  248. }
  249. }
  250. }
  251. #else
  252. /* We allocated a temporary buffer for the received data, now drop it */
  253. if ((mode & STARPU_R) && do_execute) {
  254. int mpi_rank = starpu_data_get_rank(data);
  255. if (mpi_rank != me && mpi_rank != -1) {
  256. starpu_data_invalidate_submit(data);
  257. }
  258. }
  259. #endif
  260. }
  261. int starpu_mpi_insert_task(MPI_Comm comm, struct starpu_codelet *codelet, ...)
  262. {
  263. int arg_type;
  264. va_list varg_list;
  265. int me, do_execute, xrank, nb_nodes;
  266. size_t *size_on_nodes;
  267. size_t arg_buffer_size = 0;
  268. char *arg_buffer;
  269. int dest=0, inconsistent_execute;
  270. int current_data = 0;
  271. _STARPU_MPI_LOG_IN();
  272. MPI_Comm_rank(comm, &me);
  273. MPI_Comm_size(comm, &nb_nodes);
  274. size_on_nodes = (size_t *)calloc(1, nb_nodes * sizeof(size_t));
  275. /* Get the number of buffers and the size of the arguments */
  276. va_start(varg_list, codelet);
  277. arg_buffer_size = _starpu_insert_task_get_arg_size(varg_list);
  278. va_start(varg_list, codelet);
  279. _starpu_codelet_pack_args(arg_buffer_size, &arg_buffer, varg_list);
  280. /* Find out whether we are to execute the data because we own the data to be written to. */
  281. inconsistent_execute = 0;
  282. do_execute = -1;
  283. xrank = -1;
  284. va_start(varg_list, codelet);
  285. while ((arg_type = va_arg(varg_list, int)) != 0) {
  286. if (arg_type==STARPU_EXECUTE_ON_NODE) {
  287. xrank = va_arg(varg_list, int);
  288. _STARPU_MPI_DEBUG("Executing on node %d\n", xrank);
  289. do_execute = 1;
  290. }
  291. else if (arg_type==STARPU_EXECUTE_ON_DATA) {
  292. starpu_data_handle_t data = va_arg(varg_list, starpu_data_handle_t);
  293. xrank = starpu_data_get_rank(data);
  294. _STARPU_MPI_DEBUG("Executing on data node %d\n", xrank);
  295. STARPU_ASSERT(xrank <= nb_nodes);
  296. do_execute = 1;
  297. }
  298. else if (arg_type==STARPU_R || arg_type==STARPU_W || arg_type==STARPU_RW || arg_type==STARPU_SCRATCH || arg_type==STARPU_REDUX) {
  299. starpu_data_handle_t data = va_arg(varg_list, starpu_data_handle_t);
  300. enum starpu_access_mode mode = (enum starpu_access_mode) arg_type;
  301. int ret = _starpu_mpi_find_executee_node(data, mode, me, &do_execute, &inconsistent_execute, &dest, size_on_nodes);
  302. if (ret == -EINVAL)
  303. {
  304. free(size_on_nodes);
  305. return ret;
  306. }
  307. current_data ++;
  308. }
  309. else if (arg_type == STARPU_DATA_ARRAY)
  310. {
  311. starpu_data_handle_t *datas = va_arg(varg_list, starpu_data_handle_t *);
  312. int nb_handles = va_arg(varg_list, int);
  313. int i;
  314. for(i=0 ; i<nb_handles ; i++)
  315. {
  316. enum starpu_access_mode mode = codelet->modes[current_data];
  317. int ret = _starpu_mpi_find_executee_node(datas[i], mode, me, &do_execute, &inconsistent_execute, &dest, size_on_nodes);
  318. if (ret == -EINVAL)
  319. {
  320. free(size_on_nodes);
  321. return ret;
  322. }
  323. current_data ++;
  324. }
  325. }
  326. else if (arg_type==STARPU_VALUE) {
  327. va_arg(varg_list, void *);
  328. va_arg(varg_list, size_t);
  329. }
  330. else if (arg_type==STARPU_CALLBACK) {
  331. va_arg(varg_list, void (*)(void *));
  332. }
  333. else if (arg_type==STARPU_CALLBACK_WITH_ARG) {
  334. va_arg(varg_list, void (*)(void *));
  335. va_arg(varg_list, void *);
  336. }
  337. else if (arg_type==STARPU_CALLBACK_ARG) {
  338. va_arg(varg_list, void *);
  339. }
  340. else if (arg_type==STARPU_PRIORITY) {
  341. va_arg(varg_list, int);
  342. }
  343. else if (arg_type==STARPU_EXECUTE_ON_NODE) {
  344. va_arg(varg_list, int);
  345. }
  346. else if (arg_type==STARPU_EXECUTE_ON_DATA) {
  347. va_arg(varg_list, int);
  348. }
  349. }
  350. va_end(varg_list);
  351. if (do_execute == -1) {
  352. int i;
  353. size_t max_size = 0;
  354. for(i=0 ; i<nb_nodes ; i++) {
  355. if (size_on_nodes[i] > max_size)
  356. {
  357. max_size = size_on_nodes[i];
  358. xrank = i;
  359. }
  360. }
  361. free(size_on_nodes);
  362. if (xrank != -1) {
  363. _STARPU_MPI_DEBUG("Node %d is having the most R data\n", xrank);
  364. do_execute = 1;
  365. }
  366. }
  367. STARPU_ASSERT_MSG(do_execute != -1, "StarPU needs to see a W or a REDUX data which will tell it where to execute the task");
  368. if (inconsistent_execute == 1) {
  369. if (xrank == -1) {
  370. _STARPU_MPI_DEBUG("Different tasks are owning W data. Needs to specify which one is to execute the codelet, using STARPU_EXECUTE_ON_NODE or STARPU_EXECUTE_ON_DATA\n");
  371. free(size_on_nodes);
  372. return -EINVAL;
  373. }
  374. else {
  375. do_execute = (me == xrank);
  376. dest = xrank;
  377. }
  378. }
  379. else if (xrank != -1) {
  380. do_execute = (me == xrank);
  381. dest = xrank;
  382. }
  383. /* Send and receive data as requested */
  384. va_start(varg_list, codelet);
  385. current_data = 0;
  386. while ((arg_type = va_arg(varg_list, int)) != 0) {
  387. if (arg_type==STARPU_R || arg_type==STARPU_W || arg_type==STARPU_RW || arg_type==STARPU_SCRATCH || arg_type==STARPU_REDUX) {
  388. starpu_data_handle_t data = va_arg(varg_list, starpu_data_handle_t);
  389. enum starpu_access_mode mode = (enum starpu_access_mode) arg_type;
  390. _starpu_mpi_exchange_data_before_execution(data, mode, me, dest, do_execute, comm);
  391. current_data ++;
  392. }
  393. else if (arg_type == STARPU_DATA_ARRAY)
  394. {
  395. starpu_data_handle_t *datas = va_arg(varg_list, starpu_data_handle_t *);
  396. int nb_handles = va_arg(varg_list, int);
  397. int i;
  398. for(i=0 ; i<nb_handles ; i++)
  399. {
  400. _starpu_mpi_exchange_data_before_execution(datas[i], codelet->modes[current_data], me, dest, do_execute, comm);
  401. current_data++;
  402. }
  403. }
  404. else if (arg_type==STARPU_VALUE) {
  405. va_arg(varg_list, void *);
  406. va_arg(varg_list, size_t);
  407. }
  408. else if (arg_type==STARPU_CALLBACK) {
  409. va_arg(varg_list, void (*)(void *));
  410. }
  411. else if (arg_type==STARPU_CALLBACK_WITH_ARG) {
  412. va_arg(varg_list, void (*)(void *));
  413. va_arg(varg_list, void *);
  414. }
  415. else if (arg_type==STARPU_CALLBACK_ARG) {
  416. va_arg(varg_list, void *);
  417. }
  418. else if (arg_type==STARPU_PRIORITY) {
  419. va_arg(varg_list, int);
  420. }
  421. else if (arg_type==STARPU_EXECUTE_ON_NODE) {
  422. va_arg(varg_list, int);
  423. }
  424. else if (arg_type==STARPU_EXECUTE_ON_DATA) {
  425. va_arg(varg_list, starpu_data_handle_t);
  426. }
  427. }
  428. va_end(varg_list);
  429. if (do_execute) {
  430. _STARPU_MPI_DEBUG("Execution of the codelet %p (%s)\n", codelet, codelet->name);
  431. va_start(varg_list, codelet);
  432. struct starpu_task *task = starpu_task_create();
  433. int ret = _starpu_insert_task_create_and_submit(arg_buffer, arg_buffer_size, codelet, &task, varg_list);
  434. _STARPU_MPI_DEBUG("ret: %d\n", ret);
  435. STARPU_ASSERT(ret==0);
  436. }
  437. if (inconsistent_execute) {
  438. va_start(varg_list, codelet);
  439. current_data = 0;
  440. while ((arg_type = va_arg(varg_list, int)) != 0) {
  441. if (arg_type==STARPU_R || arg_type==STARPU_W || arg_type==STARPU_RW || arg_type==STARPU_SCRATCH || arg_type==STARPU_REDUX) {
  442. starpu_data_handle_t data = va_arg(varg_list, starpu_data_handle_t);
  443. enum starpu_access_mode mode = (enum starpu_access_mode) arg_type;
  444. _starpu_mpi_exchange_data_after_execution(data, mode, me, xrank, dest, do_execute, comm);
  445. current_data++;
  446. }
  447. else if (arg_type == STARPU_DATA_ARRAY)
  448. {
  449. starpu_data_handle_t *datas = va_arg(varg_list, starpu_data_handle_t *);
  450. int nb_handles = va_arg(varg_list, int);
  451. int i;
  452. for(i=0 ; i<nb_handles ; i++)
  453. {
  454. _starpu_mpi_exchange_data_after_execution(datas[i], codelet->modes[current_data], me, xrank, dest, do_execute, comm);
  455. current_data++;
  456. }
  457. }
  458. else if (arg_type==STARPU_VALUE) {
  459. va_arg(varg_list, void *);
  460. va_arg(varg_list, size_t);
  461. }
  462. else if (arg_type==STARPU_CALLBACK) {
  463. va_arg(varg_list, void (*)(void *));
  464. }
  465. else if (arg_type==STARPU_CALLBACK_WITH_ARG) {
  466. va_arg(varg_list, void (*)(void *));
  467. va_arg(varg_list, void *);
  468. }
  469. else if (arg_type==STARPU_CALLBACK_ARG) {
  470. va_arg(varg_list, void *);
  471. }
  472. else if (arg_type==STARPU_PRIORITY) {
  473. va_arg(varg_list, int);
  474. }
  475. else if (arg_type==STARPU_EXECUTE_ON_NODE) {
  476. va_arg(varg_list, int);
  477. }
  478. else if (arg_type==STARPU_EXECUTE_ON_DATA) {
  479. va_arg(varg_list, starpu_data_handle_t);
  480. }
  481. }
  482. va_end(varg_list);
  483. }
  484. va_start(varg_list, codelet);
  485. current_data = 0;
  486. while ((arg_type = va_arg(varg_list, int)) != 0) {
  487. if (arg_type==STARPU_R || arg_type==STARPU_W || arg_type==STARPU_RW || arg_type == STARPU_SCRATCH || arg_type == STARPU_REDUX) {
  488. starpu_data_handle_t data = va_arg(varg_list, starpu_data_handle_t);
  489. enum starpu_access_mode mode = (enum starpu_access_mode) arg_type;
  490. _starpu_mpi_clear_data_after_execution(data, mode, me, do_execute, comm);
  491. current_data++;
  492. }
  493. else if (arg_type == STARPU_DATA_ARRAY)
  494. {
  495. starpu_data_handle_t *datas = va_arg(varg_list, starpu_data_handle_t *);
  496. int nb_handles = va_arg(varg_list, int);
  497. int i;
  498. for(i=0 ; i<nb_handles ; i++)
  499. {
  500. _starpu_mpi_clear_data_after_execution(datas[i], codelet->modes[current_data], me, do_execute, comm);
  501. current_data++;
  502. }
  503. }
  504. else if (arg_type==STARPU_VALUE) {
  505. va_arg(varg_list, void *);
  506. va_arg(varg_list, size_t);
  507. }
  508. else if (arg_type==STARPU_CALLBACK) {
  509. va_arg(varg_list, void (*)(void *));
  510. }
  511. else if (arg_type==STARPU_CALLBACK_WITH_ARG) {
  512. va_arg(varg_list, void (*)(void *));
  513. va_arg(varg_list, void *);
  514. }
  515. else if (arg_type==STARPU_CALLBACK_ARG) {
  516. va_arg(varg_list, void *);
  517. }
  518. else if (arg_type==STARPU_PRIORITY) {
  519. va_arg(varg_list, int);
  520. }
  521. else if (arg_type==STARPU_EXECUTE_ON_NODE) {
  522. va_arg(varg_list, int);
  523. }
  524. else if (arg_type==STARPU_EXECUTE_ON_DATA) {
  525. va_arg(varg_list, starpu_data_handle_t);
  526. }
  527. }
  528. va_end(varg_list);
  529. _STARPU_MPI_LOG_OUT();
  530. return 0;
  531. }
  532. void starpu_mpi_get_data_on_node_detached(MPI_Comm comm, starpu_data_handle_t data_handle, int node, void (*callback)(void*), void *arg)
  533. {
  534. int me, rank, tag;
  535. rank = starpu_data_get_rank(data_handle);
  536. tag = starpu_data_get_tag(data_handle);
  537. if(rank == -1) {
  538. fprintf(stderr,"StarPU needs to be told the MPI rank of this data, using starpu_data_set_rank\n");
  539. STARPU_ABORT();
  540. }
  541. if(tag == -1) {
  542. fprintf(stderr,"StarPU needs to be told the MPI tag of this data, using starpu_data_set_tag\n");
  543. STARPU_ABORT();
  544. }
  545. MPI_Comm_rank(comm, &me);
  546. if (node == rank) return;
  547. if (me == node)
  548. {
  549. starpu_mpi_irecv_detached(data_handle, rank, tag, comm, callback, arg);
  550. }
  551. else if (me == rank)
  552. {
  553. starpu_mpi_isend_detached(data_handle, node, tag, comm, NULL, NULL);
  554. }
  555. }
  556. void starpu_mpi_get_data_on_node(MPI_Comm comm, starpu_data_handle_t data_handle, int node)
  557. {
  558. int me, rank, tag;
  559. rank = starpu_data_get_rank(data_handle);
  560. tag = starpu_data_get_tag(data_handle);
  561. if(rank == -1) {
  562. fprintf(stderr,"StarPU needs to be told the MPI rank of this data, using starpu_data_set_rank\n");
  563. STARPU_ABORT();
  564. }
  565. if(tag == -1) {
  566. fprintf(stderr,"StarPU needs to be told the MPI tag of this data, using starpu_data_set_tag\n");
  567. STARPU_ABORT();
  568. }
  569. MPI_Comm_rank(comm, &me);
  570. if (node == rank) return;
  571. if (me == node)
  572. {
  573. MPI_Status status;
  574. starpu_mpi_recv(data_handle, rank, tag, comm, &status);
  575. }
  576. else if (me == rank)
  577. {
  578. starpu_mpi_send(data_handle, node, tag, comm);
  579. }
  580. }
  581. void starpu_mpi_redux_data(MPI_Comm comm, starpu_data_handle_t data_handle)
  582. {
  583. int me, rank, tag, nb_nodes;
  584. rank = starpu_data_get_rank(data_handle);
  585. tag = starpu_data_get_tag(data_handle);
  586. if(rank == -1) {
  587. fprintf(stderr,"StarPU needs to be told the MPI rank of this data, using starpu_data_set_rank\n");
  588. STARPU_ABORT();
  589. }
  590. if(tag == -1) {
  591. fprintf(stderr,"StarPU needs to be told the MPI tag of this data, using starpu_data_set_tag\n");
  592. STARPU_ABORT();
  593. }
  594. MPI_Comm_rank(comm, &me);
  595. MPI_Comm_size(comm, &nb_nodes);
  596. _STARPU_MPI_DEBUG("Doing reduction for data %p on node %d with %d nodes ...\n", data_handle, rank, nb_nodes);
  597. // need to count how many nodes have the data in redux mode
  598. if (me == rank) {
  599. int i;
  600. for(i=0 ; i<nb_nodes ; i++) {
  601. if (i != rank) {
  602. starpu_data_handle_t new_handle;
  603. starpu_data_register_same(&new_handle, data_handle);
  604. _STARPU_MPI_DEBUG("Receiving redux handle from %d in %p ...\n", i, new_handle);
  605. starpu_mpi_irecv_detached(new_handle, i, tag, comm, NULL, NULL);
  606. starpu_insert_task(data_handle->redux_cl,
  607. STARPU_RW, data_handle,
  608. STARPU_R, new_handle,
  609. 0);
  610. }
  611. }
  612. }
  613. else {
  614. _STARPU_MPI_DEBUG("Sending redux handle to %d ...\n", rank);
  615. starpu_mpi_isend_detached(data_handle, rank, tag, comm, NULL, NULL);
  616. }
  617. }