gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix an odd character in the warmup

Browse Source
This commit is contained in:
Alejandro Gallo 2023-01-23 14:43:17 +01:00
parent be96e4bf8c
commit c8bdc4239f
1 changed files with 24 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
include/atrip/SliceUnion.hpp
View File

@ -465,30 +465,30 @@ template <typename F=double>
#if defined(HAVE_CUDA) #if defined(HAVE_CUDA)
LOG(1,"Atrip") << "warming communication up " << slices.size() << "\n"; LOG(1,"Atrip") << "warming communication up " << slices.size() << "\n";
 WITH_CHRONO("cuda:warmup", WITH_CHRONO("cuda:warmup",
int nRanks=Atrip::np, requestCount=0; int nRanks=Atrip::np, requestCount=0;
int nSends=sliceBuffers.size()*nRanks; int nSends=sliceBuffers.size()*nRanks;
MPI_Request *requests = (MPI_Request*) malloc(nSends*2 * sizeof(MPI_Request)); MPI_Request *requests = (MPI_Request*) malloc(nSends*2 * sizeof(MPI_Request));
MPI_Status *statuses = (MPI_Status*) malloc(nSends*2 * sizeof(MPI_Status)); MPI_Status *statuses = (MPI_Status*) malloc(nSends*2 * sizeof(MPI_Status));
for (int sliceId=0; sliceId<sliceBuffers.size(); sliceId++){ for (int sliceId=0; sliceId<sliceBuffers.size(); sliceId++){
for (int rankId=0; rankId<nRanks; rankId++){ for (int rankId=0; rankId<nRanks; rankId++){
MPI_Isend((void*)SOURCES_DATA(sources[0]), MPI_Isend((void*)SOURCES_DATA(sources[0]),
sliceSize, sliceSize,
traits::mpi::datatypeOf<F>(), traits::mpi::datatypeOf<F>(),
rankId, rankId,
100, 100,
universe, universe,
&requests[requestCount++]); &requests[requestCount++]);
MPI_Irecv((void*)sliceBuffers[sliceId], MPI_Irecv((void*)sliceBuffers[sliceId],
sliceSize, sliceSize,
traits::mpi::datatypeOf<F>(), traits::mpi::datatypeOf<F>(),
rankId, rankId,
100, 100,
universe, universe,
&requests[requestCount++]); &requests[requestCount++]);
} }
} }
MPI_Waitall(nSends*2, requests, statuses); MPI_Waitall(nSends*2, requests, statuses);
) )
#endif #endif