diff --git a/src/KOKKOS/atom_vec_atomic_kokkos.cpp b/src/KOKKOS/atom_vec_atomic_kokkos.cpp index e9b3115d16..4b50b9b9fd 100644 --- a/src/KOKKOS/atom_vec_atomic_kokkos.cpp +++ b/src/KOKKOS/atom_vec_atomic_kokkos.cpp @@ -1297,6 +1297,8 @@ void AtomVecAtomicKokkos::data_atom(double *coord, tagint imagetmp, h_v(nlocal,1) = 0.0; h_v(nlocal,2) = 0.0; + atomKK->modified(Host,ALL_MASK); + atom->nlocal++; } diff --git a/src/KOKKOS/atom_vec_charge_kokkos.cpp b/src/KOKKOS/atom_vec_charge_kokkos.cpp index 34e7bbae93..87e9ac1cf4 100644 --- a/src/KOKKOS/atom_vec_charge_kokkos.cpp +++ b/src/KOKKOS/atom_vec_charge_kokkos.cpp @@ -1383,6 +1383,8 @@ void AtomVecChargeKokkos::data_atom(double *coord, imageint imagetmp, h_v(nlocal,1) = 0.0; h_v(nlocal,2) = 0.0; + atomKK->modified(Host,ALL_MASK); + atom->nlocal++; } /* ----------------------------------------------------------------------