Commit fc94f1bd authored by Stan Moore's avatar Stan Moore
Browse files

Fixing GPU memory issues in Kokkos

parent 5ce8e2fb

src/KOKKOS/domain_kokkos.cpp

+2 −2
Original line number Diff line number Diff line
@@ -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);

}



/* ----------------------------------------------------------------------

src/KOKKOS/verlet_kokkos.cpp

+1 −1
Original line number Diff line number Diff line
@@ -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;

}

CRA Git | Maintained and supported by SUSTech CRA and CCSE