ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

GPU Package: Switching to parallel GPU initialization / JIT compilation.

Browse Source
This commit is contained in:
W. Michael Brown
2022-10-07 13:25:14 -07:00
parent f715f174bb
commit 5a98a38e24
60 changed files with 78 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/gpu/lal_ufm_ext.cpp
View File

@ -78,7 +78,7 @@ int ufml_gpu_init(const int ntypes, double **cutsq, double **host_uf1,
offset, special_lj, inum, nall, max_nbors, maxspecial,
cell_size, gpu_split, screen);
UFMLMF.device->gpu_barrier();
UFMLMF.device->serialize_init();
if (message)
fprintf(screen,"Done.\n");
}
@ -106,7 +106,7 @@ void ufml_gpu_reinit(const int ntypes, double **cutsq, double **host_uf1,
for (int i=0; i<procs_per_gpu; i++) {
if (gpu_rank==i && world_me!=0)
UFMLMF.reinit(ntypes, cutsq, host_uf1, host_uf2, host_uf3, offset);
UFMLMF.device->gpu_barrier();
UFMLMF.device->serialize_init();
}
}