git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@5537 f3b2605a-c512-4ea7-a41b-209d697bcdaa
This commit is contained in:
@ -46,7 +46,7 @@ bool ljl_gpu_init(const int ntypes, double **cutsq,
|
||||
LJLMF.device->init_message(screen,"lj/cut",first_gpu,last_gpu);
|
||||
|
||||
bool message=false;
|
||||
if (world_me==0 && screen)
|
||||
if (LJLMF.device->replica_me()==0 && screen)
|
||||
message=true;
|
||||
|
||||
if (message) {
|
||||
@ -62,14 +62,14 @@ bool ljl_gpu_init(const int ntypes, double **cutsq,
|
||||
return false;
|
||||
}
|
||||
|
||||
MPI_Barrier(MPI_COMM_WORLD);
|
||||
LJLMF.device->world_barrier();
|
||||
if (message)
|
||||
fprintf(screen,"Done.\n");
|
||||
|
||||
for (int i=0; i<procs_per_gpu; i++) {
|
||||
if (message) {
|
||||
if (last_gpu-first_gpu==0)
|
||||
fprintf(screen,"Initializing GPU %d on core %d...",gpu_rank,i);
|
||||
fprintf(screen,"Initializing GPU %d on core %d...",first_gpu,i);
|
||||
else
|
||||
fprintf(screen,"Initializing GPUs %d-%d on core %d...",first_gpu,
|
||||
last_gpu,i);
|
||||
@ -83,7 +83,7 @@ bool ljl_gpu_init(const int ntypes, double **cutsq,
|
||||
if (!init_ok)
|
||||
return false;
|
||||
}
|
||||
MPI_Barrier(LJLMF.device->gpu_comm);
|
||||
LJLMF.device->gpu_barrier();
|
||||
if (message)
|
||||
fprintf(screen,"Done.\n");
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user