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

This commit is contained in:
Stan Moore
2025-02-13 12:41:16 -07:00
parent ea2eb930c5
commit 583f7b70b4

View File

@ -793,7 +793,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);
}