1 /* Copyright (c) 2009-2022. 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. */
13 int main(int argc, char *argv[])
20 MPI_Init(&argc, &argv);
21 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
22 MPI_Comm_size(MPI_COMM_WORLD, &size);
23 MPI_Comm_set_errhandler(MPI_COMM_WORLD, MPI_ERRORS_RETURN);
25 int* recv_counts = (int *) xbt_malloc(size * sizeof(int));
26 int* recv_disps = (int *) xbt_malloc(size * sizeof(int));
29 for (i = 0; i < size; i++) {
30 recv_counts[i] = i + 1;
31 recv_disps[i] = recv_sb_size;
32 recv_sb_size += i + 1;
35 int* sb = (int *) xbt_malloc(recv_counts[rank] * sizeof(int));
36 int* rb = (int *) xbt_malloc(recv_sb_size * sizeof(int));
38 status = MPI_Allgatherv(NULL, recv_counts[rank], MPI_INT, rb, recv_counts, recv_disps, MPI_INT, MPI_COMM_WORLD);
39 if(status!=MPI_ERR_BUFFER)
40 printf("MPI_Allgatherv did not return MPI_ERR_BUFFER for empty sendbuf\n");
41 status = MPI_Allgatherv(sb, -1, MPI_INT, rb, recv_counts, recv_disps, MPI_INT, MPI_COMM_WORLD);
42 if(status!=MPI_ERR_COUNT)
43 printf("MPI_Allgatherv did not return MPI_ERR_COUNT for -1 sendcount\n");
44 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_DATATYPE_NULL, rb, recv_counts, recv_disps, MPI_INT, MPI_COMM_WORLD);
45 if(status!=MPI_ERR_TYPE)
46 printf("MPI_Allgatherv did not return MPI_ERR_TYPE for MPI_DATATYPE_NULL sendtype\n");
47 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, NULL, recv_counts, recv_disps, MPI_INT, MPI_COMM_WORLD);
48 if(status!=MPI_ERR_BUFFER)
49 printf("MPI_Allgatherv did not return MPI_ERR_BUFFER for empty recvbuf\n");
50 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, rb, NULL, recv_disps, MPI_INT, MPI_COMM_WORLD);
51 if(status!=MPI_ERR_COUNT)
52 printf("MPI_Allgatherv did not return MPI_ERR_COUNT for NULL recvcounts\n");
53 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, rb, recv_counts, NULL, MPI_INT, MPI_COMM_WORLD);
54 if(status!=MPI_ERR_ARG)
55 printf("MPI_Allgatherv did not return MPI_ERR_ARG for NULL recvdisps\n");
56 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, rb, recv_counts, recv_disps, MPI_DATATYPE_NULL, MPI_COMM_WORLD);
57 if(status!=MPI_ERR_TYPE)
58 printf("MPI_Allgatherv did not return MPI_ERR_TYPE for MPI_DATATYPE_NULL recvtype\n");
59 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, rb, recv_counts, recv_disps, MPI_INT, MPI_COMM_NULL);
60 if(status!=MPI_ERR_COMM)
61 printf("MPI_Allgatherv did not return MPI_ERR_COMM for MPI_COMM_NULL comm\n");
63 printf("[%d] sndbuf=[", rank);
64 for (i = 0; i < recv_counts[rank]; i++){
65 sb[i] = recv_disps[rank] + i;
70 for (i = 0; i < recv_sb_size; i++)
73 status = MPI_Allgatherv(sb, recv_counts[rank], MPI_INT, rb, recv_counts, recv_disps, MPI_INT, MPI_COMM_WORLD);
75 printf("[%d] rcvbuf=[", rank);
76 for (i = 0; i < recv_sb_size; i++)
80 if (rank == 0 && status != MPI_SUCCESS) {
81 printf("allgatherv returned %d\n", status);
86 xbt_free(recv_counts);