print GPU device ID for MPI ranks within a node

2019-04-26 14:56:53 -04:00
parent d15f5aff86
commit fe5a110027
3 changed files with 1 additions and 2 deletions
--- a/gpu/Extras.cu
+++ b/gpu/Extras.cu
@@ -9,6 +9,7 @@ extern "C" int ScaLBL_SetDevice(int rank){
 	//int device = local_rank % n_devices; 
 	int device = rank % n_devices; 
 	cudaSetDevice(device); 
+	if (rank < n_devices) printf("MPI rank=%i will use GPU ID %i / %i \n",rank,device,n_devices);
 	return device;
 }