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

gt: fix dim1 nobuff

Browse Source
This commit is contained in:
Bryce Allen
2023-03-25 13:48:40 -07:00
parent a885062c83
commit b00d52af2a
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mpi_stencil2d_gt.cc
View File

@@ -299,7 +299,7 @@ void boundary_exchange_y(MPI_Comm comm, int world_size, int rank,
} else { } else {
rbuf_l_data = rv_l.data().get(); rbuf_l_data = rv_l.data().get();
} }
CHECK(MPI_Irecv(rbuf_l_data, rbuf_l.size(), MPI_DOUBLE, rank_l, 123, comm, CHECK(MPI_Irecv(rbuf_l_data, rv_l.size(), MPI_DOUBLE, rank_l, 123, comm,
&req_l[0])); &req_l[0]));
} }
@@ -310,14 +310,14 @@ void boundary_exchange_y(MPI_Comm comm, int world_size, int rank,
} else { } else {
rbuf_r_data = rv_r.data().get(); rbuf_r_data = rv_r.data().get();
} }
CHECK(MPI_Irecv(rbuf_r_data, rbuf_r.size(), MPI_DOUBLE, rank_r, 456, comm, CHECK(MPI_Irecv(rbuf_r_data, rv_r.size(), MPI_DOUBLE, rank_r, 456, comm,
&req_r[0])); &req_r[0]));
} }
// wait for send buffer fill // wait for send buffer fill
if (stage_device) { // if (stage_device) {
gt::synchronize(); gt::synchronize();
} // }
// initiate async sends // initiate async sends
if (rank_l >= 0) { if (rank_l >= 0) {
@@ -327,7 +327,7 @@ void boundary_exchange_y(MPI_Comm comm, int world_size, int rank,
} else { } else {
sbuf_l_data = sv_l.data().get(); sbuf_l_data = sv_l.data().get();
} }
CHECK(MPI_Isend(sbuf_l_data, sbuf_l.size(), MPI_DOUBLE, rank_l, 456, comm, CHECK(MPI_Isend(sbuf_l_data, sv_l.size(), MPI_DOUBLE, rank_l, 456, comm,
&req_l[1])); &req_l[1]));
} }
@@ -338,7 +338,7 @@ void boundary_exchange_y(MPI_Comm comm, int world_size, int rank,
} else { } else {
sbuf_r_data = sv_r.data().get(); sbuf_r_data = sv_r.data().get();
} }
CHECK(MPI_Isend(sbuf_r_data, sbuf_r.size(), MPI_DOUBLE, rank_r, 123, comm, CHECK(MPI_Isend(sbuf_r_data, sv_r.size(), MPI_DOUBLE, rank_r, 123, comm,
&req_r[1])); &req_r[1]));
} }