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)
187 if (datatype == MPI_DATATYPE_NULL) {
188 retval = MPI_ERR_TYPE;
189 } else if (extent == NULL) {
190 retval = MPI_ERR_ARG;
192 *extent = smpi_datatype_get_extent(datatype);
193 retval = MPI_SUCCESS;
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 if(*group!= smpi_comm_group(MPI_COMM_WORLD))// do not free the group of the comm_world
275 smpi_group_destroy(*group);
276 *group = MPI_GROUP_NULL;
277 retval = MPI_SUCCESS;
283 int PMPI_Group_size(MPI_Group group, int *size)
288 if (group == MPI_GROUP_NULL) {
289 retval = MPI_ERR_GROUP;
290 } else if (size == NULL) {
291 retval = MPI_ERR_ARG;
293 *size = smpi_group_size(group);
294 retval = MPI_SUCCESS;
300 int PMPI_Group_rank(MPI_Group group, int *rank)
305 if (group == MPI_GROUP_NULL) {
306 retval = MPI_ERR_GROUP;
307 } else if (rank == NULL) {
308 retval = MPI_ERR_ARG;
310 *rank = smpi_group_rank(group, smpi_process_index());
311 retval = MPI_SUCCESS;
317 int PMPI_Group_translate_ranks(MPI_Group group1, int n, int *ranks1,
318 MPI_Group group2, int *ranks2)
320 int retval, i, index;
323 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
324 retval = MPI_ERR_GROUP;
326 for (i = 0; i < n; i++) {
327 index = smpi_group_index(group1, ranks1[i]);
328 ranks2[i] = smpi_group_rank(group2, index);
330 retval = MPI_SUCCESS;
336 int PMPI_Group_compare(MPI_Group group1, MPI_Group group2, int *result)
341 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
342 retval = MPI_ERR_GROUP;
343 } else if (result == NULL) {
344 retval = MPI_ERR_ARG;
346 *result = smpi_group_compare(group1, group2);
347 retval = MPI_SUCCESS;
353 int PMPI_Group_union(MPI_Group group1, MPI_Group group2,
354 MPI_Group * newgroup)
356 int retval, i, proc1, proc2, size, size2;
359 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
360 retval = MPI_ERR_GROUP;
361 } else if (newgroup == NULL) {
362 retval = MPI_ERR_ARG;
364 size = smpi_group_size(group1);
365 size2 = smpi_group_size(group2);
366 for (i = 0; i < size2; i++) {
367 proc2 = smpi_group_index(group2, i);
368 proc1 = smpi_group_rank(group1, proc2);
369 if (proc1 == MPI_UNDEFINED) {
374 *newgroup = MPI_GROUP_EMPTY;
376 *newgroup = smpi_group_new(size);
377 size2 = smpi_group_size(group1);
378 for (i = 0; i < size2; i++) {
379 proc1 = smpi_group_index(group1, i);
380 smpi_group_set_mapping(*newgroup, proc1, i);
382 for (i = size2; i < size; i++) {
383 proc2 = smpi_group_index(group2, i - size2);
384 smpi_group_set_mapping(*newgroup, proc2, i);
387 smpi_group_use(*newgroup);
388 retval = MPI_SUCCESS;
394 int PMPI_Group_intersection(MPI_Group group1, MPI_Group group2,
395 MPI_Group * newgroup)
397 int retval, i, proc1, proc2, size;
400 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
401 retval = MPI_ERR_GROUP;
402 } else if (newgroup == NULL) {
403 retval = MPI_ERR_ARG;
405 size = smpi_group_size(group2);
406 for (i = 0; i < size; 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);
418 for (i = 0; i < smpi_group_size(group2); i++) {
419 proc2 = smpi_group_index(group2, i);
420 proc1 = smpi_group_rank(group1, proc2);
421 if (proc1 != MPI_UNDEFINED) {
422 smpi_group_set_mapping(*newgroup, proc2, j);
427 smpi_group_use(*newgroup);
428 retval = MPI_SUCCESS;
434 int PMPI_Group_difference(MPI_Group group1, MPI_Group group2, MPI_Group * newgroup)
436 int retval, i, proc1, proc2, size, size2;
439 if (group1 == MPI_GROUP_NULL || group2 == MPI_GROUP_NULL) {
440 retval = MPI_ERR_GROUP;
441 } else if (newgroup == NULL) {
442 retval = MPI_ERR_ARG;
444 size = size2 = smpi_group_size(group1);
445 for (i = 0; i < size2; i++) {
446 proc1 = smpi_group_index(group1, i);
447 proc2 = smpi_group_rank(group2, proc1);
448 if (proc2 != MPI_UNDEFINED) {
453 *newgroup = MPI_GROUP_EMPTY;
455 *newgroup = smpi_group_new(size);
456 for (i = 0; i < size2; i++) {
457 proc1 = smpi_group_index(group1, i);
458 proc2 = smpi_group_rank(group2, proc1);
459 if (proc2 == MPI_UNDEFINED) {
460 smpi_group_set_mapping(*newgroup, proc1, i);
464 smpi_group_use(*newgroup);
465 retval = MPI_SUCCESS;
471 int PMPI_Group_incl(MPI_Group group, int n, int *ranks, MPI_Group * newgroup)
473 int retval, i, index;
476 if (group == MPI_GROUP_NULL) {
477 retval = MPI_ERR_GROUP;
478 } else if (newgroup == NULL) {
479 retval = MPI_ERR_ARG;
482 *newgroup = MPI_GROUP_EMPTY;
483 } else if (n == smpi_group_size(group)) {
486 *newgroup = smpi_group_new(n);
487 for (i = 0; i < n; i++) {
488 index = smpi_group_index(group, ranks[i]);
489 smpi_group_set_mapping(*newgroup, index, i);
492 smpi_group_use(*newgroup);
493 retval = MPI_SUCCESS;
499 int PMPI_Group_excl(MPI_Group group, int n, int *ranks, MPI_Group * newgroup)
501 int retval, i, j, newsize, oldsize, index;
504 if (group == MPI_GROUP_NULL) {
505 retval = MPI_ERR_GROUP;
506 } else if (newgroup == NULL) {
507 retval = MPI_ERR_ARG;
511 } else if (n == smpi_group_size(group)) {
512 *newgroup = MPI_GROUP_EMPTY;
514 oldsize=smpi_group_size(group);
515 newsize = oldsize - n;
516 *newgroup = smpi_group_new(newsize);
518 int* to_exclude=xbt_new(int, smpi_group_size(group));
519 for(i=0; i<oldsize; i++)
522 to_exclude[ranks[i]]=1;
525 for(i=0; i<oldsize; i++){
526 if(to_exclude[i]==0){
527 index = smpi_group_index(group, i);
528 smpi_group_set_mapping(*newgroup, index, j);
533 xbt_free(to_exclude);
535 smpi_group_use(*newgroup);
536 retval = MPI_SUCCESS;
542 int PMPI_Group_range_incl(MPI_Group group, int n, int ranges[][3],
543 MPI_Group * newgroup)
545 int retval, i, j, rank, size, index;
548 if (group == MPI_GROUP_NULL) {
549 retval = MPI_ERR_GROUP;
550 } else if (newgroup == NULL) {
551 retval = MPI_ERR_ARG;
554 *newgroup = MPI_GROUP_EMPTY;
557 for (i = 0; i < n; i++) {
558 for (rank = ranges[i][0]; /* First */
559 rank >= 0 && rank <= ranges[i][1]; /* Last */
560 rank += ranges[i][2] /* Stride */ ) {
564 if (size == smpi_group_size(group)) {
567 *newgroup = smpi_group_new(size);
569 for (i = 0; i < n; i++) {
570 for (rank = ranges[i][0]; /* First */
571 rank >= 0 && rank <= ranges[i][1]; /* Last */
572 rank += ranges[i][2] /* Stride */ ) {
573 index = smpi_group_index(group, rank);
574 smpi_group_set_mapping(*newgroup, index, j);
580 smpi_group_use(*newgroup);
581 retval = MPI_SUCCESS;
587 int PMPI_Group_range_excl(MPI_Group group, int n, int ranges[][3],
588 MPI_Group * newgroup)
590 int retval, i, newrank, rank, size, index, add;
593 if (group == MPI_GROUP_NULL) {
594 retval = MPI_ERR_GROUP;
595 } else if (newgroup == NULL) {
596 retval = MPI_ERR_ARG;
601 size = smpi_group_size(group);
602 for (i = 0; i < n; i++) {
603 for (rank = ranges[i][0]; /* First */
604 rank >= 0 && rank <= ranges[i][1]; /* Last */
605 rank += ranges[i][2] /* Stride */ ) {
610 *newgroup = MPI_GROUP_EMPTY;
612 *newgroup = smpi_group_new(size);
614 while (newrank < size) {
615 for (i = 0; i < n; i++) {
617 for (rank = ranges[i][0]; /* First */
618 rank >= 0 && rank <= ranges[i][1]; /* Last */
619 rank += ranges[i][2] /* Stride */ ) {
620 if (rank == newrank) {
626 index = smpi_group_index(group, newrank);
627 smpi_group_set_mapping(*newgroup, index, newrank);
630 newrank++; //added to avoid looping, need to be checked ..
634 smpi_group_use(*newgroup);
635 retval = MPI_SUCCESS;
641 int PMPI_Comm_rank(MPI_Comm comm, int *rank)
646 if (comm == MPI_COMM_NULL) {
647 retval = MPI_ERR_COMM;
648 } else if (rank == NULL) {
649 retval = MPI_ERR_ARG;
651 *rank = smpi_comm_rank(comm);
652 retval = MPI_SUCCESS;
658 int PMPI_Comm_size(MPI_Comm comm, int *size)
663 if (comm == MPI_COMM_NULL) {
664 retval = MPI_ERR_COMM;
665 } else if (size == NULL) {
666 retval = MPI_ERR_ARG;
668 *size = smpi_comm_size(comm);
669 retval = MPI_SUCCESS;
675 int PMPI_Comm_get_name (MPI_Comm comm, char* name, int* len)
680 if (comm == MPI_COMM_NULL) {
681 retval = MPI_ERR_COMM;
682 } else if (name == NULL || len == NULL) {
683 retval = MPI_ERR_ARG;
685 smpi_comm_get_name(comm, name, len);
686 retval = MPI_SUCCESS;
692 int PMPI_Comm_group(MPI_Comm comm, MPI_Group * group)
697 if (comm == MPI_COMM_NULL) {
698 retval = MPI_ERR_COMM;
699 } else if (group == NULL) {
700 retval = MPI_ERR_ARG;
702 *group = smpi_comm_group(comm);
703 retval = MPI_SUCCESS;
709 int PMPI_Comm_compare(MPI_Comm comm1, MPI_Comm comm2, int *result)
714 if (comm1 == MPI_COMM_NULL || comm2 == MPI_COMM_NULL) {
715 retval = MPI_ERR_COMM;
716 } else if (result == NULL) {
717 retval = MPI_ERR_ARG;
719 if (comm1 == comm2) { /* Same communicators means same groups */
723 smpi_group_compare(smpi_comm_group(comm1),
724 smpi_comm_group(comm2));
725 if (*result == MPI_IDENT) {
726 *result = MPI_CONGRUENT;
729 retval = MPI_SUCCESS;
735 int PMPI_Comm_dup(MPI_Comm comm, MPI_Comm * newcomm)
740 if (comm == MPI_COMM_NULL) {
741 retval = MPI_ERR_COMM;
742 } else if (newcomm == NULL) {
743 retval = MPI_ERR_ARG;
745 *newcomm = smpi_comm_new(smpi_comm_group(comm));
746 retval = MPI_SUCCESS;
752 int PMPI_Comm_create(MPI_Comm comm, MPI_Group group, MPI_Comm * newcomm)
757 if (comm == MPI_COMM_NULL) {
758 retval = MPI_ERR_COMM;
759 } else if (group == MPI_GROUP_NULL) {
760 retval = MPI_ERR_GROUP;
761 } else if (newcomm == NULL) {
762 retval = MPI_ERR_ARG;
764 *newcomm = smpi_comm_new(group);
765 retval = MPI_SUCCESS;
771 int PMPI_Comm_free(MPI_Comm * comm)
777 retval = MPI_ERR_ARG;
778 } else if (*comm == MPI_COMM_NULL) {
779 retval = MPI_ERR_COMM;
781 smpi_comm_destroy(*comm);
782 *comm = MPI_COMM_NULL;
783 retval = MPI_SUCCESS;
789 int PMPI_Comm_disconnect(MPI_Comm * comm)
791 /* TODO: wait until all communication in comm are done */
796 retval = MPI_ERR_ARG;
797 } else if (*comm == MPI_COMM_NULL) {
798 retval = MPI_ERR_COMM;
800 smpi_comm_destroy(*comm);
801 *comm = MPI_COMM_NULL;
802 retval = MPI_SUCCESS;
808 int PMPI_Comm_split(MPI_Comm comm, int color, int key, MPI_Comm* comm_out)
813 if (comm_out == NULL) {
814 retval = MPI_ERR_ARG;
815 } else if (comm == MPI_COMM_NULL) {
816 retval = MPI_ERR_COMM;
818 *comm_out = smpi_comm_split(comm, color, key);
819 retval = MPI_SUCCESS;
825 int PMPI_Send_init(void *buf, int count, MPI_Datatype datatype, int dst,
826 int tag, MPI_Comm comm, MPI_Request * request)
831 if (request == NULL) {
832 retval = MPI_ERR_ARG;
833 } else if (comm == MPI_COMM_NULL) {
834 retval = MPI_ERR_COMM;
835 } else if (dst == MPI_PROC_NULL) {
836 retval = MPI_SUCCESS;
838 *request = smpi_mpi_send_init(buf, count, datatype, dst, tag, comm);
839 retval = MPI_SUCCESS;
845 int PMPI_Recv_init(void *buf, int count, MPI_Datatype datatype, int src,
846 int tag, MPI_Comm comm, MPI_Request * request)
851 if (request == NULL) {
852 retval = MPI_ERR_ARG;
853 } else if (comm == MPI_COMM_NULL) {
854 retval = MPI_ERR_COMM;
855 } else if (src == MPI_PROC_NULL) {
856 retval = MPI_SUCCESS;
858 *request = smpi_mpi_recv_init(buf, count, datatype, src, tag, comm);
859 retval = MPI_SUCCESS;
865 int PMPI_Ssend_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm, MPI_Request* request) {
869 if (request == NULL) {
870 retval = MPI_ERR_ARG;
871 } else if (comm == MPI_COMM_NULL) {
872 retval = MPI_ERR_COMM;
873 } else if (dst == MPI_PROC_NULL) {
874 retval = MPI_SUCCESS;
876 *request = smpi_mpi_ssend_init(buf, count, datatype, dst, tag, comm);
877 retval = MPI_SUCCESS;
883 int PMPI_Start(MPI_Request * request)
888 if (request == NULL || *request == MPI_REQUEST_NULL) {
889 retval = MPI_ERR_ARG;
891 smpi_mpi_start(*request);
892 retval = MPI_SUCCESS;
898 int PMPI_Startall(int count, MPI_Request * requests)
903 if (requests == NULL) {
904 retval = MPI_ERR_ARG;
906 smpi_mpi_startall(count, requests);
907 retval = MPI_SUCCESS;
913 int PMPI_Request_free(MPI_Request * request)
918 if (*request == MPI_REQUEST_NULL) {
919 retval = MPI_ERR_ARG;
921 if((*request)->flags & PERSISTENT)(*request)->refcount--;
922 smpi_mpi_request_free(request);
923 retval = MPI_SUCCESS;
929 int PMPI_Irecv(void *buf, int count, MPI_Datatype datatype, int src,
930 int tag, MPI_Comm comm, MPI_Request * request)
936 if (request == NULL) {
937 retval = MPI_ERR_ARG;
938 } else if (comm == MPI_COMM_NULL) {
939 retval = MPI_ERR_COMM;
940 } else if (src == MPI_PROC_NULL) {
941 *request = MPI_REQUEST_NULL;
942 retval = MPI_SUCCESS;
943 } else if (src!=MPI_ANY_SOURCE && (src >= smpi_group_size(smpi_comm_group(comm)) || src <0)){
944 retval = MPI_ERR_COMM;
945 } else if (count < 0) {
946 retval = MPI_ERR_COUNT;
947 } else if (buf==NULL && count > 0) {
948 retval = MPI_ERR_COUNT;
949 } else if (datatype == MPI_DATATYPE_NULL){
950 retval = MPI_ERR_TYPE;
951 } else if(tag<0 && tag != MPI_ANY_TAG){
952 retval = MPI_ERR_TAG;
956 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
957 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
958 TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
961 *request = smpi_mpi_irecv(buf, count, datatype, src, tag, comm);
962 retval = MPI_SUCCESS;
965 TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
966 (*request)->recv = 1;
975 int PMPI_Isend(void *buf, int count, MPI_Datatype datatype, int dst,
976 int tag, MPI_Comm comm, MPI_Request * request)
981 if (request == NULL) {
982 retval = MPI_ERR_ARG;
983 } else if (comm == MPI_COMM_NULL) {
984 retval = MPI_ERR_COMM;
985 } else if (dst == MPI_PROC_NULL) {
986 *request = MPI_REQUEST_NULL;
987 retval = MPI_SUCCESS;
988 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
989 retval = MPI_ERR_COMM;
990 } else if (count < 0) {
991 retval = MPI_ERR_COUNT;
992 } else if (buf==NULL && count > 0) {
993 retval = MPI_ERR_COUNT;
994 } else if (datatype == MPI_DATATYPE_NULL){
995 retval = MPI_ERR_TYPE;
996 } else if(tag<0 && tag != MPI_ANY_TAG){
997 retval = MPI_ERR_TAG;
1001 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1002 TRACE_smpi_computing_out(rank);
1003 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1004 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
1005 TRACE_smpi_send(rank, rank, dst_traced);
1008 *request = smpi_mpi_isend(buf, count, datatype, dst, tag, comm);
1009 retval = MPI_SUCCESS;
1012 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
1013 (*request)->send = 1;
1014 TRACE_smpi_computing_in(rank);
1022 int PMPI_Issend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm, MPI_Request* request) {
1026 if (request == NULL) {
1027 retval = MPI_ERR_ARG;
1028 } else if (comm == MPI_COMM_NULL) {
1029 retval = MPI_ERR_COMM;
1030 } else if (dst == MPI_PROC_NULL) {
1031 *request = MPI_REQUEST_NULL;
1032 retval = MPI_SUCCESS;
1033 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
1034 retval = MPI_ERR_COMM;
1035 } else if (count < 0) {
1036 retval = MPI_ERR_COUNT;
1037 } else if (buf==NULL && count > 0) {
1038 retval = MPI_ERR_COUNT;
1039 } else if (datatype == MPI_DATATYPE_NULL){
1040 retval = MPI_ERR_TYPE;
1041 } else if(tag<0 && tag != MPI_ANY_TAG){
1042 retval = MPI_ERR_TAG;
1046 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1047 TRACE_smpi_computing_out(rank);
1048 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1049 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
1050 TRACE_smpi_send(rank, rank, dst_traced);
1053 *request = smpi_mpi_issend(buf, count, datatype, dst, tag, comm);
1054 retval = MPI_SUCCESS;
1057 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
1058 (*request)->send = 1;
1059 TRACE_smpi_computing_in(rank);
1067 int PMPI_Recv(void *buf, int count, MPI_Datatype datatype, int src, int tag,
1068 MPI_Comm comm, MPI_Status * status)
1073 if (comm == MPI_COMM_NULL) {
1074 retval = MPI_ERR_COMM;
1075 } else if (src == MPI_PROC_NULL) {
1076 smpi_empty_status(status);
1077 status->MPI_SOURCE = MPI_PROC_NULL;
1078 retval = MPI_SUCCESS;
1079 } else if (src!=MPI_ANY_SOURCE && (src >= smpi_group_size(smpi_comm_group(comm)) || src <0)){
1080 retval = MPI_ERR_COMM;
1081 } else if (count < 0) {
1082 retval = MPI_ERR_COUNT;
1083 } else if (buf==NULL && count > 0) {
1084 retval = MPI_ERR_COUNT;
1085 } else if (datatype == MPI_DATATYPE_NULL){
1086 retval = MPI_ERR_TYPE;
1087 } else if(tag<0 && tag != MPI_ANY_TAG){
1088 retval = MPI_ERR_TAG;
1091 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1092 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
1093 TRACE_smpi_computing_out(rank);
1095 TRACE_smpi_ptp_in(rank, src_traced, rank, __FUNCTION__);
1098 smpi_mpi_recv(buf, count, datatype, src, tag, comm, status);
1099 retval = MPI_SUCCESS;
1102 //the src may not have been known at the beginning of the recv (MPI_ANY_SOURCE)
1103 if(status!=MPI_STATUS_IGNORE)src_traced = smpi_group_index(smpi_comm_group(comm), status->MPI_SOURCE);
1104 TRACE_smpi_ptp_out(rank, src_traced, rank, __FUNCTION__);
1105 TRACE_smpi_recv(rank, src_traced, rank);
1106 TRACE_smpi_computing_in(rank);
1114 int PMPI_Send(void *buf, int count, MPI_Datatype datatype, int dst, int tag,
1121 if (comm == MPI_COMM_NULL) {
1122 retval = MPI_ERR_COMM;
1123 } else if (dst == MPI_PROC_NULL) {
1124 retval = MPI_SUCCESS;
1125 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
1126 retval = MPI_ERR_COMM;
1127 } else if (count < 0) {
1128 retval = MPI_ERR_COUNT;
1129 } else if (buf==NULL && count > 0) {
1130 retval = MPI_ERR_COUNT;
1131 } else if (datatype == MPI_DATATYPE_NULL){
1132 retval = MPI_ERR_TYPE;
1133 } else if(tag<0 && tag != MPI_ANY_TAG){
1134 retval = MPI_ERR_TAG;
1138 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1139 TRACE_smpi_computing_out(rank);
1140 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1141 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
1142 TRACE_smpi_send(rank, rank, dst_traced);
1145 smpi_mpi_send(buf, count, datatype, dst, tag, comm);
1146 retval = MPI_SUCCESS;
1149 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
1150 TRACE_smpi_computing_in(rank);
1160 int PMPI_Ssend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm) {
1165 if (comm == MPI_COMM_NULL) {
1166 retval = MPI_ERR_COMM;
1167 } else if (dst == MPI_PROC_NULL) {
1168 retval = MPI_SUCCESS;
1169 } else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0){
1170 retval = MPI_ERR_COMM;
1171 } else if (count < 0) {
1172 retval = MPI_ERR_COUNT;
1173 } else if (buf==NULL && count > 0) {
1174 retval = MPI_ERR_COUNT;
1175 } else if (datatype == MPI_DATATYPE_NULL){
1176 retval = MPI_ERR_TYPE;
1177 } else if(tag<0 && tag != MPI_ANY_TAG){
1178 retval = MPI_ERR_TAG;
1182 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1183 TRACE_smpi_computing_out(rank);
1184 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1185 TRACE_smpi_ptp_in(rank, rank, dst_traced, __FUNCTION__);
1186 TRACE_smpi_send(rank, rank, dst_traced);
1189 smpi_mpi_ssend(buf, count, datatype, dst, tag, comm);
1190 retval = MPI_SUCCESS;
1193 TRACE_smpi_ptp_out(rank, rank, dst_traced, __FUNCTION__);
1194 TRACE_smpi_computing_in(rank);
1202 int PMPI_Sendrecv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1203 int dst, int sendtag, void *recvbuf, int recvcount,
1204 MPI_Datatype recvtype, int src, int recvtag,
1205 MPI_Comm comm, MPI_Status * status)
1211 if (comm == MPI_COMM_NULL) {
1212 retval = MPI_ERR_COMM;
1213 } else if (sendtype == MPI_DATATYPE_NULL
1214 || recvtype == MPI_DATATYPE_NULL) {
1215 retval = MPI_ERR_TYPE;
1216 } else if (src == MPI_PROC_NULL || dst == MPI_PROC_NULL) {
1217 smpi_empty_status(status);
1218 status->MPI_SOURCE = MPI_PROC_NULL;
1219 retval = MPI_SUCCESS;
1220 }else if (dst >= smpi_group_size(smpi_comm_group(comm)) || dst <0 ||
1221 (src!=MPI_ANY_SOURCE && (src >= smpi_group_size(smpi_comm_group(comm)) || src <0))){
1222 retval = MPI_ERR_COMM;
1223 } else if (sendcount < 0 || recvcount<0) {
1224 retval = MPI_ERR_COUNT;
1225 } else if ((sendbuf==NULL && sendcount > 0)||(recvbuf==NULL && recvcount>0)) {
1226 retval = MPI_ERR_COUNT;
1227 } else if((sendtag<0 && sendtag != MPI_ANY_TAG)||(recvtag<0 && recvtag != MPI_ANY_TAG)){
1228 retval = MPI_ERR_TAG;
1232 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1233 TRACE_smpi_computing_out(rank);
1234 int dst_traced = smpi_group_index(smpi_comm_group(comm), dst);
1235 int src_traced = smpi_group_index(smpi_comm_group(comm), src);
1236 TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
1237 TRACE_smpi_send(rank, rank, dst_traced);
1238 TRACE_smpi_send(rank, src_traced, rank);
1242 smpi_mpi_sendrecv(sendbuf, sendcount, sendtype, dst, sendtag, recvbuf,
1243 recvcount, recvtype, src, recvtag, comm, status);
1244 retval = MPI_SUCCESS;
1247 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1248 TRACE_smpi_recv(rank, rank, dst_traced);
1249 TRACE_smpi_recv(rank, src_traced, rank);
1250 TRACE_smpi_computing_in(rank);
1259 int PMPI_Sendrecv_replace(void *buf, int count, MPI_Datatype datatype,
1260 int dst, int sendtag, int src, int recvtag,
1261 MPI_Comm comm, MPI_Status * status)
1263 //TODO: suboptimal implementation
1266 if ((datatype == MPI_DATATYPE_NULL)||(datatype->has_subtype==1)) {
1267 retval = MPI_ERR_TYPE;
1268 } else if (count < 0) {
1269 retval = MPI_ERR_COUNT;
1271 int size = smpi_datatype_size(datatype) * count;
1272 recvbuf = xbt_new(char, size);
1274 MPI_Sendrecv(buf, count, datatype, dst, sendtag, recvbuf, count,
1275 datatype, src, recvtag, comm, status);
1276 if(retval==MPI_SUCCESS){
1277 memcpy(buf, recvbuf, size * sizeof(char));
1285 int PMPI_Test(MPI_Request * request, int *flag, MPI_Status * status)
1290 if (request == MPI_REQUEST_NULL || flag == NULL) {
1291 retval = MPI_ERR_ARG;
1292 } else if (*request == MPI_REQUEST_NULL) {
1294 retval = MPI_ERR_REQUEST;
1296 *flag = smpi_mpi_test(request, status);
1297 retval = MPI_SUCCESS;
1303 int PMPI_Testany(int count, MPI_Request requests[], int *index, int *flag,
1304 MPI_Status * status)
1309 if (index == NULL || flag == NULL) {
1310 retval = MPI_ERR_ARG;
1312 *flag = smpi_mpi_testany(count, requests, index, status);
1313 retval = MPI_SUCCESS;
1319 int PMPI_Testall(int count, MPI_Request* requests, int* flag, MPI_Status* statuses)
1325 retval = MPI_ERR_ARG;
1327 *flag = smpi_mpi_testall(count, requests, statuses);
1328 retval = MPI_SUCCESS;
1334 int PMPI_Probe(int source, int tag, MPI_Comm comm, MPI_Status* status) {
1338 if (status == NULL) {
1339 retval = MPI_ERR_ARG;
1340 } else if (comm == MPI_COMM_NULL) {
1341 retval = MPI_ERR_COMM;
1342 } else if (source == MPI_PROC_NULL) {
1343 smpi_empty_status(status);
1344 status->MPI_SOURCE = MPI_PROC_NULL;
1345 retval = MPI_SUCCESS;
1347 smpi_mpi_probe(source, tag, comm, status);
1348 retval = MPI_SUCCESS;
1355 int PMPI_Iprobe(int source, int tag, MPI_Comm comm, int* flag, MPI_Status* status) {
1360 retval = MPI_ERR_ARG;
1361 } else if (status == NULL) {
1362 retval = MPI_ERR_ARG;
1363 } else if (comm == MPI_COMM_NULL) {
1364 retval = MPI_ERR_COMM;
1365 } else if (source == MPI_PROC_NULL) {
1366 smpi_empty_status(status);
1367 status->MPI_SOURCE = MPI_PROC_NULL;
1368 retval = MPI_SUCCESS;
1370 smpi_mpi_iprobe(source, tag, comm, flag, status);
1371 retval = MPI_SUCCESS;
1377 int PMPI_Wait(MPI_Request * request, MPI_Status * status)
1383 if (request == NULL) {
1384 retval = MPI_ERR_ARG;
1385 } else if (*request == MPI_REQUEST_NULL) {
1386 retval = MPI_ERR_REQUEST;
1390 int rank = request && (*request)->comm != MPI_COMM_NULL
1391 ? smpi_process_index()
1393 TRACE_smpi_computing_out(rank);
1395 MPI_Group group = smpi_comm_group((*request)->comm);
1396 int src_traced = smpi_group_index(group, (*request)->src);
1397 int dst_traced = smpi_group_index(group, (*request)->dst);
1398 int is_wait_for_receive = (*request)->recv;
1399 TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
1402 smpi_mpi_wait(request, status);
1403 retval = MPI_SUCCESS;
1406 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1407 if (is_wait_for_receive) {
1408 TRACE_smpi_recv(rank, src_traced, dst_traced);
1410 TRACE_smpi_computing_in(rank);
1419 int PMPI_Waitany(int count, MPI_Request requests[], int *index, MPI_Status * status)
1425 //save requests information for tracing
1427 xbt_dynar_t srcs = xbt_dynar_new(sizeof(int), NULL);
1428 xbt_dynar_t dsts = xbt_dynar_new(sizeof(int), NULL);
1429 xbt_dynar_t recvs = xbt_dynar_new(sizeof(int), NULL);
1430 for (i = 0; i < count; i++) {
1431 MPI_Request req = requests[i]; //already received requests are no longer valid
1433 int *asrc = xbt_new(int, 1);
1434 int *adst = xbt_new(int, 1);
1435 int *arecv = xbt_new(int, 1);
1439 xbt_dynar_insert_at(srcs, i, asrc);
1440 xbt_dynar_insert_at(dsts, i, adst);
1441 xbt_dynar_insert_at(recvs, i, arecv);
1446 int *t = xbt_new(int, 1);
1447 xbt_dynar_insert_at(srcs, i, t);
1448 xbt_dynar_insert_at(dsts, i, t);
1449 xbt_dynar_insert_at(recvs, i, t);
1453 int rank_traced = smpi_process_index();
1454 TRACE_smpi_computing_out(rank_traced);
1456 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1459 if (index == NULL) {
1460 retval = MPI_ERR_ARG;
1462 *index = smpi_mpi_waitany(count, requests, status);
1463 retval = MPI_SUCCESS;
1466 if(*index!=MPI_UNDEFINED){
1467 int src_traced, dst_traced, is_wait_for_receive;
1468 xbt_dynar_get_cpy(srcs, *index, &src_traced);
1469 xbt_dynar_get_cpy(dsts, *index, &dst_traced);
1470 xbt_dynar_get_cpy(recvs, *index, &is_wait_for_receive);
1471 if (is_wait_for_receive) {
1472 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1474 TRACE_smpi_ptp_out(rank_traced, src_traced, dst_traced, __FUNCTION__);
1475 //clean-up of dynars
1476 xbt_dynar_free(&srcs);
1477 xbt_dynar_free(&dsts);
1478 xbt_dynar_free(&recvs);
1480 TRACE_smpi_computing_in(rank_traced);
1486 int PMPI_Waitall(int count, MPI_Request requests[], MPI_Status status[])
1491 //save information from requests
1493 xbt_dynar_t srcs = xbt_dynar_new(sizeof(int), NULL);
1494 xbt_dynar_t dsts = xbt_dynar_new(sizeof(int), NULL);
1495 xbt_dynar_t recvs = xbt_dynar_new(sizeof(int), NULL);
1496 for (i = 0; i < count; i++) {
1497 MPI_Request req = requests[i];
1499 int *asrc = xbt_new(int, 1);
1500 int *adst = xbt_new(int, 1);
1501 int *arecv = xbt_new(int, 1);
1505 xbt_dynar_insert_at(srcs, i, asrc);
1506 xbt_dynar_insert_at(dsts, i, adst);
1507 xbt_dynar_insert_at(recvs, i, arecv);
1512 int *t = xbt_new(int, 1);
1513 xbt_dynar_insert_at(srcs, i, t);
1514 xbt_dynar_insert_at(dsts, i, t);
1515 xbt_dynar_insert_at(recvs, i, t);
1519 int rank_traced = smpi_process_index();
1520 TRACE_smpi_computing_out(rank_traced);
1522 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1524 int retval = smpi_mpi_waitall(count, requests, status);
1526 for (i = 0; i < count; i++) {
1527 int src_traced, dst_traced, is_wait_for_receive;
1528 xbt_dynar_get_cpy(srcs, i, &src_traced);
1529 xbt_dynar_get_cpy(dsts, i, &dst_traced);
1530 xbt_dynar_get_cpy(recvs, i, &is_wait_for_receive);
1531 if (is_wait_for_receive) {
1532 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1535 TRACE_smpi_ptp_out(rank_traced, -1, -1, __FUNCTION__);
1536 //clean-up of dynars
1537 xbt_dynar_free(&srcs);
1538 xbt_dynar_free(&dsts);
1539 xbt_dynar_free(&recvs);
1540 TRACE_smpi_computing_in(rank_traced);
1546 int PMPI_Waitsome(int incount, MPI_Request requests[], int *outcount,
1547 int *indices, MPI_Status status[])
1552 if (outcount == NULL || indices == NULL) {
1553 retval = MPI_ERR_ARG;
1555 *outcount = smpi_mpi_waitsome(incount, requests, indices, status);
1556 retval = MPI_SUCCESS;
1562 int PMPI_Testsome(int incount, MPI_Request requests[], int* outcount,
1563 int* indices, MPI_Status status[])
1568 if (outcount == NULL || indices == NULL) {
1569 retval = MPI_ERR_ARG;
1571 *outcount = smpi_mpi_testsome(incount, requests, indices, status);
1572 retval = MPI_SUCCESS;
1579 int PMPI_Bcast(void *buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm)
1585 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1586 TRACE_smpi_computing_out(rank);
1587 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1588 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1590 if (comm == MPI_COMM_NULL) {
1591 retval = MPI_ERR_COMM;
1593 smpi_mpi_bcast(buf, count, datatype, root, comm);
1594 retval = MPI_SUCCESS;
1597 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1598 TRACE_smpi_computing_in(rank);
1604 int PMPI_Barrier(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;
1617 smpi_mpi_barrier(comm);
1618 retval = MPI_SUCCESS;
1621 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1622 TRACE_smpi_computing_in(rank);
1628 int PMPI_Gather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1629 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1630 int root, MPI_Comm comm)
1636 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1637 TRACE_smpi_computing_out(rank);
1638 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1639 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1641 if (comm == MPI_COMM_NULL) {
1642 retval = MPI_ERR_COMM;
1643 } else if (sendtype == MPI_DATATYPE_NULL
1644 || recvtype == MPI_DATATYPE_NULL) {
1645 retval = MPI_ERR_TYPE;
1647 smpi_mpi_gather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1648 recvtype, root, comm);
1649 retval = MPI_SUCCESS;
1652 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1653 TRACE_smpi_computing_in(rank);
1659 int PMPI_Gatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1660 void *recvbuf, int *recvcounts, int *displs,
1661 MPI_Datatype recvtype, int root, MPI_Comm comm)
1667 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1668 TRACE_smpi_computing_out(rank);
1669 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1670 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1672 if (comm == MPI_COMM_NULL) {
1673 retval = MPI_ERR_COMM;
1674 } else if (sendtype == MPI_DATATYPE_NULL
1675 || recvtype == MPI_DATATYPE_NULL) {
1676 retval = MPI_ERR_TYPE;
1677 } else if (recvcounts == NULL || displs == NULL) {
1678 retval = MPI_ERR_ARG;
1680 smpi_mpi_gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1681 displs, recvtype, root, comm);
1682 retval = MPI_SUCCESS;
1685 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1686 TRACE_smpi_computing_in(rank);
1692 int PMPI_Allgather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1693 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1700 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1701 TRACE_smpi_computing_out(rank);
1702 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1704 if (comm == MPI_COMM_NULL) {
1705 retval = MPI_ERR_COMM;
1706 } else if (sendtype == MPI_DATATYPE_NULL
1707 || recvtype == MPI_DATATYPE_NULL) {
1708 retval = MPI_ERR_TYPE;
1710 smpi_mpi_allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1712 retval = MPI_SUCCESS;
1715 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1721 int PMPI_Allgatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1722 void *recvbuf, int *recvcounts, int *displs,
1723 MPI_Datatype recvtype, MPI_Comm comm)
1729 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1730 TRACE_smpi_computing_out(rank);
1731 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1733 if (comm == MPI_COMM_NULL) {
1734 retval = MPI_ERR_COMM;
1735 } else if (sendtype == MPI_DATATYPE_NULL
1736 || recvtype == MPI_DATATYPE_NULL) {
1737 retval = MPI_ERR_TYPE;
1738 } else if (recvcounts == NULL || displs == NULL) {
1739 retval = MPI_ERR_ARG;
1741 smpi_mpi_allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1742 displs, recvtype, comm);
1743 retval = MPI_SUCCESS;
1746 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1747 TRACE_smpi_computing_in(rank);
1753 int PMPI_Scatter(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1754 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1755 int root, MPI_Comm comm)
1761 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1762 TRACE_smpi_computing_out(rank);
1763 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1765 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1767 if (comm == MPI_COMM_NULL) {
1768 retval = MPI_ERR_COMM;
1769 } else if (sendtype == MPI_DATATYPE_NULL
1770 || recvtype == MPI_DATATYPE_NULL) {
1771 retval = MPI_ERR_TYPE;
1773 smpi_mpi_scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1774 recvtype, root, comm);
1775 retval = MPI_SUCCESS;
1778 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1779 TRACE_smpi_computing_in(rank);
1785 int PMPI_Scatterv(void *sendbuf, int *sendcounts, int *displs,
1786 MPI_Datatype sendtype, void *recvbuf, int recvcount,
1787 MPI_Datatype recvtype, int root, MPI_Comm comm)
1793 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1794 TRACE_smpi_computing_out(rank);
1795 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1796 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1798 if (comm == MPI_COMM_NULL) {
1799 retval = MPI_ERR_COMM;
1800 } else if (sendtype == MPI_DATATYPE_NULL
1801 || recvtype == MPI_DATATYPE_NULL) {
1802 retval = MPI_ERR_TYPE;
1803 } else if (sendcounts == NULL || displs == NULL) {
1804 retval = MPI_ERR_ARG;
1806 smpi_mpi_scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf,
1807 recvcount, recvtype, root, comm);
1808 retval = MPI_SUCCESS;
1811 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1812 TRACE_smpi_computing_in(rank);
1818 int PMPI_Reduce(void *sendbuf, void *recvbuf, int count,
1819 MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm)
1825 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1826 TRACE_smpi_computing_out(rank);
1827 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1828 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1830 if (comm == MPI_COMM_NULL) {
1831 retval = MPI_ERR_COMM;
1832 } else if (datatype == MPI_DATATYPE_NULL || op == MPI_OP_NULL) {
1833 retval = MPI_ERR_ARG;
1835 smpi_mpi_reduce(sendbuf, recvbuf, count, datatype, op, root, comm);
1836 retval = MPI_SUCCESS;
1839 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1840 TRACE_smpi_computing_in(rank);
1846 int PMPI_Allreduce(void *sendbuf, void *recvbuf, int count,
1847 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1853 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1854 TRACE_smpi_computing_out(rank);
1855 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1857 if (comm == MPI_COMM_NULL) {
1858 retval = MPI_ERR_COMM;
1859 } else if (datatype == MPI_DATATYPE_NULL) {
1860 retval = MPI_ERR_TYPE;
1861 } else if (op == MPI_OP_NULL) {
1862 retval = MPI_ERR_OP;
1864 smpi_mpi_allreduce(sendbuf, recvbuf, count, datatype, op, comm);
1865 retval = MPI_SUCCESS;
1868 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1869 TRACE_smpi_computing_in(rank);
1875 int PMPI_Scan(void *sendbuf, void *recvbuf, int count,
1876 MPI_Datatype datatype, MPI_Op op, 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 (datatype == MPI_DATATYPE_NULL) {
1889 retval = MPI_ERR_TYPE;
1890 } else if (op == MPI_OP_NULL) {
1891 retval = MPI_ERR_OP;
1893 smpi_mpi_scan(sendbuf, recvbuf, count, datatype, op, comm);
1894 retval = MPI_SUCCESS;
1897 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1898 TRACE_smpi_computing_in(rank);
1904 int PMPI_Reduce_scatter(void *sendbuf, void *recvbuf, int *recvcounts,
1905 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1907 int retval, i, size, count;
1909 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1913 TRACE_smpi_computing_out(rank);
1914 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1916 if (comm == MPI_COMM_NULL) {
1917 retval = MPI_ERR_COMM;
1918 } else if (datatype == MPI_DATATYPE_NULL) {
1919 retval = MPI_ERR_TYPE;
1920 } else if (op == MPI_OP_NULL) {
1921 retval = MPI_ERR_OP;
1922 } else if (recvcounts == NULL) {
1923 retval = MPI_ERR_ARG;
1925 /* arbitrarily choose root as rank 0 */
1926 /* TODO: faster direct implementation ? */
1927 size = smpi_comm_size(comm);
1929 displs = xbt_new(int, size);
1930 for (i = 0; i < size; i++) {
1931 count += recvcounts[i];
1934 smpi_mpi_reduce(sendbuf, recvbuf, count, datatype, op, 0, comm);
1935 smpi_mpi_scatterv(recvbuf, recvcounts, displs, datatype, recvbuf,
1936 recvcounts[rank], datatype, 0, comm);
1938 retval = MPI_SUCCESS;
1941 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1942 TRACE_smpi_computing_in(rank);
1948 int PMPI_Alltoall(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1949 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1952 int retval, size, sendsize;
1956 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1957 TRACE_smpi_computing_out(rank);
1958 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1960 if (comm == MPI_COMM_NULL) {
1961 retval = MPI_ERR_COMM;
1962 } else if (sendtype == MPI_DATATYPE_NULL
1963 || recvtype == MPI_DATATYPE_NULL) {
1964 retval = MPI_ERR_TYPE;
1966 size = smpi_comm_size(comm);
1967 sendsize = smpi_datatype_size(sendtype) * sendcount;
1968 if (sendsize < 200 && size > 12) {
1970 smpi_coll_tuned_alltoall_bruck(sendbuf, sendcount, sendtype,
1971 recvbuf, recvcount, recvtype,
1973 } else if (sendsize < 3000) {
1975 smpi_coll_tuned_alltoall_basic_linear(sendbuf, sendcount,
1977 recvcount, recvtype, comm);
1980 smpi_coll_tuned_alltoall_pairwise(sendbuf, sendcount, sendtype,
1981 recvbuf, recvcount, recvtype,
1986 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1987 TRACE_smpi_computing_in(rank);
1993 int PMPI_Alltoallv(void *sendbuf, int *sendcounts, int *senddisps,
1994 MPI_Datatype sendtype, void *recvbuf, int *recvcounts,
1995 int *recvdisps, MPI_Datatype recvtype, MPI_Comm comm)
2001 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
2002 TRACE_smpi_computing_out(rank);
2003 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
2005 if (comm == MPI_COMM_NULL) {
2006 retval = MPI_ERR_COMM;
2007 } else if (sendtype == MPI_DATATYPE_NULL
2008 || recvtype == MPI_DATATYPE_NULL) {
2009 retval = MPI_ERR_TYPE;
2010 } else if (sendcounts == NULL || senddisps == NULL || recvcounts == NULL
2011 || recvdisps == NULL) {
2012 retval = MPI_ERR_ARG;
2015 smpi_coll_basic_alltoallv(sendbuf, sendcounts, senddisps, sendtype,
2016 recvbuf, recvcounts, recvdisps, recvtype,
2020 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
2021 TRACE_smpi_computing_in(rank);
2028 int PMPI_Get_processor_name(char *name, int *resultlen)
2030 int retval = MPI_SUCCESS;
2033 strncpy(name, SIMIX_host_get_name(SIMIX_host_self()),
2034 strlen(SIMIX_host_get_name(SIMIX_host_self())) < MPI_MAX_PROCESSOR_NAME - 1 ?
2035 strlen(SIMIX_host_get_name(SIMIX_host_self())) +1 :
2036 MPI_MAX_PROCESSOR_NAME - 1 );
2039 MPI_MAX_PROCESSOR_NAME ? MPI_MAX_PROCESSOR_NAME : strlen(name);
2045 int PMPI_Get_count(MPI_Status * status, MPI_Datatype datatype, int *count)
2047 int retval = MPI_SUCCESS;
2051 if (status == NULL || count == NULL) {
2052 retval = MPI_ERR_ARG;
2053 } else if (datatype == MPI_DATATYPE_NULL) {
2054 retval = MPI_ERR_TYPE;
2056 size = smpi_datatype_size(datatype);
2059 } else if (status->count % size != 0) {
2060 retval = MPI_UNDEFINED;
2062 *count = smpi_mpi_get_count(status, datatype);
2069 int PMPI_Type_contiguous(int count, MPI_Datatype old_type, MPI_Datatype* new_type) {
2073 if (old_type == MPI_DATATYPE_NULL) {
2074 retval = MPI_ERR_TYPE;
2075 } else if (count<0){
2076 retval = MPI_ERR_COUNT;
2078 retval = smpi_datatype_contiguous(count, old_type, new_type);
2084 int PMPI_Type_commit(MPI_Datatype* datatype) {
2088 if (datatype == MPI_DATATYPE_NULL) {
2089 retval = MPI_ERR_TYPE;
2091 smpi_datatype_commit(datatype);
2092 retval = MPI_SUCCESS;
2099 int PMPI_Type_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
2103 if (old_type == MPI_DATATYPE_NULL) {
2104 retval = MPI_ERR_TYPE;
2105 } else if (count<0 || blocklen<0){
2106 retval = MPI_ERR_COUNT;
2108 retval = smpi_datatype_vector(count, blocklen, stride, old_type, new_type);
2114 int PMPI_Type_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
2118 if (old_type == MPI_DATATYPE_NULL) {
2119 retval = MPI_ERR_TYPE;
2120 } else if (count<0 || blocklen<0){
2121 retval = MPI_ERR_COUNT;
2123 retval = smpi_datatype_hvector(count, blocklen, stride, old_type, new_type);
2130 int PMPI_Type_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2134 if (old_type == MPI_DATATYPE_NULL) {
2135 retval = MPI_ERR_TYPE;
2136 } else if (count<0){
2137 retval = MPI_ERR_COUNT;
2139 retval = smpi_datatype_indexed(count, blocklens, indices, old_type, new_type);
2145 int PMPI_Type_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2149 if (old_type == MPI_DATATYPE_NULL) {
2150 retval = MPI_ERR_TYPE;
2151 } else if (count<0){
2152 retval = MPI_ERR_COUNT;
2154 retval = smpi_datatype_hindexed(count, blocklens, indices, old_type, new_type);
2161 int PMPI_Type_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type) {
2166 retval = MPI_ERR_COUNT;
2168 retval = smpi_datatype_struct(count, blocklens, indices, old_types, new_type);
2173 int PMPI_Error_class(int errorcode, int* errorclass) {
2174 // assume smpi uses only standard mpi error codes
2175 *errorclass=errorcode;
2180 int PMPI_Initialized(int* flag) {
2181 *flag=(smpi_process_data()!=NULL);
2185 /* The following calls are not yet implemented and will fail at runtime. */
2186 /* Once implemented, please move them above this notice. */
2188 static int not_yet_implemented(void) {
2189 XBT_WARN("Not yet implemented");
2193 int PMPI_Pack_size(int incount, MPI_Datatype datatype, MPI_Comm comm, int* size) {
2194 return not_yet_implemented();
2197 int PMPI_Cart_coords(MPI_Comm comm, int rank, int maxdims, int* coords) {
2198 return not_yet_implemented();
2201 int PMPI_Cart_create(MPI_Comm comm_old, int ndims, int* dims, int* periods, int reorder, MPI_Comm* comm_cart) {
2202 return not_yet_implemented();
2205 int PMPI_Cart_get(MPI_Comm comm, int maxdims, int* dims, int* periods, int* coords) {
2206 return not_yet_implemented();
2209 int PMPI_Cart_map(MPI_Comm comm_old, int ndims, int* dims, int* periods, int* newrank) {
2210 return not_yet_implemented();
2213 int PMPI_Cart_rank(MPI_Comm comm, int* coords, int* rank) {
2214 return not_yet_implemented();
2217 int PMPI_Cart_shift(MPI_Comm comm, int direction, int displ, int* source, int* dest) {
2218 return not_yet_implemented();
2221 int PMPI_Cart_sub(MPI_Comm comm, int* remain_dims, MPI_Comm* comm_new) {
2222 return not_yet_implemented();
2225 int PMPI_Cartdim_get(MPI_Comm comm, int* ndims) {
2226 return not_yet_implemented();
2229 int PMPI_Graph_create(MPI_Comm comm_old, int nnodes, int* index, int* edges, int reorder, MPI_Comm* comm_graph) {
2230 return not_yet_implemented();
2233 int PMPI_Graph_get(MPI_Comm comm, int maxindex, int maxedges, int* index, int* edges) {
2234 return not_yet_implemented();
2237 int PMPI_Graph_map(MPI_Comm comm_old, int nnodes, int* index, int* edges, int* newrank) {
2238 return not_yet_implemented();
2241 int PMPI_Graph_neighbors(MPI_Comm comm, int rank, int maxneighbors, int* neighbors) {
2242 return not_yet_implemented();
2245 int PMPI_Graph_neighbors_count(MPI_Comm comm, int rank, int* nneighbors) {
2246 return not_yet_implemented();
2249 int PMPI_Graphdims_get(MPI_Comm comm, int* nnodes, int* nedges) {
2250 return not_yet_implemented();
2253 int PMPI_Topo_test(MPI_Comm comm, int* top_type) {
2254 return not_yet_implemented();
2257 int PMPI_Errhandler_create(MPI_Handler_function* function, MPI_Errhandler* errhandler) {
2258 return not_yet_implemented();
2261 int PMPI_Errhandler_free(MPI_Errhandler* errhandler) {
2262 return not_yet_implemented();
2265 int PMPI_Errhandler_get(MPI_Comm comm, MPI_Errhandler* errhandler) {
2266 return not_yet_implemented();
2269 int PMPI_Error_string(int errorcode, char* string, int* resultlen) {
2270 return not_yet_implemented();
2273 int PMPI_Errhandler_set(MPI_Comm comm, MPI_Errhandler errhandler) {
2274 return not_yet_implemented();
2278 int PMPI_Cancel(MPI_Request* request) {
2279 return not_yet_implemented();
2282 int PMPI_Buffer_attach(void* buffer, int size) {
2283 return not_yet_implemented();
2286 int PMPI_Buffer_detach(void* buffer, int* size) {
2287 return not_yet_implemented();
2290 int PMPI_Comm_test_inter(MPI_Comm comm, int* flag) {
2291 return not_yet_implemented();
2294 int PMPI_Comm_get_attr (MPI_Comm comm, int comm_keyval, void *attribute_val, int *flag)
2296 return not_yet_implemented();
2299 int PMPI_Pcontrol(const int level )
2301 return not_yet_implemented();
2304 int PMPI_Unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Datatype type, MPI_Comm comm) {
2305 return not_yet_implemented();
2310 int PMPI_Intercomm_create(MPI_Comm local_comm, int local_leader, MPI_Comm peer_comm, int remote_leader, int tag, MPI_Comm* comm_out) {
2311 return not_yet_implemented();
2314 int PMPI_Intercomm_merge(MPI_Comm comm, int high, MPI_Comm* comm_out) {
2315 return not_yet_implemented();
2318 int PMPI_Bsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2319 return not_yet_implemented();
2322 int PMPI_Bsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2323 return not_yet_implemented();
2326 int PMPI_Ibsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2327 return not_yet_implemented();
2330 int PMPI_Comm_remote_group(MPI_Comm comm, MPI_Group* group) {
2331 return not_yet_implemented();
2334 int PMPI_Comm_remote_size(MPI_Comm comm, int* size) {
2335 return not_yet_implemented();
2338 int PMPI_Attr_delete(MPI_Comm comm, int keyval) {
2339 return not_yet_implemented();
2342 int PMPI_Attr_get(MPI_Comm comm, int keyval, void* attr_value, int* flag) {
2343 return not_yet_implemented();
2346 int PMPI_Attr_put(MPI_Comm comm, int keyval, void* attr_value) {
2347 return not_yet_implemented();
2350 int PMPI_Rsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2351 return not_yet_implemented();
2354 int PMPI_Rsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2355 return not_yet_implemented();
2358 int PMPI_Irsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2359 return not_yet_implemented();
2362 int PMPI_Keyval_create(MPI_Copy_function* copy_fn, MPI_Delete_function* delete_fn, int* keyval, void* extra_state) {
2363 return not_yet_implemented();
2366 int PMPI_Keyval_free(int* keyval) {
2367 return not_yet_implemented();
2370 int PMPI_Test_cancelled(MPI_Status* status, int* flag) {
2371 return not_yet_implemented();
2374 int PMPI_Pack(void* inbuf, int incount, MPI_Datatype type, void* outbuf, int outcount, int* position, MPI_Comm comm) {
2375 return not_yet_implemented();
2378 int PMPI_Get_elements(MPI_Status* status, MPI_Datatype datatype, int* elements) {
2379 return not_yet_implemented();
2382 int PMPI_Dims_create(int nnodes, int ndims, int* dims) {
2383 return not_yet_implemented();