Commit 529eeb60 authored by Stan Moore's avatar Stan Moore
Browse files

Reduce GPU data transfer

parent a86572f4

src/KOKKOS/comm_kokkos.cpp

+5 −3
Original line number Diff line number Diff line
@@ -916,8 +916,10 @@ void CommKokkos::borders_device() { 


  if (exec_space == Host) k_sendlist.sync<LMPDeviceType>();

  atomKK->modified(exec_space,ALL_MASK);

  if (map_style) {

    atomKK->sync(Host,TAG_MASK);

  if (map_style) atom->map_set();

    atom->map_set();

  }

}

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

   realloc the size of the send buffer as needed with BUFFACTOR and bufextra

src/KOKKOS/neighbor_kokkos.cpp

+1 −1
Original line number Diff line number Diff line
@@ -310,9 +310,9 @@ void NeighborKokkos::build_kokkos(int topoflag) 
  // build pairwise lists for all perpetual NPair/NeighList

  // grow() with nlocal/nall args so that only realloc if have to



  atomKK->sync(Host,ALL_MASK);

  for (i = 0; i < npair_perpetual; i++) {

    m = plist[i];

    if (!lists[m]->kokkos) atomKK->sync(Host,ALL_MASK);

    if (!lists[m]->copy) lists[m]->grow(nlocal,nall);

    neigh_pair[m]->build_setup();

    neigh_pair[m]->build(lists[m]);

CRA Git | Maintained and supported by SUSTech CRA and CCSE