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

add n_global arg, print sizes in rank 0

Browse Source
This commit is contained in:
Bryce Allen
2022-10-23 18:51:17 +00:00
parent 74bfc20d50
commit 4143c5f06f
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
mpi_stencil_gt.cc
View File

@@ -125,9 +125,14 @@ void boundary_exchange(MPI_Comm comm, int world_size, int rank,
} }
int main(int argc, char **argv) { int main(int argc, char **argv) {
constexpr int n_global = 8 * 1024 * 1024; int n_global = 32 * 1024 * 1024;
constexpr int n_sten = 5;
constexpr int n_bnd = (n_sten - 1) / 2; if (argc > 1) {
n_global = std::atoi(argv[1]) * 1024 * 1024;
}
int n_sten = 5;
int n_bnd = (n_sten - 1) / 2;
int world_size, world_rank, device_id; int world_size, world_rank, device_id;
uint32_t vendor_id; uint32_t vendor_id;
@@ -143,6 +148,12 @@ int main(int argc, char **argv) {
device_id = gt::backend::clib::device_get(); device_id = gt::backend::clib::device_get();
vendor_id = gt::backend::clib::device_get_vendor_id(device_id); vendor_id = gt::backend::clib::device_get_vendor_id(device_id);
if (world_rank == 0) {
printf("n procs = %d\n", world_size);
printf("n_global = %d\n", n_global);
printf("n_local = %d\n", n_local);
}
auto h_y = gt::empty<double>({n_local_with_ghost}); auto h_y = gt::empty<double>({n_local_with_ghost});
auto d_y = gt::empty_device<double>({n_local_with_ghost}); auto d_y = gt::empty_device<double>({n_local_with_ghost});