1 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
2 * All rights reserved. */
4 /* This program is free software; you can redistribute it and/or modify it
5 * under the terms of the license (GNU LGPL) which comes with this package. */
8 #include "smpi_mpi_dt_private.h"
10 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_pmpi, smpi,
11 "Logging specific to SMPI (pmpi)");
14 //this function need to be here because of the calls to smpi_bench
15 void TRACE_smpi_set_category(const char *category)
17 //need to end bench otherwise categories for execution tasks are wrong
19 TRACE_internal_smpi_set_category (category);
20 //begin bench after changing process's category
25 /* PMPI User level calls */
27 int PMPI_Init(int *argc, char ***argv)
29 smpi_process_init(argc, argv);
31 int rank = smpi_process_index();
32 TRACE_smpi_init(rank);
34 TRACE_smpi_computing_init(rank);
40 int PMPI_Finalize(void)
42 smpi_process_finalize();
45 int rank = smpi_process_index();
46 TRACE_smpi_computing_out(rank);
47 TRACE_smpi_finalize(smpi_process_index());
49 smpi_process_destroy();
53 int PMPI_Init_thread(int *argc, char ***argv, int required, int *provided)
55 if (provided != NULL) {
56 *provided = MPI_THREAD_MULTIPLE;
58 return MPI_Init(argc, argv);
61 int PMPI_Query_thread(int *provided)
66 if (provided == NULL) {
69 *provided = MPI_THREAD_MULTIPLE;
76 int PMPI_Is_thread_main(int *flag)
84 *flag = smpi_process_index() == 0;
91 int PMPI_Abort(MPI_Comm comm, int errorcode)
94 smpi_process_destroy();
96 int rank = smpi_process_index();
97 TRACE_smpi_computing_out(rank);
99 // FIXME: should kill all processes in comm instead
100 simcall_process_kill(SIMIX_process_self());
104 double PMPI_Wtime(void)
109 time = SIMIX_get_clock();
113 extern double sg_maxmin_precision;
114 double PMPI_Wtick(void)
116 return sg_maxmin_precision;
119 int PMPI_Address(void *location, MPI_Aint * address)
125 retval = MPI_ERR_ARG;
127 *address = (MPI_Aint) location;
128 retval = MPI_SUCCESS;
134 int PMPI_Type_free(MPI_Datatype * datatype)
140 retval = MPI_ERR_ARG;
142 smpi_datatype_free(datatype);
143 retval = MPI_SUCCESS;
149 int PMPI_Type_size(MPI_Datatype datatype, int *size)
154 if (datatype == MPI_DATATYPE_NULL) {
155 retval = MPI_ERR_TYPE;
156 } else if (size == NULL) {
157 retval = MPI_ERR_ARG;
159 *size = (int) smpi_datatype_size(datatype);
160 retval = MPI_SUCCESS;
166 int PMPI_Type_get_extent(MPI_Datatype datatype, MPI_Aint * lb, MPI_Aint * extent)
171 if (datatype == MPI_DATATYPE_NULL) {
172 retval = MPI_ERR_TYPE;
173 } else if (lb == NULL || extent == NULL) {
174 retval = MPI_ERR_ARG;
176 retval = smpi_datatype_extent(datatype, lb, extent);
182 int PMPI_Type_extent(MPI_Datatype datatype, MPI_Aint * extent)
188 if (datatype == MPI_DATATYPE_NULL) {
189 retval = MPI_ERR_TYPE;
190 } else if (extent == NULL) {
191 retval = MPI_ERR_ARG;
193 retval = smpi_datatype_extent(datatype, &dummy, extent);
199 int PMPI_Type_lb(MPI_Datatype datatype, MPI_Aint * disp)
204 if (datatype == MPI_DATATYPE_NULL) {
205 retval = MPI_ERR_TYPE;
206 } else if (disp == NULL) {
207 retval = MPI_ERR_ARG;
209 *disp = smpi_datatype_lb(datatype);
210 retval = MPI_SUCCESS;
216 int PMPI_Type_ub(MPI_Datatype datatype, MPI_Aint * disp)
221 if (datatype == MPI_DATATYPE_NULL) {
222 retval = MPI_ERR_TYPE;
223 } else if (disp == NULL) {
224 retval = MPI_ERR_ARG;
226 *disp = smpi_datatype_ub(datatype);
227 retval = MPI_SUCCESS;
233 int PMPI_Op_create(MPI_User_function * function, int commute, MPI_Op * op)
238 if (function == NULL || op == NULL) {
239 retval = MPI_ERR_ARG;
241 *op = smpi_op_new(function, commute);
242 retval = MPI_SUCCESS;
248 int PMPI_Op_free(MPI_Op * op)
254 retval = MPI_ERR_ARG;
255 } else if (*op == MPI_OP_NULL) {
258 smpi_op_destroy(*op);
260 retval = MPI_SUCCESS;
266 int PMPI_Group_free(MPI_Group * group)
272 retval = MPI_ERR_ARG;
274 smpi_group_destroy(*group);
275 *group = MPI_GROUP_NULL;
276 retval = MPI_SUCCESS;
282 int PMPI_Group_size(MPI_Group group, int *size)
287 if (group == MPI_GROUP_NULL) {
288 retval = MPI_ERR_GROUP;
289 } else if (size == NULL) {
290 retval = MPI_ERR_ARG;
292 *size = smpi_group_size(group);
293 retval = MPI_SUCCESS;
299 int PMPI_Group_rank(MPI_Group group, int *rank)
304 if (group == MPI_GROUP_NULL) {
305 retval = MPI_ERR_GROUP;
306 } else if (rank == NULL) {
307 retval = MPI_ERR_ARG;
309 *rank = smpi_group_rank(group, smpi_process_index());
310 retval = MPI_SUCCESS;
316 int PMPI_Group_translate_ranks(MPI_Group group1, int n, int *ranks1,
317 MPI_Group group2, int *ranks2)
319 int retval, i, index;
322 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
323 retval = MPI_ERR_GROUP;
325 for (i = 0; i < n; i++) {
326 index = smpi_group_index(group1, ranks1[i]);
327 ranks2[i] = smpi_group_rank(group2, index);
329 retval = MPI_SUCCESS;
335 int PMPI_Group_compare(MPI_Group group1, MPI_Group group2, int *result)
340 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
341 retval = MPI_ERR_GROUP;
342 } else if (result == NULL) {
343 retval = MPI_ERR_ARG;
345 *result = smpi_group_compare(group1, group2);
346 retval = MPI_SUCCESS;
352 int PMPI_Group_union(MPI_Group group1, MPI_Group group2,
353 MPI_Group * newgroup)
355 int retval, i, proc1, proc2, size, size2;
358 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
359 retval = MPI_ERR_GROUP;
360 } else if (newgroup == NULL) {
361 retval = MPI_ERR_ARG;
363 size = smpi_group_size(group1);
364 size2 = smpi_group_size(group2);
365 for (i = 0; i < size2; i++) {
366 proc2 = smpi_group_index(group2, i);
367 proc1 = smpi_group_rank(group1, proc2);
368 if (proc1 == MPI_UNDEFINED) {
373 *newgroup = MPI_GROUP_EMPTY;
375 *newgroup = smpi_group_new(size);
376 size2 = smpi_group_size(group1);
377 for (i = 0; i < size2; i++) {
378 proc1 = smpi_group_index(group1, i);
379 smpi_group_set_mapping(*newgroup, proc1, i);
381 for (i = size2; i < size; i++) {
382 proc2 = smpi_group_index(group2, i - size2);
383 smpi_group_set_mapping(*newgroup, proc2, i);
386 smpi_group_use(*newgroup);
387 retval = MPI_SUCCESS;
393 int PMPI_Group_intersection(MPI_Group group1, MPI_Group group2,
394 MPI_Group * newgroup)
396 int retval, i, proc1, proc2, size, size2;
399 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
400 retval = MPI_ERR_GROUP;
401 } else if (newgroup == NULL) {
402 retval = MPI_ERR_ARG;
404 size = smpi_group_size(group1);
405 size2 = smpi_group_size(group2);
406 for (i = 0; i < size2; i++) {
407 proc2 = smpi_group_index(group2, i);
408 proc1 = smpi_group_rank(group1, proc2);
409 if (proc1 == MPI_UNDEFINED) {
414 *newgroup = MPI_GROUP_EMPTY;
416 *newgroup = smpi_group_new(size);
417 size2 = smpi_group_size(group1);
418 for (i = 0; i < size2; i++) {
419 proc1 = smpi_group_index(group1, i);
420 proc2 = smpi_group_rank(group2, proc1);
421 if (proc2 != MPI_UNDEFINED) {
422 smpi_group_set_mapping(*newgroup, proc1, i);
426 smpi_group_use(*newgroup);
427 retval = MPI_SUCCESS;
433 int PMPI_Group_difference(MPI_Group group1, MPI_Group group2, MPI_Group * newgroup)
435 int retval, i, proc1, proc2, size, size2;
438 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
439 retval = MPI_ERR_GROUP;
440 } else if (newgroup == NULL) {
441 retval = MPI_ERR_ARG;
443 size = size2 = smpi_group_size(group1);
444 for (i = 0; i < size2; i++) {
445 proc1 = smpi_group_index(group1, i);
446 proc2 = smpi_group_rank(group2, proc1);
447 if (proc2 != MPI_UNDEFINED) {
452 *newgroup = MPI_GROUP_EMPTY;
454 *newgroup = smpi_group_new(size);
455 for (i = 0; i < size2; i++) {
456 proc1 = smpi_group_index(group1, i);
457 proc2 = smpi_group_rank(group2, proc1);
458 if (proc2 == MPI_UNDEFINED) {
459 smpi_group_set_mapping(*newgroup, proc1, i);
463 smpi_group_use(*newgroup);
464 retval = MPI_SUCCESS;
470 int PMPI_Group_incl(MPI_Group group, int n, int *ranks, MPI_Group * newgroup)
472 int retval, i, index;
475 if (group == MPI_GROUP_NULL) {
476 retval = MPI_ERR_GROUP;
477 } else if (newgroup == NULL) {
478 retval = MPI_ERR_ARG;
481 *newgroup = MPI_GROUP_EMPTY;
482 } else if (n == smpi_group_size(group)) {
485 *newgroup = smpi_group_new(n);
486 for (i = 0; i < n; i++) {
487 index = smpi_group_index(group, ranks[i]);
488 smpi_group_set_mapping(*newgroup, index, i);
491 smpi_group_use(*newgroup);
492 retval = MPI_SUCCESS;
498 int PMPI_Group_excl(MPI_Group group, int n, int *ranks, MPI_Group * newgroup)
500 int retval, i, size, rank, index;
503 if (group == MPI_GROUP_NULL) {
504 retval = MPI_ERR_GROUP;
505 } else if (newgroup == NULL) {
506 retval = MPI_ERR_ARG;
510 } else if (n == smpi_group_size(group)) {
511 *newgroup = MPI_GROUP_EMPTY;
513 size = smpi_group_size(group) - n;
514 *newgroup = smpi_group_new(size);
516 while (rank < size) {
517 for (i = 0; i < n; i++) {
518 if (ranks[i] == rank) {
523 index = smpi_group_index(group, rank);
524 smpi_group_set_mapping(*newgroup, index, rank);
529 smpi_group_use(*newgroup);
530 retval = MPI_SUCCESS;
536 int PMPI_Group_range_incl(MPI_Group group, int n, int ranges[][3],
537 MPI_Group * newgroup)
539 int retval, i, j, rank, size, index;
542 if (group == MPI_GROUP_NULL) {
543 retval = MPI_ERR_GROUP;
544 } else if (newgroup == NULL) {
545 retval = MPI_ERR_ARG;
548 *newgroup = MPI_GROUP_EMPTY;
551 for (i = 0; i < n; i++) {
552 for (rank = ranges[i][0]; /* First */
553 rank >= 0 && rank <= ranges[i][1]; /* Last */
554 rank += ranges[i][2] /* Stride */ ) {
558 if (size == smpi_group_size(group)) {
561 *newgroup = smpi_group_new(size);
563 for (i = 0; i < n; i++) {
564 for (rank = ranges[i][0]; /* First */
565 rank >= 0 && rank <= ranges[i][1]; /* Last */
566 rank += ranges[i][2] /* Stride */ ) {
567 index = smpi_group_index(group, rank);
568 smpi_group_set_mapping(*newgroup, index, j);
574 smpi_group_use(*newgroup);
575 retval = MPI_SUCCESS;
581 int PMPI_Group_range_excl(MPI_Group group, int n, int ranges[][3],
582 MPI_Group * newgroup)
584 int retval, i, newrank, rank, size, index, add;
587 if (group == MPI_GROUP_NULL) {
588 retval = MPI_ERR_GROUP;
589 } else if (newgroup == NULL) {
590 retval = MPI_ERR_ARG;
595 size = smpi_group_size(group);
596 for (i = 0; i < n; i++) {
597 for (rank = ranges[i][0]; /* First */
598 rank >= 0 && rank <= ranges[i][1]; /* Last */
599 rank += ranges[i][2] /* Stride */ ) {
604 *newgroup = MPI_GROUP_EMPTY;
606 *newgroup = smpi_group_new(size);
608 while (newrank < size) {
609 for (i = 0; i < n; i++) {
611 for (rank = ranges[i][0]; /* First */
612 rank >= 0 && rank <= ranges[i][1]; /* Last */
613 rank += ranges[i][2] /* Stride */ ) {
614 if (rank == newrank) {
620 index = smpi_group_index(group, newrank);
621 smpi_group_set_mapping(*newgroup, index, newrank);
627 smpi_group_use(*newgroup);
628 retval = MPI_SUCCESS;
634 int PMPI_Comm_rank(MPI_Comm comm, int *rank)
639 if (comm == MPI_COMM_NULL) {
640 retval = MPI_ERR_COMM;
641 } else if (rank == NULL) {
642 retval = MPI_ERR_ARG;
644 *rank = smpi_comm_rank(comm);
645 retval = MPI_SUCCESS;
651 int PMPI_Comm_size(MPI_Comm comm, int *size)
656 if (comm == MPI_COMM_NULL) {
657 retval = MPI_ERR_COMM;
658 } else if (size == NULL) {
659 retval = MPI_ERR_ARG;
661 *size = smpi_comm_size(comm);
662 retval = MPI_SUCCESS;
668 int PMPI_Comm_get_name (MPI_Comm comm, char* name, int* len)
673 if (comm == MPI_COMM_NULL) {
674 retval = MPI_ERR_COMM;
675 } else if (name == NULL || len == NULL) {
676 retval = MPI_ERR_ARG;
678 smpi_comm_get_name(comm, name, len);
679 retval = MPI_SUCCESS;
685 int PMPI_Comm_group(MPI_Comm comm, MPI_Group * group)
690 if (comm == MPI_COMM_NULL) {
691 retval = MPI_ERR_COMM;
692 } else if (group == NULL) {
693 retval = MPI_ERR_ARG;
695 *group = smpi_comm_group(comm);
696 retval = MPI_SUCCESS;
702 int PMPI_Comm_compare(MPI_Comm comm1, MPI_Comm comm2, int *result)
707 if (comm1 == MPI_COMM_NULL || comm2 == MPI_COMM_NULL) {
708 retval = MPI_ERR_COMM;
709 } else if (result == NULL) {
710 retval = MPI_ERR_ARG;
712 if (comm1 == comm2) { /* Same communicators means same groups */
716 smpi_group_compare(smpi_comm_group(comm1),
717 smpi_comm_group(comm2));
718 if (*result == MPI_IDENT) {
719 *result = MPI_CONGRUENT;
722 retval = MPI_SUCCESS;
728 int PMPI_Comm_dup(MPI_Comm comm, MPI_Comm * newcomm)
733 if (comm == MPI_COMM_NULL) {
734 retval = MPI_ERR_COMM;
735 } else if (newcomm == NULL) {
736 retval = MPI_ERR_ARG;
738 *newcomm = smpi_comm_new(smpi_comm_group(comm));
739 retval = MPI_SUCCESS;
745 int PMPI_Comm_create(MPI_Comm comm, MPI_Group group, MPI_Comm * newcomm)
750 if (comm == MPI_COMM_NULL) {
751 retval = MPI_ERR_COMM;
752 } else if (group == MPI_GROUP_NULL) {
753 retval = MPI_ERR_GROUP;
754 } else if (newcomm == NULL) {
755 retval = MPI_ERR_ARG;
757 *newcomm = smpi_comm_new(group);
758 retval = MPI_SUCCESS;
764 int PMPI_Comm_free(MPI_Comm * comm)
770 retval = MPI_ERR_ARG;
771 } else if (*comm == MPI_COMM_NULL) {
772 retval = MPI_ERR_COMM;
774 smpi_comm_destroy(*comm);
775 *comm = MPI_COMM_NULL;
776 retval = MPI_SUCCESS;
782 int PMPI_Comm_disconnect(MPI_Comm * comm)
784 /* TODO: wait until all communication in comm are done */
789 retval = MPI_ERR_ARG;
790 } else if (*comm == MPI_COMM_NULL) {
791 retval = MPI_ERR_COMM;
793 smpi_comm_destroy(*comm);
794 *comm = MPI_COMM_NULL;
795 retval = MPI_SUCCESS;
801 int PMPI_Comm_split(MPI_Comm comm, int color, int key, MPI_Comm* comm_out)
806 if (comm_out == NULL) {
807 retval = MPI_ERR_ARG;
808 } else if (comm == MPI_COMM_NULL) {
809 retval = MPI_ERR_COMM;
811 *comm_out = smpi_comm_split(comm, color, key);
812 retval = MPI_SUCCESS;
818 int PMPI_Send_init(void *buf, int count, MPI_Datatype datatype, int dst,
819 int tag, MPI_Comm comm, MPI_Request * request)
824 if (request == NULL) {
825 retval = MPI_ERR_ARG;
826 } else if (comm == MPI_COMM_NULL) {
827 retval = MPI_ERR_COMM;
828 } else if (dst == MPI_PROC_NULL) {
829 retval = MPI_SUCCESS;
831 *request = smpi_mpi_send_init(buf, count, datatype, dst, tag, comm);
832 retval = MPI_SUCCESS;
838 int PMPI_Recv_init(void *buf, int count, MPI_Datatype datatype, int src,
839 int tag, MPI_Comm comm, MPI_Request * request)
844 if (request == NULL) {
845 retval = MPI_ERR_ARG;
846 } else if (comm == MPI_COMM_NULL) {
847 retval = MPI_ERR_COMM;
848 } else if (src == MPI_PROC_NULL) {
849 retval = MPI_SUCCESS;
851 *request = smpi_mpi_recv_init(buf, count, datatype, src, tag, comm);
852 retval = MPI_SUCCESS;
858 int PMPI_Start(MPI_Request * request)
863 if (request == NULL || *request == MPI_REQUEST_NULL) {
864 retval = MPI_ERR_ARG;
866 smpi_mpi_start(*request);
867 retval = MPI_SUCCESS;
873 int PMPI_Startall(int count, MPI_Request * requests)
878 if (requests == NULL) {
879 retval = MPI_ERR_ARG;
881 smpi_mpi_startall(count, requests);
882 retval = MPI_SUCCESS;
888 int PMPI_Request_free(MPI_Request * request)
893 if (request == MPI_REQUEST_NULL) {
894 retval = MPI_ERR_ARG;
896 smpi_mpi_request_free(request);
897 retval = MPI_SUCCESS;
903 int PMPI_Irecv(void *buf, int count, MPI_Datatype datatype, int src,
904 int tag, MPI_Comm comm, MPI_Request * request)
910 if (request == NULL) {
911 retval = MPI_ERR_ARG;
912 } else if (comm == MPI_COMM_NULL) {
913 retval = MPI_ERR_COMM;
914 } else if (src == MPI_PROC_NULL) {
915 *request = MPI_REQUEST_NULL;
916 retval = MPI_SUCCESS;
917 } else if (src >= smpi_group_size(smpi_comm_group(comm)) || src <0){
918 retval = MPI_ERR_COMM;
919 } else if (count < 0) {
920 retval = MPI_ERR_COUNT;
921 } else if (buf==NULL && count > 0) {
922 retval = MPI_ERR_COUNT;
923 } else if (datatype == MPI_DATATYPE_NULL){
924 retval = MPI_ERR_TYPE;
925 } else if(tag<0 && tag != MPI_ANY_TAG){
926 retval = MPI_ERR_TAG;
930 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
931 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
932 TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
935 *request = smpi_mpi_irecv(buf, count, datatype, src, tag, comm);
936 retval = MPI_SUCCESS;
939 TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
940 (*request)->recv = 1;
949 int PMPI_Isend(void *buf, int count, MPI_Datatype datatype, int dst,
950 int tag, MPI_Comm comm, MPI_Request * request)
955 if (request == NULL) {
956 retval = MPI_ERR_ARG;
957 } else if (comm == MPI_COMM_NULL) {
958 retval = MPI_ERR_COMM;
959 } else if (dst == MPI_PROC_NULL) {
960 *request = MPI_REQUEST_NULL;
961 retval = MPI_SUCCESS;
962 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
963 retval = MPI_ERR_COMM;
964 } else if (count < 0) {
965 retval = MPI_ERR_COUNT;
966 } else if (buf==NULL && count > 0) {
967 retval = MPI_ERR_COUNT;
968 } else if (datatype == MPI_DATATYPE_NULL){
969 retval = MPI_ERR_TYPE;
970 } else if(tag<0 && tag != MPI_ANY_TAG){
971 retval = MPI_ERR_TAG;
975 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
976 TRACE_smpi_computing_out(rank);
977 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
978 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
979 TRACE_smpi_send(rank, rank, dst_traced);
982 *request = smpi_mpi_isend(buf, count, datatype, dst, tag, comm);
983 retval = MPI_SUCCESS;
986 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
987 (*request)->send = 1;
988 TRACE_smpi_computing_in(rank);
998 int PMPI_Recv(void *buf, int count, MPI_Datatype datatype, int src, int tag,
999 MPI_Comm comm, MPI_Status * status)
1005 if (comm == MPI_COMM_NULL) {
1006 retval = MPI_ERR_COMM;
1007 } else if (src == MPI_PROC_NULL) {
1008 smpi_empty_status(status);
1009 status->MPI_SOURCE = MPI_PROC_NULL;
1010 retval = MPI_SUCCESS;
1011 }else if(src >= smpi_group_size(smpi_comm_group(comm)) || src <0){
1012 retval = MPI_ERR_COMM;
1013 } else if (count < 0) {
1014 retval = MPI_ERR_COUNT;
1015 } else if (buf==NULL && count > 0) {
1016 retval = MPI_ERR_COUNT;
1017 } else if (datatype == MPI_DATATYPE_NULL){
1018 retval = MPI_ERR_TYPE;
1019 } else if(tag<0 && tag != MPI_ANY_TAG){
1020 retval = MPI_ERR_TAG;
1023 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1024 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
1025 TRACE_smpi_computing_out(rank);
1027 TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
1030 smpi_mpi_recv(buf, count, datatype, src, tag, comm, status);
1031 retval = MPI_SUCCESS;
1034 //the src may not have been known at the beginning of the recv (MPI_ANY_SOURCE)
1035 if(status!=MPI_STATUS_IGNORE)src_traced = smpi_group_index(smpi_comm_group(comm), status->MPI_SOURCE);
1036 TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
1037 TRACE_smpi_recv(rank, src_traced, rank);
1038 TRACE_smpi_computing_in(rank);
1046 int PMPI_Send(void *buf, int count, MPI_Datatype datatype, int dst, int tag,
1053 if (comm == MPI_COMM_NULL) {
1054 retval = MPI_ERR_COMM;
1055 } else if (dst == MPI_PROC_NULL) {
1056 retval = MPI_SUCCESS;
1057 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
1058 retval = MPI_ERR_COMM;
1059 } else if (count < 0) {
1060 retval = MPI_ERR_COUNT;
1061 } else if (buf==NULL && count > 0) {
1062 retval = MPI_ERR_COUNT;
1063 } else if (datatype == MPI_DATATYPE_NULL){
1064 retval = MPI_ERR_TYPE;
1065 } else if(tag<0 && tag != MPI_ANY_TAG){
1066 retval = MPI_ERR_TAG;
1070 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1071 TRACE_smpi_computing_out(rank);
1072 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1073 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
1074 TRACE_smpi_send(rank, rank, dst_traced);
1077 smpi_mpi_send(buf, count, datatype, dst, tag, comm);
1078 retval = MPI_SUCCESS;
1081 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
1082 TRACE_smpi_computing_in(rank);
1090 int PMPI_Sendrecv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1091 int dst, int sendtag, void *recvbuf, int recvcount,
1092 MPI_Datatype recvtype, int src, int recvtag,
1093 MPI_Comm comm, MPI_Status * status)
1099 if (comm == MPI_COMM_NULL) {
1100 retval = MPI_ERR_COMM;
1101 } else if (sendtype == MPI_DATATYPE_NULL
1102 || recvtype == MPI_DATATYPE_NULL) {
1103 retval = MPI_ERR_TYPE;
1104 } else if (src == MPI_PROC_NULL || dst == MPI_PROC_NULL) {
1105 smpi_empty_status(status);
1106 status->MPI_SOURCE = MPI_PROC_NULL;
1107 retval = MPI_SUCCESS;
1108 }else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0 || src >= smpi_group_size(smpi_comm_group(comm)) || src <0){
1109 retval = MPI_ERR_COMM;
1110 } else if (sendcount < 0 || recvcount<0) {
1111 retval = MPI_ERR_COUNT;
1112 } else if ((sendbuf==NULL && sendcount > 0)||(recvbuf==NULL && recvcount>0)) {
1113 retval = MPI_ERR_COUNT;
1114 } else if((sendtag<0 && sendtag != MPI_ANY_TAG)||(recvtag<0 && recvtag != MPI_ANY_TAG)){
1115 retval = MPI_ERR_TAG;
1119 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1120 TRACE_smpi_computing_out(rank);
1121 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1122 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
1123 TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
1124 TRACE_smpi_send(rank, rank, dst_traced);
1125 TRACE_smpi_send(rank, src_traced, rank);
1129 smpi_mpi_sendrecv(sendbuf, sendcount, sendtype, dst, sendtag, recvbuf,
1130 recvcount, recvtype, src, recvtag, comm, status);
1131 retval = MPI_SUCCESS;
1134 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1135 TRACE_smpi_recv(rank, rank, dst_traced);
1136 TRACE_smpi_recv(rank, src_traced, rank);
1137 TRACE_smpi_computing_in(rank);
1146 int PMPI_Sendrecv_replace(void *buf, int count, MPI_Datatype datatype,
1147 int dst, int sendtag, int src, int recvtag,
1148 MPI_Comm comm, MPI_Status * status)
1150 //TODO: suboptimal implementation
1154 size = smpi_datatype_size(datatype) * count;
1155 recvbuf = xbt_new(char, size);
1157 MPI_Sendrecv(buf, count, datatype, dst, sendtag, recvbuf, count,
1158 datatype, src, recvtag, comm, status);
1159 memcpy(buf, recvbuf, size * sizeof(char));
1164 int PMPI_Test(MPI_Request * request, int *flag, MPI_Status * status)
1169 if (request == MPI_REQUEST_NULL || flag == NULL) {
1170 retval = MPI_ERR_ARG;
1171 } else if (*request == MPI_REQUEST_NULL) {
1173 retval = MPI_ERR_REQUEST;
1175 *flag = smpi_mpi_test(request, status);
1176 retval = MPI_SUCCESS;
1182 int PMPI_Testany(int count, MPI_Request requests[], int *index, int *flag,
1183 MPI_Status * status)
1188 if (index == NULL || flag == NULL) {
1189 retval = MPI_ERR_ARG;
1191 *flag = smpi_mpi_testany(count, requests, index, status);
1192 retval = MPI_SUCCESS;
1198 int PMPI_Testall(int count, MPI_Request* requests, int* flag, MPI_Status* statuses)
1204 retval = MPI_ERR_ARG;
1206 *flag = smpi_mpi_testall(count, requests, statuses);
1207 retval = MPI_SUCCESS;
1213 int PMPI_Probe(int source, int tag, MPI_Comm comm, MPI_Status* status) {
1217 if (status == NULL) {
1218 retval = MPI_ERR_ARG;
1219 } else if (comm == MPI_COMM_NULL) {
1220 retval = MPI_ERR_COMM;
1221 } else if (source == MPI_PROC_NULL) {
1222 smpi_empty_status(status);
1223 status->MPI_SOURCE = MPI_PROC_NULL;
1224 retval = MPI_SUCCESS;
1226 smpi_mpi_probe(source, tag, comm, status);
1227 retval = MPI_SUCCESS;
1234 int PMPI_Iprobe(int source, int tag, MPI_Comm comm, int* flag, MPI_Status* status) {
1239 retval = MPI_ERR_ARG;
1240 } else if (status == NULL) {
1241 retval = MPI_ERR_ARG;
1242 } else if (comm == MPI_COMM_NULL) {
1243 retval = MPI_ERR_COMM;
1244 } else if (source == MPI_PROC_NULL) {
1245 smpi_empty_status(status);
1246 status->MPI_SOURCE = MPI_PROC_NULL;
1247 retval = MPI_SUCCESS;
1249 smpi_mpi_iprobe(source, tag, comm, flag, status);
1250 retval = MPI_SUCCESS;
1256 int PMPI_Wait(MPI_Request * request, MPI_Status * status)
1262 if (request == NULL) {
1263 retval = MPI_ERR_ARG;
1264 } else if (*request == MPI_REQUEST_NULL) {
1265 retval = MPI_ERR_REQUEST;
1269 int rank = request && (*request)->comm != MPI_COMM_NULL
1270 ? smpi_process_index()
1272 TRACE_smpi_computing_out(rank);
1274 MPI_Group group = smpi_comm_group((*request)->comm);
1275 int src_traced = smpi_group_index(group, (*request)->src);
1276 int dst_traced = smpi_group_index(group, (*request)->dst);
1277 int is_wait_for_receive = (*request)->recv;
1278 TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
1281 smpi_mpi_wait(request, status);
1282 retval = MPI_SUCCESS;
1285 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1286 if (is_wait_for_receive) {
1287 TRACE_smpi_recv(rank, src_traced, dst_traced);
1289 TRACE_smpi_computing_in(rank);
1298 int PMPI_Waitany(int count, MPI_Request requests[], int *index, MPI_Status * status)
1304 //save requests information for tracing
1306 xbt_dynar_t srcs = xbt_dynar_new(sizeof(int), NULL);
1307 xbt_dynar_t dsts = xbt_dynar_new(sizeof(int), NULL);
1308 xbt_dynar_t recvs = xbt_dynar_new(sizeof(int), NULL);
1309 for (i = 0; i < count; i++) {
1310 MPI_Request req = requests[i]; //already received requests are no longer valid
1312 int *asrc = xbt_new(int, 1);
1313 int *adst = xbt_new(int, 1);
1314 int *arecv = xbt_new(int, 1);
1318 xbt_dynar_insert_at(srcs, i, asrc);
1319 xbt_dynar_insert_at(dsts, i, adst);
1320 xbt_dynar_insert_at(recvs, i, arecv);
1325 int *t = xbt_new(int, 1);
1326 xbt_dynar_insert_at(srcs, i, t);
1327 xbt_dynar_insert_at(dsts, i, t);
1328 xbt_dynar_insert_at(recvs, i, t);
1332 int rank_traced = smpi_process_index();
1333 TRACE_smpi_computing_out(rank_traced);
1335 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1338 if (index == NULL) {
1339 retval = MPI_ERR_ARG;
1341 *index = smpi_mpi_waitany(count, requests, status);
1342 retval = MPI_SUCCESS;
1345 if(*index!=MPI_UNDEFINED){
1346 int src_traced, dst_traced, is_wait_for_receive;
1347 xbt_dynar_get_cpy(srcs, *index, &src_traced);
1348 xbt_dynar_get_cpy(dsts, *index, &dst_traced);
1349 xbt_dynar_get_cpy(recvs, *index, &is_wait_for_receive);
1350 if (is_wait_for_receive) {
1351 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1353 TRACE_smpi_ptp_out(rank_traced, src_traced, dst_traced, __FUNCTION__);
1354 //clean-up of dynars
1355 xbt_dynar_free(&srcs);
1356 xbt_dynar_free(&dsts);
1357 xbt_dynar_free(&recvs);
1359 TRACE_smpi_computing_in(rank_traced);
1367 int PMPI_Waitall(int count, MPI_Request requests[], MPI_Status status[])
1372 //save information from requests
1374 xbt_dynar_t srcs = xbt_dynar_new(sizeof(int), NULL);
1375 xbt_dynar_t dsts = xbt_dynar_new(sizeof(int), NULL);
1376 xbt_dynar_t recvs = xbt_dynar_new(sizeof(int), NULL);
1377 for (i = 0; i < count; i++) {
1378 MPI_Request req = requests[i];
1380 int *asrc = xbt_new(int, 1);
1381 int *adst = xbt_new(int, 1);
1382 int *arecv = xbt_new(int, 1);
1386 xbt_dynar_insert_at(srcs, i, asrc);
1387 xbt_dynar_insert_at(dsts, i, adst);
1388 xbt_dynar_insert_at(recvs, i, arecv);
1393 int *t = xbt_new(int, 1);
1394 xbt_dynar_insert_at(srcs, i, t);
1395 xbt_dynar_insert_at(dsts, i, t);
1396 xbt_dynar_insert_at(recvs, i, t);
1400 int rank_traced = smpi_process_index();
1401 TRACE_smpi_computing_out(rank_traced);
1403 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1405 smpi_mpi_waitall(count, requests, status);
1407 for (i = 0; i < count; i++) {
1408 int src_traced, dst_traced, is_wait_for_receive;
1409 xbt_dynar_get_cpy(srcs, i, &src_traced);
1410 xbt_dynar_get_cpy(dsts, i, &dst_traced);
1411 xbt_dynar_get_cpy(recvs, i, &is_wait_for_receive);
1412 if (is_wait_for_receive) {
1413 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1416 TRACE_smpi_ptp_out(rank_traced, -1, -1, __FUNCTION__);
1417 //clean-up of dynars
1418 xbt_dynar_free(&srcs);
1419 xbt_dynar_free(&dsts);
1420 xbt_dynar_free(&recvs);
1421 TRACE_smpi_computing_in(rank_traced);
1427 int PMPI_Waitsome(int incount, MPI_Request requests[], int *outcount,
1428 int *indices, MPI_Status status[])
1433 if (outcount == NULL || indices == NULL) {
1434 retval = MPI_ERR_ARG;
1436 *outcount = smpi_mpi_waitsome(incount, requests, indices, status);
1437 retval = MPI_SUCCESS;
1443 int PMPI_Testsome(int incount, MPI_Request requests[], int* outcount,
1444 int* indices, MPI_Status status[])
1449 if (outcount == NULL || indices == NULL) {
1450 retval = MPI_ERR_ARG;
1452 *outcount = smpi_mpi_testsome(incount, requests, indices, status);
1453 retval = MPI_SUCCESS;
1460 int PMPI_Bcast(void *buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm)
1466 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1467 TRACE_smpi_computing_out(rank);
1468 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1469 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1471 if (comm == MPI_COMM_NULL) {
1472 retval = MPI_ERR_COMM;
1474 smpi_mpi_bcast(buf, count, datatype, root, comm);
1475 retval = MPI_SUCCESS;
1478 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1479 TRACE_smpi_computing_in(rank);
1485 int PMPI_Barrier(MPI_Comm comm)
1491 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1492 TRACE_smpi_computing_out(rank);
1493 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1495 if (comm == MPI_COMM_NULL) {
1496 retval = MPI_ERR_COMM;
1498 smpi_mpi_barrier(comm);
1499 retval = MPI_SUCCESS;
1502 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1503 TRACE_smpi_computing_in(rank);
1509 int PMPI_Gather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1510 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1511 int root, MPI_Comm comm)
1517 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1518 TRACE_smpi_computing_out(rank);
1519 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1520 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1522 if (comm == MPI_COMM_NULL) {
1523 retval = MPI_ERR_COMM;
1524 } else if (sendtype == MPI_DATATYPE_NULL
1525 || recvtype == MPI_DATATYPE_NULL) {
1526 retval = MPI_ERR_TYPE;
1528 smpi_mpi_gather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1529 recvtype, root, comm);
1530 retval = MPI_SUCCESS;
1533 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1534 TRACE_smpi_computing_in(rank);
1540 int PMPI_Gatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1541 void *recvbuf, int *recvcounts, int *displs,
1542 MPI_Datatype recvtype, int root, MPI_Comm comm)
1548 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1549 TRACE_smpi_computing_out(rank);
1550 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1551 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1553 if (comm == MPI_COMM_NULL) {
1554 retval = MPI_ERR_COMM;
1555 } else if (sendtype == MPI_DATATYPE_NULL
1556 || recvtype == MPI_DATATYPE_NULL) {
1557 retval = MPI_ERR_TYPE;
1558 } else if (recvcounts == NULL || displs == NULL) {
1559 retval = MPI_ERR_ARG;
1561 smpi_mpi_gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1562 displs, recvtype, root, comm);
1563 retval = MPI_SUCCESS;
1566 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1567 TRACE_smpi_computing_in(rank);
1573 int PMPI_Allgather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1574 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1581 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1582 TRACE_smpi_computing_out(rank);
1583 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1585 if (comm == MPI_COMM_NULL) {
1586 retval = MPI_ERR_COMM;
1587 } else if (sendtype == MPI_DATATYPE_NULL
1588 || recvtype == MPI_DATATYPE_NULL) {
1589 retval = MPI_ERR_TYPE;
1591 smpi_mpi_allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1593 retval = MPI_SUCCESS;
1596 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1602 int PMPI_Allgatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1603 void *recvbuf, int *recvcounts, int *displs,
1604 MPI_Datatype recvtype, MPI_Comm comm)
1610 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1611 TRACE_smpi_computing_out(rank);
1612 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1614 if (comm == MPI_COMM_NULL) {
1615 retval = MPI_ERR_COMM;
1616 } else if (sendtype == MPI_DATATYPE_NULL
1617 || recvtype == MPI_DATATYPE_NULL) {
1618 retval = MPI_ERR_TYPE;
1619 } else if (recvcounts == NULL || displs == NULL) {
1620 retval = MPI_ERR_ARG;
1622 smpi_mpi_allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1623 displs, recvtype, comm);
1624 retval = MPI_SUCCESS;
1627 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1628 TRACE_smpi_computing_in(rank);
1634 int PMPI_Scatter(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1635 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1636 int root, MPI_Comm comm)
1642 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1643 TRACE_smpi_computing_out(rank);
1644 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1646 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1648 if (comm == MPI_COMM_NULL) {
1649 retval = MPI_ERR_COMM;
1650 } else if (sendtype == MPI_DATATYPE_NULL
1651 || recvtype == MPI_DATATYPE_NULL) {
1652 retval = MPI_ERR_TYPE;
1654 smpi_mpi_scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1655 recvtype, root, comm);
1656 retval = MPI_SUCCESS;
1659 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1660 TRACE_smpi_computing_in(rank);
1666 int PMPI_Scatterv(void *sendbuf, int *sendcounts, int *displs,
1667 MPI_Datatype sendtype, void *recvbuf, int recvcount,
1668 MPI_Datatype recvtype, int root, MPI_Comm comm)
1674 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1675 TRACE_smpi_computing_out(rank);
1676 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1677 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1679 if (comm == MPI_COMM_NULL) {
1680 retval = MPI_ERR_COMM;
1681 } else if (sendtype == MPI_DATATYPE_NULL
1682 || recvtype == MPI_DATATYPE_NULL) {
1683 retval = MPI_ERR_TYPE;
1684 } else if (sendcounts == NULL || displs == NULL) {
1685 retval = MPI_ERR_ARG;
1687 smpi_mpi_scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf,
1688 recvcount, recvtype, root, comm);
1689 retval = MPI_SUCCESS;
1692 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1693 TRACE_smpi_computing_in(rank);
1699 int PMPI_Reduce(void *sendbuf, void *recvbuf, int count,
1700 MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm)
1706 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1707 TRACE_smpi_computing_out(rank);
1708 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1709 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1711 if (comm == MPI_COMM_NULL) {
1712 retval = MPI_ERR_COMM;
1713 } else if (datatype == MPI_DATATYPE_NULL || op == MPI_OP_NULL) {
1714 retval = MPI_ERR_ARG;
1716 smpi_mpi_reduce(sendbuf, recvbuf, count, datatype, op, root, comm);
1717 retval = MPI_SUCCESS;
1720 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1721 TRACE_smpi_computing_in(rank);
1727 int PMPI_Allreduce(void *sendbuf, void *recvbuf, int count,
1728 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1734 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1735 TRACE_smpi_computing_out(rank);
1736 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1738 if (comm == MPI_COMM_NULL) {
1739 retval = MPI_ERR_COMM;
1740 } else if (datatype == MPI_DATATYPE_NULL) {
1741 retval = MPI_ERR_TYPE;
1742 } else if (op == MPI_OP_NULL) {
1743 retval = MPI_ERR_OP;
1745 smpi_mpi_allreduce(sendbuf, recvbuf, count, datatype, op, comm);
1746 retval = MPI_SUCCESS;
1749 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1750 TRACE_smpi_computing_in(rank);
1756 int PMPI_Scan(void *sendbuf, void *recvbuf, int count,
1757 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1763 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1764 TRACE_smpi_computing_out(rank);
1765 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1767 if (comm == MPI_COMM_NULL) {
1768 retval = MPI_ERR_COMM;
1769 } else if (datatype == MPI_DATATYPE_NULL) {
1770 retval = MPI_ERR_TYPE;
1771 } else if (op == MPI_OP_NULL) {
1772 retval = MPI_ERR_OP;
1774 smpi_mpi_scan(sendbuf, recvbuf, count, datatype, op, comm);
1775 retval = MPI_SUCCESS;
1778 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1779 TRACE_smpi_computing_in(rank);
1785 int PMPI_Reduce_scatter(void *sendbuf, void *recvbuf, int *recvcounts,
1786 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1788 int retval, i, size, count;
1790 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1794 TRACE_smpi_computing_out(rank);
1795 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1797 if (comm == MPI_COMM_NULL) {
1798 retval = MPI_ERR_COMM;
1799 } else if (datatype == MPI_DATATYPE_NULL) {
1800 retval = MPI_ERR_TYPE;
1801 } else if (op == MPI_OP_NULL) {
1802 retval = MPI_ERR_OP;
1803 } else if (recvcounts == NULL) {
1804 retval = MPI_ERR_ARG;
1806 /* arbitrarily choose root as rank 0 */
1807 /* TODO: faster direct implementation ? */
1808 size = smpi_comm_size(comm);
1810 displs = xbt_new(int, size);
1811 for (i = 0; i < size; i++) {
1812 count += recvcounts[i];
1815 smpi_mpi_reduce(sendbuf, recvbuf, count, datatype, op, 0, comm);
1816 smpi_mpi_scatterv(recvbuf, recvcounts, displs, datatype, recvbuf,
1817 recvcounts[rank], datatype, 0, comm);
1819 retval = MPI_SUCCESS;
1822 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1823 TRACE_smpi_computing_in(rank);
1829 int PMPI_Alltoall(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1830 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1833 int retval, size, sendsize;
1837 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1838 TRACE_smpi_computing_out(rank);
1839 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1841 if (comm == MPI_COMM_NULL) {
1842 retval = MPI_ERR_COMM;
1843 } else if (sendtype == MPI_DATATYPE_NULL
1844 || recvtype == MPI_DATATYPE_NULL) {
1845 retval = MPI_ERR_TYPE;
1847 size = smpi_comm_size(comm);
1848 sendsize = smpi_datatype_size(sendtype) * sendcount;
1849 if (sendsize < 200 && size > 12) {
1851 smpi_coll_tuned_alltoall_bruck(sendbuf, sendcount, sendtype,
1852 recvbuf, recvcount, recvtype,
1854 } else if (sendsize < 3000) {
1856 smpi_coll_tuned_alltoall_basic_linear(sendbuf, sendcount,
1858 recvcount, recvtype, comm);
1861 smpi_coll_tuned_alltoall_pairwise(sendbuf, sendcount, sendtype,
1862 recvbuf, recvcount, recvtype,
1867 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1868 TRACE_smpi_computing_in(rank);
1874 int PMPI_Alltoallv(void *sendbuf, int *sendcounts, int *senddisps,
1875 MPI_Datatype sendtype, void *recvbuf, int *recvcounts,
1876 int *recvdisps, MPI_Datatype recvtype, MPI_Comm comm)
1882 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1883 TRACE_smpi_computing_out(rank);
1884 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1886 if (comm == MPI_COMM_NULL) {
1887 retval = MPI_ERR_COMM;
1888 } else if (sendtype == MPI_DATATYPE_NULL
1889 || recvtype == MPI_DATATYPE_NULL) {
1890 retval = MPI_ERR_TYPE;
1891 } else if (sendcounts == NULL || senddisps == NULL || recvcounts == NULL
1892 || recvdisps == NULL) {
1893 retval = MPI_ERR_ARG;
1896 smpi_coll_basic_alltoallv(sendbuf, sendcounts, senddisps, sendtype,
1897 recvbuf, recvcounts, recvdisps, recvtype,
1901 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1902 TRACE_smpi_computing_in(rank);
1909 int PMPI_Get_processor_name(char *name, int *resultlen)
1911 int retval = MPI_SUCCESS;
1914 strncpy(name, SIMIX_host_get_name(SIMIX_host_self()),
1915 MPI_MAX_PROCESSOR_NAME - 1);
1918 MPI_MAX_PROCESSOR_NAME ? MPI_MAX_PROCESSOR_NAME : strlen(name);
1924 int PMPI_Get_count(MPI_Status * status, MPI_Datatype datatype, int *count)
1926 int retval = MPI_SUCCESS;
1930 if (status == NULL || count == NULL) {
1931 retval = MPI_ERR_ARG;
1932 } else if (datatype == MPI_DATATYPE_NULL) {
1933 retval = MPI_ERR_TYPE;
1935 size = smpi_datatype_size(datatype);
1938 } else if (status->count % size != 0) {
1939 retval = MPI_UNDEFINED;
1941 *count = smpi_mpi_get_count(status, datatype);
1948 int PMPI_Type_contiguous(int count, MPI_Datatype old_type, MPI_Datatype* new_type) {
1952 if (old_type == MPI_DATATYPE_NULL) {
1953 retval = MPI_ERR_TYPE;
1954 } else if (count<=0){
1955 retval = MPI_ERR_COUNT;
1957 retval = smpi_datatype_contiguous(count, old_type, new_type);
1963 int PMPI_Type_commit(MPI_Datatype* datatype) {
1967 if (datatype == MPI_DATATYPE_NULL) {
1968 retval = MPI_ERR_TYPE;
1970 smpi_datatype_commit(datatype);
1971 retval = MPI_SUCCESS;
1978 int PMPI_Type_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
1982 if (old_type == MPI_DATATYPE_NULL) {
1983 retval = MPI_ERR_TYPE;
1984 } else if (count<=0 || blocklen<=0){
1985 retval = MPI_ERR_COUNT;
1987 retval = smpi_datatype_vector(count, blocklen, stride, old_type, new_type);
1993 int PMPI_Type_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
1997 if (old_type == MPI_DATATYPE_NULL) {
1998 retval = MPI_ERR_TYPE;
1999 } else if (count<=0 || blocklen<=0){
2000 retval = MPI_ERR_COUNT;
2002 retval = smpi_datatype_hvector(count, blocklen, stride, old_type, new_type);
2009 int PMPI_Type_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2013 if (old_type == MPI_DATATYPE_NULL) {
2014 retval = MPI_ERR_TYPE;
2015 } else if (count<=0){
2016 retval = MPI_ERR_COUNT;
2018 retval = smpi_datatype_indexed(count, blocklens, indices, old_type, new_type);
2024 int PMPI_Type_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2028 if (old_type == MPI_DATATYPE_NULL) {
2029 retval = MPI_ERR_TYPE;
2030 } else if (count<=0){
2031 retval = MPI_ERR_COUNT;
2033 retval = smpi_datatype_hindexed(count, blocklens, indices, old_type, new_type);
2040 int PMPI_Type_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type) {
2045 retval = MPI_ERR_COUNT;
2047 retval = smpi_datatype_struct(count, blocklens, indices, old_types, new_type);
2053 /* The following calls are not yet implemented and will fail at runtime. */
2054 /* Once implemented, please move them above this notice. */
2056 static int not_yet_implemented(void) {
2057 XBT_WARN("Not yet implemented");
2061 int PMPI_Pack_size(int incount, MPI_Datatype datatype, MPI_Comm comm, int* size) {
2062 return not_yet_implemented();
2065 int PMPI_Cart_coords(MPI_Comm comm, int rank, int maxdims, int* coords) {
2066 return not_yet_implemented();
2069 int PMPI_Cart_create(MPI_Comm comm_old, int ndims, int* dims, int* periods, int reorder, MPI_Comm* comm_cart) {
2070 return not_yet_implemented();
2073 int PMPI_Cart_get(MPI_Comm comm, int maxdims, int* dims, int* periods, int* coords) {
2074 return not_yet_implemented();
2077 int PMPI_Cart_map(MPI_Comm comm_old, int ndims, int* dims, int* periods, int* newrank) {
2078 return not_yet_implemented();
2081 int PMPI_Cart_rank(MPI_Comm comm, int* coords, int* rank) {
2082 return not_yet_implemented();
2085 int PMPI_Cart_shift(MPI_Comm comm, int direction, int displ, int* source, int* dest) {
2086 return not_yet_implemented();
2089 int PMPI_Cart_sub(MPI_Comm comm, int* remain_dims, MPI_Comm* comm_new) {
2090 return not_yet_implemented();
2093 int PMPI_Cartdim_get(MPI_Comm comm, int* ndims) {
2094 return not_yet_implemented();
2097 int PMPI_Graph_create(MPI_Comm comm_old, int nnodes, int* index, int* edges, int reorder, MPI_Comm* comm_graph) {
2098 return not_yet_implemented();
2101 int PMPI_Graph_get(MPI_Comm comm, int maxindex, int maxedges, int* index, int* edges) {
2102 return not_yet_implemented();
2105 int PMPI_Graph_map(MPI_Comm comm_old, int nnodes, int* index, int* edges, int* newrank) {
2106 return not_yet_implemented();
2109 int PMPI_Graph_neighbors(MPI_Comm comm, int rank, int maxneighbors, int* neighbors) {
2110 return not_yet_implemented();
2113 int PMPI_Graph_neighbors_count(MPI_Comm comm, int rank, int* nneighbors) {
2114 return not_yet_implemented();
2117 int PMPI_Graphdims_get(MPI_Comm comm, int* nnodes, int* nedges) {
2118 return not_yet_implemented();
2121 int PMPI_Topo_test(MPI_Comm comm, int* top_type) {
2122 return not_yet_implemented();
2125 int PMPI_Error_class(int errorcode, int* errorclass) {
2126 return not_yet_implemented();
2129 int PMPI_Errhandler_create(MPI_Handler_function* function, MPI_Errhandler* errhandler) {
2130 return not_yet_implemented();
2133 int PMPI_Errhandler_free(MPI_Errhandler* errhandler) {
2134 return not_yet_implemented();
2137 int PMPI_Errhandler_get(MPI_Comm comm, MPI_Errhandler* errhandler) {
2138 return not_yet_implemented();
2141 int PMPI_Error_string(int errorcode, char* string, int* resultlen) {
2142 return not_yet_implemented();
2145 int PMPI_Errhandler_set(MPI_Comm comm, MPI_Errhandler errhandler) {
2146 return not_yet_implemented();
2150 int PMPI_Cancel(MPI_Request* request) {
2151 return not_yet_implemented();
2154 int PMPI_Buffer_attach(void* buffer, int size) {
2155 return not_yet_implemented();
2158 int PMPI_Buffer_detach(void* buffer, int* size) {
2159 return not_yet_implemented();
2162 int PMPI_Comm_test_inter(MPI_Comm comm, int* flag) {
2163 return not_yet_implemented();
2166 int PMPI_Comm_get_attr (MPI_Comm comm, int comm_keyval, void *attribute_val, int *flag)
2168 return not_yet_implemented();
2171 int PMPI_Pcontrol(const int level )
2173 return not_yet_implemented();
2176 int PMPI_Unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Datatype type, MPI_Comm comm) {
2177 return not_yet_implemented();
2180 int PMPI_Ssend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2181 return not_yet_implemented();
2184 int PMPI_Ssend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2185 return not_yet_implemented();
2188 int PMPI_Intercomm_create(MPI_Comm local_comm, int local_leader, MPI_Comm peer_comm, int remote_leader, int tag, MPI_Comm* comm_out) {
2189 return not_yet_implemented();
2192 int PMPI_Intercomm_merge(MPI_Comm comm, int high, MPI_Comm* comm_out) {
2193 return not_yet_implemented();
2196 int PMPI_Bsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2197 return not_yet_implemented();
2200 int PMPI_Bsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2201 return not_yet_implemented();
2204 int PMPI_Ibsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2205 return not_yet_implemented();
2208 int PMPI_Comm_remote_group(MPI_Comm comm, MPI_Group* group) {
2209 return not_yet_implemented();
2212 int PMPI_Comm_remote_size(MPI_Comm comm, int* size) {
2213 return not_yet_implemented();
2216 int PMPI_Issend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2217 return not_yet_implemented();
2221 int PMPI_Attr_delete(MPI_Comm comm, int keyval) {
2222 return not_yet_implemented();
2225 int PMPI_Attr_get(MPI_Comm comm, int keyval, void* attr_value, int* flag) {
2226 return not_yet_implemented();
2229 int PMPI_Attr_put(MPI_Comm comm, int keyval, void* attr_value) {
2230 return not_yet_implemented();
2233 int PMPI_Rsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2234 return not_yet_implemented();
2237 int PMPI_Rsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2238 return not_yet_implemented();
2241 int PMPI_Irsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2242 return not_yet_implemented();
2245 int PMPI_Keyval_create(MPI_Copy_function* copy_fn, MPI_Delete_function* delete_fn, int* keyval, void* extra_state) {
2246 return not_yet_implemented();
2249 int PMPI_Keyval_free(int* keyval) {
2250 return not_yet_implemented();
2253 int PMPI_Test_cancelled(MPI_Status* status, int* flag) {
2254 return not_yet_implemented();
2257 int PMPI_Pack(void* inbuf, int incount, MPI_Datatype type, void* outbuf, int outcount, int* position, MPI_Comm comm) {
2258 return not_yet_implemented();
2261 int PMPI_Get_elements(MPI_Status* status, MPI_Datatype datatype, int* elements) {
2262 return not_yet_implemented();
2265 int PMPI_Dims_create(int nnodes, int ndims, int* dims) {
2266 return not_yet_implemented();
2269 int PMPI_Initialized(int* flag) {
2270 return not_yet_implemented();