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_lj_ext.cpp
View File

@ -76,7 +76,7 @@ int ljl_gpu_init(const int ntypes, double **cutsq, double **host_lj1,
offset, special_lj, inum, nall, max_nbors, maxspecial,
cell_size, gpu_split, screen);
LJLMF.device->gpu_barrier();
LJLMF.device->serialize_init();
if (message)
fprintf(screen,"Done.\n");
}
@ -105,7 +105,7 @@ void ljl_gpu_reinit(const int ntypes, double **cutsq, double **host_lj1,
for (int i=0; i<procs_per_gpu; i++) {
if (gpu_rank==i && world_me!=0)
LJLMF.reinit(ntypes, cutsq, host_lj1, host_lj2, host_lj3, host_lj4, offset);
LJLMF.device->gpu_barrier();
LJLMF.device->serialize_init();
}
}