portelli/lattice-benchmarks
3
1
Fork 2
Code Issues Pull Requests 2 Packages Projects Releases Wiki Activity

Grid comms warmup sequence

Browse Source
This commit is contained in:
Antonin Portelli 2023-02-03 20:59:20 +00:00
parent cc4c0255bc
commit ce0d4d9457
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Grid/Benchmark_Grid.cpp
View File

@ -168,6 +168,7 @@ class Benchmark
} }
double dbytes; double dbytes;
#define NWARMUP 50
for (int dir = 0; dir < 8; dir++) for (int dir = 0; dir < 8; dir++)
{ {
@ -176,6 +177,24 @@ class Benchmark
{ {
std::vector<double> times(Nloop); std::vector<double> times(Nloop);
for (int i = 0; i < NWARMUP; i++)
{
int xmit_to_rank;
int recv_from_rank;
if (dir == mu)
{
int comm_proc = 1;
Grid.ShiftedRanks(mu, comm_proc, xmit_to_rank, recv_from_rank);
}
else
{
int comm_proc = mpi_layout[mu] - 1;
Grid.ShiftedRanks(mu, comm_proc, xmit_to_rank, recv_from_rank);
}
Grid.SendToRecvFrom((void *)&xbuf[dir][0], xmit_to_rank,
(void *)&rbuf[dir][0], recv_from_rank, bytes);
}
for (int i = 0; i < Nloop; i++) for (int i = 0; i < Nloop; i++)
{ {
@ -249,7 +268,6 @@ class Benchmark
uint64_t NN; uint64_t NN;
uint64_t lmax = 64; uint64_t lmax = 64;
#define NLOOP (200 * lmax * lmax * lmax / lat / lat / lat) #define NLOOP (200 * lmax * lmax * lmax / lat / lat / lat)
#define NWARMUP 50
GridSerialRNG sRNG; GridSerialRNG sRNG;
sRNG.SeedFixedIntegers(std::vector<int>({45, 12, 81, 9})); sRNG.SeedFixedIntegers(std::vector<int>({45, 12, 81, 9}));