bda/gpu-mpi-tests
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix allgather recv size

Browse Source
This commit is contained in:
Bryce Allen
2020-08-06 07:42:46 -04:00
parent 4d504dd5b1
commit 3e99cf443b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mpi_daxpy_nvtx.cc
View File

@@ -185,10 +185,10 @@ int main(int argc, char **argv) {
nvtxRangePushA("allGather"); nvtxRangePushA("allGather");
nvtxRangePushA("x"); nvtxRangePushA("x");
MPI_Allgather(x, n, MPI_DOUBLE, allx, n*world_size, MPI_DOUBLE, MPI_COMM_WORLD); MPI_Allgather(x, n, MPI_DOUBLE, allx, n, MPI_DOUBLE, MPI_COMM_WORLD);
nvtxRangePop(); nvtxRangePop();
nvtxRangePushA("y"); nvtxRangePushA("y");
MPI_Allgather(y, n, MPI_DOUBLE, ally, n*world_size, MPI_DOUBLE, MPI_COMM_WORLD); MPI_Allgather(y, n, MPI_DOUBLE, ally, n, MPI_DOUBLE, MPI_COMM_WORLD);
nvtxRangePop(); nvtxRangePop();
nvtxRangePop(); nvtxRangePop();