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

Fix rare bug in KOKKOS, manifest when load balancing on GPUs and exchange comm is on host

Browse Source
This commit is contained in:
Stan Moore
2025-02-13 12:41:16 -07:00
committed by Axel Kohlmeyer
parent 07f56405a4
commit bae0fe562e
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/KOKKOS/comm_kokkos.cpp
View File

@ -768,7 +768,10 @@ void CommKokkos::exchange()
}
atomKK->sync(Host,ALL_MASK);
int prev_auto_sync = lmp->kokkos->auto_sync;
lmp->kokkos->auto_sync = 1;
CommBrick::exchange();
lmp->kokkos->auto_sync = prev_auto_sync;
atomKK->modified(Host,ALL_MASK);
}