Fix issue in comm_kokkos

This commit is contained in:
Stan Moore 2019-04-10 10:17:14 -06:00
parent cf35ebe5fa
commit e4e2249b63
1 changed files with 3 additions and 2 deletions

View File

@ -617,8 +617,9 @@ void CommKokkos::exchange_device()
k_count.sync<LMPHostType>();
if (k_count.h_view()>=k_exchange_sendlist.h_view.extent(0)) {
k_exchange_sendlist.resize(k_count.h_view()*1.1);
k_exchange_copylist.resize(k_count.h_view()*1.1);
k_exchange_lists.resize(2,k_count.h_view()*1.1);
k_exchange_sendlist = Kokkos::subview(k_exchange_lists,0,Kokkos::ALL);
k_exchange_copylist = Kokkos::subview(k_exchange_lists,1,Kokkos::ALL);
k_count.h_view()=k_exchange_sendlist.h_view.extent(0);
}
}