From fc94f1bd18ef0375fde631c82b0779c4165b0890 Mon Sep 17 00:00:00 2001 From: Stan Moore <stanmoore1@gmail.com> Date: Thu, 19 Jan 2017 11:55:22 -0700 Subject: [PATCH] Fixing GPU memory issues in Kokkos --- src/KOKKOS/domain_kokkos.cpp | 4 ++-- src/KOKKOS/verlet_kokkos.cpp | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/KOKKOS/domain_kokkos.cpp b/src/KOKKOS/domain_kokkos.cpp index cf65316ec9..5c1f1a60b9 100644 --- a/src/KOKKOS/domain_kokkos.cpp +++ b/src/KOKKOS/domain_kokkos.cpp @@ -354,7 +354,6 @@ void DomainKokkos::pbc() } atomKK->sync(Device,X_MASK|V_MASK|MASK_MASK|IMAGE_MASK); - atomKK->modified(Device,X_MASK|V_MASK|IMAGE_MASK); if (xperiodic || yperiodic || zperiodic) { if (deform_vremap) { @@ -385,8 +384,9 @@ void DomainKokkos::pbc() Kokkos::parallel_for(nlocal,f); } } - LMPDeviceType::fence(); + + atomKK->modified(Device,X_MASK|V_MASK|IMAGE_MASK); } /* ---------------------------------------------------------------------- diff --git a/src/KOKKOS/verlet_kokkos.cpp b/src/KOKKOS/verlet_kokkos.cpp index 20c4035276..53b4042376 100644 --- a/src/KOKKOS/verlet_kokkos.cpp +++ b/src/KOKKOS/verlet_kokkos.cpp @@ -170,7 +170,7 @@ void VerletKokkos::setup() modify->setup(vflag); output->setup(); - lmp->kokkos->auto_sync = 0; + lmp->kokkos->auto_sync = 1; update->setupflag = 1; } -- GitLab