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);
1241 smpi_mpi_sendrecv(sendbuf, sendcount, sendtype, dst, sendtag, recvbuf,
1242 recvcount, recvtype, src, recvtag, comm, status);
1243 retval = MPI_SUCCESS;
1246 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1247 TRACE_smpi_recv(rank, src_traced, rank);
1248 TRACE_smpi_computing_in(rank);
1257 int PMPI_Sendrecv_replace(void *buf, int count, MPI_Datatype datatype,
1258 int dst, int sendtag, int src, int recvtag,
1259 MPI_Comm comm, MPI_Status * status)
1261 //TODO: suboptimal implementation
1264 if (datatype == MPI_DATATYPE_NULL) {
1265 retval = MPI_ERR_TYPE;
1266 } else if (count < 0) {
1267 retval = MPI_ERR_COUNT;
1269 int size = smpi_datatype_get_extent(datatype) * count;
1270 recvbuf = xbt_new(char, size);
1272 MPI_Sendrecv(buf, count, datatype, dst, sendtag, recvbuf, count,
1273 datatype, src, recvtag, comm, status);
1274 if(retval==MPI_SUCCESS){
1275 smpi_datatype_copy(recvbuf, count, datatype, buf, count, datatype);
1283 int PMPI_Test(MPI_Request * request, int *flag, MPI_Status * status)
1288 if (request == MPI_REQUEST_NULL || flag == NULL) {
1289 retval = MPI_ERR_ARG;
1290 } else if (*request == MPI_REQUEST_NULL) {
1292 retval = MPI_ERR_REQUEST;
1294 *flag = smpi_mpi_test(request, status);
1295 retval = MPI_SUCCESS;
1301 int PMPI_Testany(int count, MPI_Request requests[], int *index, int *flag,
1302 MPI_Status * status)
1307 if (index == NULL || flag == NULL) {
1308 retval = MPI_ERR_ARG;
1310 *flag = smpi_mpi_testany(count, requests, index, status);
1311 retval = MPI_SUCCESS;
1317 int PMPI_Testall(int count, MPI_Request* requests, int* flag, MPI_Status* statuses)
1323 retval = MPI_ERR_ARG;
1325 *flag = smpi_mpi_testall(count, requests, statuses);
1326 retval = MPI_SUCCESS;
1332 int PMPI_Probe(int source, int tag, MPI_Comm comm, MPI_Status* status) {
1336 if (status == NULL) {
1337 retval = MPI_ERR_ARG;
1338 } else if (comm == MPI_COMM_NULL) {
1339 retval = MPI_ERR_COMM;
1340 } else if (source == MPI_PROC_NULL) {
1341 smpi_empty_status(status);
1342 status->MPI_SOURCE = MPI_PROC_NULL;
1343 retval = MPI_SUCCESS;
1345 smpi_mpi_probe(source, tag, comm, status);
1346 retval = MPI_SUCCESS;
1353 int PMPI_Iprobe(int source, int tag, MPI_Comm comm, int* flag, MPI_Status* status) {
1358 retval = MPI_ERR_ARG;
1359 } else if (status == NULL) {
1360 retval = MPI_ERR_ARG;
1361 } else if (comm == MPI_COMM_NULL) {
1362 retval = MPI_ERR_COMM;
1363 } else if (source == MPI_PROC_NULL) {
1364 smpi_empty_status(status);
1365 status->MPI_SOURCE = MPI_PROC_NULL;
1366 retval = MPI_SUCCESS;
1368 smpi_mpi_iprobe(source, tag, comm, flag, status);
1369 retval = MPI_SUCCESS;
1375 int PMPI_Wait(MPI_Request * request, MPI_Status * status)
1381 if (request == NULL) {
1382 retval = MPI_ERR_ARG;
1383 } else if (*request == MPI_REQUEST_NULL) {
1384 retval = MPI_ERR_REQUEST;
1388 int rank = request && (*request)->comm != MPI_COMM_NULL
1389 ? smpi_process_index()
1391 TRACE_smpi_computing_out(rank);
1393 MPI_Group group = smpi_comm_group((*request)->comm);
1394 int src_traced = (*request)->src;
1395 int dst_traced = (*request)->dst;
1396 int is_wait_for_receive = (*request)->recv;
1397 TRACE_smpi_ptp_in(rank, src_traced, dst_traced, __FUNCTION__);
1400 smpi_mpi_wait(request, status);
1401 retval = MPI_SUCCESS;
1404 TRACE_smpi_ptp_out(rank, src_traced, dst_traced, __FUNCTION__);
1405 if (is_wait_for_receive) {
1406 TRACE_smpi_recv(rank, src_traced, dst_traced);
1408 TRACE_smpi_computing_in(rank);
1417 int PMPI_Waitany(int count, MPI_Request requests[], int *index, MPI_Status * status)
1423 //save requests information for tracing
1425 int *srcs = xbt_new(int, count);
1426 int *dsts = xbt_new(int, count);
1427 int *recvs = xbt_new(int, count);
1428 for (i = 0; i < count; i++) {
1429 MPI_Request req = requests[i]; //already received requests are no longer valid
1433 recvs[i] = req->recv;
1436 int rank_traced = smpi_process_index();
1437 TRACE_smpi_computing_out(rank_traced);
1439 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1442 if (index == NULL) {
1443 retval = MPI_ERR_ARG;
1445 *index = smpi_mpi_waitany(count, requests, status);
1446 retval = MPI_SUCCESS;
1449 if(*index!=MPI_UNDEFINED){
1450 int src_traced = srcs[*index];
1451 int dst_traced = dsts[*index];
1452 int is_wait_for_receive = recvs[*index];
1453 if (is_wait_for_receive) {
1454 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1456 TRACE_smpi_ptp_out(rank_traced, src_traced, dst_traced, __FUNCTION__);
1461 TRACE_smpi_computing_in(rank_traced);
1467 int PMPI_Waitall(int count, MPI_Request requests[], MPI_Status status[])
1472 //save information from requests
1474 int *srcs = xbt_new(int, count);
1475 int *dsts = xbt_new(int, count);
1476 int *recvs = xbt_new(int, count);
1477 int valid_count = 0;
1478 for (i = 0; i < count; i++) {
1479 MPI_Request req = requests[i];
1480 if(req!=MPI_REQUEST_NULL){
1481 srcs[valid_count] = req->src;
1482 dsts[valid_count] = req->dst;
1483 recvs[valid_count] = req->recv;
1487 int rank_traced = smpi_process_index();
1488 TRACE_smpi_computing_out(rank_traced);
1490 TRACE_smpi_ptp_in(rank_traced, -1, -1, __FUNCTION__);
1492 int retval = smpi_mpi_waitall(count, requests, status);
1494 for (i = 0; i < valid_count; i++) {
1495 int src_traced = srcs[i];
1496 int dst_traced = dsts[i];
1497 int is_wait_for_receive = recvs[i];
1498 if (is_wait_for_receive) {
1499 TRACE_smpi_recv(rank_traced, src_traced, dst_traced);
1502 TRACE_smpi_ptp_out(rank_traced, -1, -1, __FUNCTION__);
1506 TRACE_smpi_computing_in(rank_traced);
1512 int PMPI_Waitsome(int incount, MPI_Request requests[], int *outcount,
1513 int *indices, MPI_Status status[])
1518 if (outcount == NULL || indices == NULL) {
1519 retval = MPI_ERR_ARG;
1521 *outcount = smpi_mpi_waitsome(incount, requests, indices, status);
1522 retval = MPI_SUCCESS;
1528 int PMPI_Testsome(int incount, MPI_Request requests[], int* outcount,
1529 int* indices, MPI_Status status[])
1534 if (outcount == NULL || indices == NULL) {
1535 retval = MPI_ERR_ARG;
1537 *outcount = smpi_mpi_testsome(incount, requests, indices, status);
1538 retval = MPI_SUCCESS;
1545 int PMPI_Bcast(void *buf, int count, MPI_Datatype datatype, int root, MPI_Comm comm)
1551 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1552 TRACE_smpi_computing_out(rank);
1553 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1554 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1556 if (comm == MPI_COMM_NULL) {
1557 retval = MPI_ERR_COMM;
1559 mpi_coll_bcast_fun(buf, count, datatype, root, comm);
1560 retval = MPI_SUCCESS;
1563 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1564 TRACE_smpi_computing_in(rank);
1570 int PMPI_Barrier(MPI_Comm comm)
1576 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1577 TRACE_smpi_computing_out(rank);
1578 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1580 if (comm == MPI_COMM_NULL) {
1581 retval = MPI_ERR_COMM;
1583 smpi_mpi_barrier(comm);
1584 retval = MPI_SUCCESS;
1587 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1588 TRACE_smpi_computing_in(rank);
1594 int PMPI_Gather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1595 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1596 int root, MPI_Comm comm)
1602 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1603 TRACE_smpi_computing_out(rank);
1604 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1605 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1607 if (comm == MPI_COMM_NULL) {
1608 retval = MPI_ERR_COMM;
1609 } else if (sendtype == MPI_DATATYPE_NULL
1610 || recvtype == MPI_DATATYPE_NULL) {
1611 retval = MPI_ERR_TYPE;
1613 smpi_mpi_gather(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1614 recvtype, root, comm);
1615 retval = MPI_SUCCESS;
1618 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1619 TRACE_smpi_computing_in(rank);
1625 int PMPI_Gatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1626 void *recvbuf, int *recvcounts, int *displs,
1627 MPI_Datatype recvtype, int root, MPI_Comm comm)
1633 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1634 TRACE_smpi_computing_out(rank);
1635 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1636 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1638 if (comm == MPI_COMM_NULL) {
1639 retval = MPI_ERR_COMM;
1640 } else if (sendtype == MPI_DATATYPE_NULL
1641 || recvtype == MPI_DATATYPE_NULL) {
1642 retval = MPI_ERR_TYPE;
1643 } else if (recvcounts == NULL || displs == NULL) {
1644 retval = MPI_ERR_ARG;
1646 smpi_mpi_gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1647 displs, recvtype, root, comm);
1648 retval = MPI_SUCCESS;
1651 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1652 TRACE_smpi_computing_in(rank);
1658 int PMPI_Allgather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1659 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1666 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1667 TRACE_smpi_computing_out(rank);
1668 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1670 if (comm == MPI_COMM_NULL) {
1671 retval = MPI_ERR_COMM;
1672 } else if (sendtype == MPI_DATATYPE_NULL
1673 || recvtype == MPI_DATATYPE_NULL) {
1674 retval = MPI_ERR_TYPE;
1676 mpi_coll_allgather_fun(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1678 retval = MPI_SUCCESS;
1681 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1687 int PMPI_Allgatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1688 void *recvbuf, int *recvcounts, int *displs,
1689 MPI_Datatype recvtype, MPI_Comm comm)
1695 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1696 TRACE_smpi_computing_out(rank);
1697 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1699 if (comm == MPI_COMM_NULL) {
1700 retval = MPI_ERR_COMM;
1701 } else if (sendtype == MPI_DATATYPE_NULL
1702 || recvtype == MPI_DATATYPE_NULL) {
1703 retval = MPI_ERR_TYPE;
1704 } else if (recvcounts == NULL || displs == NULL) {
1705 retval = MPI_ERR_ARG;
1707 mpi_coll_allgatherv_fun(sendbuf, sendcount, sendtype, recvbuf, recvcounts,
1708 displs, recvtype, comm);
1709 retval = MPI_SUCCESS;
1712 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1713 TRACE_smpi_computing_in(rank);
1719 int PMPI_Scatter(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1720 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1721 int root, MPI_Comm comm)
1727 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1728 TRACE_smpi_computing_out(rank);
1729 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1731 TRACE_smpi_collective_in(rank, root_traced, __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;
1739 smpi_mpi_scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount,
1740 recvtype, root, comm);
1741 retval = MPI_SUCCESS;
1744 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1745 TRACE_smpi_computing_in(rank);
1751 int PMPI_Scatterv(void *sendbuf, int *sendcounts, int *displs,
1752 MPI_Datatype sendtype, void *recvbuf, int recvcount,
1753 MPI_Datatype recvtype, int root, MPI_Comm comm)
1759 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1760 TRACE_smpi_computing_out(rank);
1761 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1762 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1764 if (comm == MPI_COMM_NULL) {
1765 retval = MPI_ERR_COMM;
1766 } else if (sendtype == MPI_DATATYPE_NULL
1767 || recvtype == MPI_DATATYPE_NULL) {
1768 retval = MPI_ERR_TYPE;
1769 } else if (sendcounts == NULL || displs == NULL) {
1770 retval = MPI_ERR_ARG;
1772 smpi_mpi_scatterv(sendbuf, sendcounts, displs, sendtype, recvbuf,
1773 recvcount, recvtype, root, comm);
1774 retval = MPI_SUCCESS;
1777 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1778 TRACE_smpi_computing_in(rank);
1784 int PMPI_Reduce(void *sendbuf, void *recvbuf, int count,
1785 MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm)
1791 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1792 TRACE_smpi_computing_out(rank);
1793 int root_traced = smpi_group_index(smpi_comm_group(comm), root);
1794 TRACE_smpi_collective_in(rank, root_traced, __FUNCTION__);
1796 if (comm == MPI_COMM_NULL) {
1797 retval = MPI_ERR_COMM;
1798 } else if (datatype == MPI_DATATYPE_NULL || op == MPI_OP_NULL) {
1799 retval = MPI_ERR_ARG;
1801 mpi_coll_reduce_fun(sendbuf, recvbuf, count, datatype, op, root, comm);
1802 retval = MPI_SUCCESS;
1805 TRACE_smpi_collective_out(rank, root_traced, __FUNCTION__);
1806 TRACE_smpi_computing_in(rank);
1812 int PMPI_Allreduce(void *sendbuf, void *recvbuf, int count,
1813 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1819 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1820 TRACE_smpi_computing_out(rank);
1821 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1823 if (comm == MPI_COMM_NULL) {
1824 retval = MPI_ERR_COMM;
1825 } else if (datatype == MPI_DATATYPE_NULL) {
1826 retval = MPI_ERR_TYPE;
1827 } else if (op == MPI_OP_NULL) {
1828 retval = MPI_ERR_OP;
1830 mpi_coll_allreduce_fun(sendbuf, recvbuf, count, datatype, op, comm);
1831 retval = MPI_SUCCESS;
1834 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1835 TRACE_smpi_computing_in(rank);
1841 int PMPI_Scan(void *sendbuf, void *recvbuf, int count,
1842 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1848 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1849 TRACE_smpi_computing_out(rank);
1850 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1852 if (comm == MPI_COMM_NULL) {
1853 retval = MPI_ERR_COMM;
1854 } else if (datatype == MPI_DATATYPE_NULL) {
1855 retval = MPI_ERR_TYPE;
1856 } else if (op == MPI_OP_NULL) {
1857 retval = MPI_ERR_OP;
1859 smpi_mpi_scan(sendbuf, recvbuf, count, datatype, op, comm);
1860 retval = MPI_SUCCESS;
1863 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1864 TRACE_smpi_computing_in(rank);
1870 int PMPI_Reduce_scatter(void *sendbuf, void *recvbuf, int *recvcounts,
1871 MPI_Datatype datatype, MPI_Op op, MPI_Comm comm)
1873 int retval, i, size, count;
1875 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1879 TRACE_smpi_computing_out(rank);
1880 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1882 if (comm == MPI_COMM_NULL) {
1883 retval = MPI_ERR_COMM;
1884 } else if (datatype == MPI_DATATYPE_NULL) {
1885 retval = MPI_ERR_TYPE;
1886 } else if (op == MPI_OP_NULL) {
1887 retval = MPI_ERR_OP;
1888 } else if (recvcounts == NULL) {
1889 retval = MPI_ERR_ARG;
1891 /* arbitrarily choose root as rank 0 */
1892 /* TODO: faster direct implementation ? */
1893 size = smpi_comm_size(comm);
1895 displs = xbt_new(int, size);
1896 for (i = 0; i < size; i++) {
1897 count += recvcounts[i];
1900 mpi_coll_reduce_fun(sendbuf, recvbuf, count, datatype, op, 0, comm);
1901 smpi_mpi_scatterv(recvbuf, recvcounts, displs, datatype, recvbuf,
1902 recvcounts[rank], datatype, 0, comm);
1904 retval = MPI_SUCCESS;
1907 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1908 TRACE_smpi_computing_in(rank);
1914 int PMPI_Alltoall(void *sendbuf, int sendcount, MPI_Datatype sendtype,
1915 void *recvbuf, int recvcount, MPI_Datatype recvtype,
1922 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1923 TRACE_smpi_computing_out(rank);
1924 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1926 if (comm == MPI_COMM_NULL) {
1927 retval = MPI_ERR_COMM;
1928 } else if (sendtype == MPI_DATATYPE_NULL
1929 || recvtype == MPI_DATATYPE_NULL) {
1930 retval = MPI_ERR_TYPE;
1932 retval = mpi_coll_alltoall_fun(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm);
1935 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1936 TRACE_smpi_computing_in(rank);
1942 int PMPI_Alltoallv(void *sendbuf, int *sendcounts, int *senddisps,
1943 MPI_Datatype sendtype, void *recvbuf, int *recvcounts,
1944 int *recvdisps, MPI_Datatype recvtype, MPI_Comm comm)
1950 int rank = comm != MPI_COMM_NULL ? smpi_process_index() : -1;
1951 TRACE_smpi_computing_out(rank);
1952 TRACE_smpi_collective_in(rank, -1, __FUNCTION__);
1954 if (comm == MPI_COMM_NULL) {
1955 retval = MPI_ERR_COMM;
1956 } else if (sendtype == MPI_DATATYPE_NULL
1957 || recvtype == MPI_DATATYPE_NULL) {
1958 retval = MPI_ERR_TYPE;
1959 } else if (sendcounts == NULL || senddisps == NULL || recvcounts == NULL
1960 || recvdisps == NULL) {
1961 retval = MPI_ERR_ARG;
1964 mpi_coll_alltoallv_fun(sendbuf, sendcounts, senddisps, sendtype,
1965 recvbuf, recvcounts, recvdisps, recvtype,
1969 TRACE_smpi_collective_out(rank, -1, __FUNCTION__);
1970 TRACE_smpi_computing_in(rank);
1977 int PMPI_Get_processor_name(char *name, int *resultlen)
1979 int retval = MPI_SUCCESS;
1982 strncpy(name, SIMIX_host_get_name(SIMIX_host_self()),
1983 strlen(SIMIX_host_get_name(SIMIX_host_self())) < MPI_MAX_PROCESSOR_NAME - 1 ?
1984 strlen(SIMIX_host_get_name(SIMIX_host_self())) +1 :
1985 MPI_MAX_PROCESSOR_NAME - 1 );
1988 MPI_MAX_PROCESSOR_NAME ? MPI_MAX_PROCESSOR_NAME : strlen(name);
1994 int PMPI_Get_count(MPI_Status * status, MPI_Datatype datatype, int *count)
1996 int retval = MPI_SUCCESS;
2000 if (status == NULL || count == NULL) {
2001 retval = MPI_ERR_ARG;
2002 } else if (datatype == MPI_DATATYPE_NULL) {
2003 retval = MPI_ERR_TYPE;
2005 size = smpi_datatype_size(datatype);
2008 } else if (status->count % size != 0) {
2009 retval = MPI_UNDEFINED;
2011 *count = smpi_mpi_get_count(status, datatype);
2018 int PMPI_Type_contiguous(int count, MPI_Datatype old_type, MPI_Datatype* new_type) {
2022 if (old_type == MPI_DATATYPE_NULL) {
2023 retval = MPI_ERR_TYPE;
2024 } else if (count<0){
2025 retval = MPI_ERR_COUNT;
2027 retval = smpi_datatype_contiguous(count, old_type, new_type);
2033 int PMPI_Type_commit(MPI_Datatype* datatype) {
2037 if (datatype == MPI_DATATYPE_NULL) {
2038 retval = MPI_ERR_TYPE;
2040 smpi_datatype_commit(datatype);
2041 retval = MPI_SUCCESS;
2048 int PMPI_Type_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
2052 if (old_type == MPI_DATATYPE_NULL) {
2053 retval = MPI_ERR_TYPE;
2054 } else if (count<0 || blocklen<0){
2055 retval = MPI_ERR_COUNT;
2057 retval = smpi_datatype_vector(count, blocklen, stride, old_type, new_type);
2063 int PMPI_Type_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type) {
2067 if (old_type == MPI_DATATYPE_NULL) {
2068 retval = MPI_ERR_TYPE;
2069 } else if (count<0 || blocklen<0){
2070 retval = MPI_ERR_COUNT;
2072 retval = smpi_datatype_hvector(count, blocklen, stride, old_type, new_type);
2079 int PMPI_Type_indexed(int count, int* blocklens, int* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2083 if (old_type == MPI_DATATYPE_NULL) {
2084 retval = MPI_ERR_TYPE;
2085 } else if (count<0){
2086 retval = MPI_ERR_COUNT;
2088 retval = smpi_datatype_indexed(count, blocklens, indices, old_type, new_type);
2094 int PMPI_Type_hindexed(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype old_type, MPI_Datatype* new_type) {
2098 if (old_type == MPI_DATATYPE_NULL) {
2099 retval = MPI_ERR_TYPE;
2100 } else if (count<0){
2101 retval = MPI_ERR_COUNT;
2103 retval = smpi_datatype_hindexed(count, blocklens, indices, old_type, new_type);
2110 int PMPI_Type_struct(int count, int* blocklens, MPI_Aint* indices, MPI_Datatype* old_types, MPI_Datatype* new_type) {
2115 retval = MPI_ERR_COUNT;
2117 retval = smpi_datatype_struct(count, blocklens, indices, old_types, new_type);
2122 int PMPI_Error_class(int errorcode, int* errorclass) {
2123 // assume smpi uses only standard mpi error codes
2124 *errorclass=errorcode;
2129 int PMPI_Initialized(int* flag) {
2130 *flag=(smpi_process_data()!=NULL);
2134 /* The following calls are not yet implemented and will fail at runtime. */
2135 /* Once implemented, please move them above this notice. */
2137 static int not_yet_implemented(void) {
2138 XBT_WARN("Not yet implemented");
2142 int PMPI_Pack_size(int incount, MPI_Datatype datatype, MPI_Comm comm, int* size) {
2143 return not_yet_implemented();
2146 int PMPI_Cart_coords(MPI_Comm comm, int rank, int maxdims, int* coords) {
2147 return not_yet_implemented();
2150 int PMPI_Cart_create(MPI_Comm comm_old, int ndims, int* dims, int* periods, int reorder, MPI_Comm* comm_cart) {
2151 return not_yet_implemented();
2154 int PMPI_Cart_get(MPI_Comm comm, int maxdims, int* dims, int* periods, int* coords) {
2155 return not_yet_implemented();
2158 int PMPI_Cart_map(MPI_Comm comm_old, int ndims, int* dims, int* periods, int* newrank) {
2159 return not_yet_implemented();
2162 int PMPI_Cart_rank(MPI_Comm comm, int* coords, int* rank) {
2163 return not_yet_implemented();
2166 int PMPI_Cart_shift(MPI_Comm comm, int direction, int displ, int* source, int* dest) {
2167 return not_yet_implemented();
2170 int PMPI_Cart_sub(MPI_Comm comm, int* remain_dims, MPI_Comm* comm_new) {
2171 return not_yet_implemented();
2174 int PMPI_Cartdim_get(MPI_Comm comm, int* ndims) {
2175 return not_yet_implemented();
2178 int PMPI_Graph_create(MPI_Comm comm_old, int nnodes, int* index, int* edges, int reorder, MPI_Comm* comm_graph) {
2179 return not_yet_implemented();
2182 int PMPI_Graph_get(MPI_Comm comm, int maxindex, int maxedges, int* index, int* edges) {
2183 return not_yet_implemented();
2186 int PMPI_Graph_map(MPI_Comm comm_old, int nnodes, int* index, int* edges, int* newrank) {
2187 return not_yet_implemented();
2190 int PMPI_Graph_neighbors(MPI_Comm comm, int rank, int maxneighbors, int* neighbors) {
2191 return not_yet_implemented();
2194 int PMPI_Graph_neighbors_count(MPI_Comm comm, int rank, int* nneighbors) {
2195 return not_yet_implemented();
2198 int PMPI_Graphdims_get(MPI_Comm comm, int* nnodes, int* nedges) {
2199 return not_yet_implemented();
2202 int PMPI_Topo_test(MPI_Comm comm, int* top_type) {
2203 return not_yet_implemented();
2206 int PMPI_Errhandler_create(MPI_Handler_function* function, MPI_Errhandler* errhandler) {
2207 return not_yet_implemented();
2210 int PMPI_Errhandler_free(MPI_Errhandler* errhandler) {
2211 return not_yet_implemented();
2214 int PMPI_Errhandler_get(MPI_Comm comm, MPI_Errhandler* errhandler) {
2215 return not_yet_implemented();
2218 int PMPI_Error_string(int errorcode, char* string, int* resultlen) {
2219 return not_yet_implemented();
2222 int PMPI_Errhandler_set(MPI_Comm comm, MPI_Errhandler errhandler) {
2223 return not_yet_implemented();
2227 int PMPI_Cancel(MPI_Request* request) {
2228 return not_yet_implemented();
2231 int PMPI_Buffer_attach(void* buffer, int size) {
2232 return not_yet_implemented();
2235 int PMPI_Buffer_detach(void* buffer, int* size) {
2236 return not_yet_implemented();
2239 int PMPI_Comm_test_inter(MPI_Comm comm, int* flag) {
2240 return not_yet_implemented();
2243 int PMPI_Comm_get_attr (MPI_Comm comm, int comm_keyval, void *attribute_val, int *flag)
2245 return not_yet_implemented();
2248 int PMPI_Pcontrol(const int level )
2250 return not_yet_implemented();
2253 int PMPI_Unpack(void* inbuf, int insize, int* position, void* outbuf, int outcount, MPI_Datatype type, MPI_Comm comm) {
2254 return not_yet_implemented();
2259 int PMPI_Intercomm_create(MPI_Comm local_comm, int local_leader, MPI_Comm peer_comm, int remote_leader, int tag, MPI_Comm* comm_out) {
2260 return not_yet_implemented();
2263 int PMPI_Intercomm_merge(MPI_Comm comm, int high, MPI_Comm* comm_out) {
2264 return not_yet_implemented();
2267 int PMPI_Bsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2268 return not_yet_implemented();
2271 int PMPI_Bsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2272 return not_yet_implemented();
2275 int PMPI_Ibsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2276 return not_yet_implemented();
2279 int PMPI_Comm_remote_group(MPI_Comm comm, MPI_Group* group) {
2280 return not_yet_implemented();
2283 int PMPI_Comm_remote_size(MPI_Comm comm, int* size) {
2284 return not_yet_implemented();
2287 int PMPI_Attr_delete(MPI_Comm comm, int keyval) {
2288 return not_yet_implemented();
2291 int PMPI_Attr_get(MPI_Comm comm, int keyval, void* attr_value, int* flag) {
2292 return not_yet_implemented();
2295 int PMPI_Attr_put(MPI_Comm comm, int keyval, void* attr_value) {
2296 return not_yet_implemented();
2299 int PMPI_Rsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm) {
2300 return not_yet_implemented();
2303 int PMPI_Rsend_init(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2304 return not_yet_implemented();
2307 int PMPI_Irsend(void* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm, MPI_Request* request) {
2308 return not_yet_implemented();
2311 int PMPI_Keyval_create(MPI_Copy_function* copy_fn, MPI_Delete_function* delete_fn, int* keyval, void* extra_state) {
2312 return not_yet_implemented();
2315 int PMPI_Keyval_free(int* keyval) {
2316 return not_yet_implemented();
2319 int PMPI_Test_cancelled(MPI_Status* status, int* flag) {
2320 return not_yet_implemented();
2323 int PMPI_Pack(void* inbuf, int incount, MPI_Datatype type, void* outbuf, int outcount, int* position, MPI_Comm comm) {
2324 return not_yet_implemented();
2327 int PMPI_Get_elements(MPI_Status* status, MPI_Datatype datatype, int* elements) {
2328 return not_yet_implemented();
2331 int PMPI_Dims_create(int nnodes, int ndims, int* dims) {
2332 return not_yet_implemented();
2335 int PMPI_Win_fence( int assert, MPI_Win win){
2336 return not_yet_implemented();
2339 int PMPI_Win_free( MPI_Win* win){
2340 return not_yet_implemented();
2343 int PMPI_Win_create( void *base, MPI_Aint size, int disp_unit, MPI_Info info, MPI_Comm comm, MPI_Win *win){
2344 return not_yet_implemented();
2347 int PMPI_Info_create( MPI_Info *info){
2348 return not_yet_implemented();
2351 int PMPI_Info_set( MPI_Info *info, char *key, char *value){
2352 return not_yet_implemented();
2355 int PMPI_Info_free( MPI_Info *info){
2356 return not_yet_implemented();
2359 int PMPI_Get( void *origin_addr, int origin_count, MPI_Datatype origin_datatype, int target_rank,
2360 MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Win win){
2361 return not_yet_implemented();