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

print stage_host

Browse Source
This commit is contained in:
Bryce Allen
2022-10-24 18:12:09 +00:00
parent 7c332265d9
commit 2309afb2ab
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
mpi_stencil2d_gt.cc
View File

@@ -259,9 +259,10 @@ int main(int argc, char** argv)
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) { if (world_rank == 0) {
printf("n procs = %d\n", world_size); printf("n procs = %d\n", world_size);
printf("n_global = %d\n", n_global); printf("n_global = %d\n", n_global);
printf("n_local = %d\n", n_local); printf("n_local = %d\n", n_local);
printf("stage_host = %d\n", stage_host);
} }
auto h_z = gt::empty<double>({n_global, n_local_with_ghost}); auto h_z = gt::empty<double>({n_global, n_local_with_ghost});