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

Fix execution space issues in Kokkos EAM pairstyles

parent 557e5b96

src/KOKKOS/pair_eam_alloy_kokkos.cpp

+2 −2
Original line number Diff line number Diff line
@@ -405,7 +405,7 @@ int PairEAMAlloyKokkos<DeviceType>::pack_forward_comm_kokkos(int n, DAT::tdual_i 
  d_sendlist = k_sendlist.view<DeviceType>();

  iswap = iswap_in;

  v_buf = buf.view<DeviceType>();

  Kokkos::parallel_for(Kokkos::RangePolicy<LMPDeviceType, TagPairEAMAlloyPackForwardComm>(0,n),*this);

  Kokkos::parallel_for(Kokkos::RangePolicy<DeviceType, TagPairEAMAlloyPackForwardComm>(0,n),*this);

  return n;

}
@@ -423,7 +423,7 @@ void PairEAMAlloyKokkos<DeviceType>::unpack_forward_comm_kokkos(int n, int first 
{

  first = first_in;

  v_buf = buf.view<DeviceType>();

  Kokkos::parallel_for(Kokkos::RangePolicy<LMPDeviceType, TagPairEAMAlloyUnpackForwardComm>(0,n),*this);

  Kokkos::parallel_for(Kokkos::RangePolicy<DeviceType, TagPairEAMAlloyUnpackForwardComm>(0,n),*this);

}



template<class DeviceType>

src/KOKKOS/pair_eam_fs_kokkos.cpp

+2 −2
Original line number Diff line number Diff line
@@ -405,7 +405,7 @@ int PairEAMFSKokkos<DeviceType>::pack_forward_comm_kokkos(int n, DAT::tdual_int_ 
  d_sendlist = k_sendlist.view<DeviceType>();

  iswap = iswap_in;

  v_buf = buf.view<DeviceType>();

  Kokkos::parallel_for(Kokkos::RangePolicy<LMPDeviceType, TagPairEAMFSPackForwardComm>(0,n),*this);

  Kokkos::parallel_for(Kokkos::RangePolicy<DeviceType, TagPairEAMFSPackForwardComm>(0,n),*this);

  return n;

}
@@ -423,7 +423,7 @@ void PairEAMFSKokkos<DeviceType>::unpack_forward_comm_kokkos(int n, int first_in 
{

  first = first_in;

  v_buf = buf.view<DeviceType>();

  Kokkos::parallel_for(Kokkos::RangePolicy<LMPDeviceType, TagPairEAMFSUnpackForwardComm>(0,n),*this);

  Kokkos::parallel_for(Kokkos::RangePolicy<DeviceType, TagPairEAMFSUnpackForwardComm>(0,n),*this);

}



template<class DeviceType>

CRA Git | Maintained and supported by SUSTech CRA and CCSE