From 633ca33f2f9163a4bbe740801e28efa9d4a5d7ce Mon Sep 17 00:00:00 2001
From: Stan Moore <stamoor@sandia.gov>
Date: Tue, 25 Jul 2017 13:08:24 -0600
Subject: [PATCH] Fix issue in Kokkos neighborlist

---
 src/KOKKOS/npair_kokkos.cpp | 10 ++++++++--
 src/KOKKOS/npair_kokkos.h   |  2 +-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/KOKKOS/npair_kokkos.cpp b/src/KOKKOS/npair_kokkos.cpp
index b7b550369d..fd5f9373f2 100644
--- a/src/KOKKOS/npair_kokkos.cpp
+++ b/src/KOKKOS/npair_kokkos.cpp
@@ -201,14 +201,20 @@ if (GHOST) {
   if (newton_pair) {
     NPairKokkosBuildFunctor<DeviceType,TRI?0:HALF_NEIGH,1,TRI> f(data,atoms_per_bin * 5 * sizeof(X_FLOAT) * factor);
 #ifdef KOKKOS_HAVE_CUDA
-    Kokkos::parallel_for(config, f);
+    if (ExecutionSpaceFromDevice<DeviceType>::space == Device)
+      Kokkos::parallel_for(config, f);
+    else
+      Kokkos::parallel_for(nall, f);
 #else
     Kokkos::parallel_for(nall, f);
 #endif
   } else {
     NPairKokkosBuildFunctor<DeviceType,HALF_NEIGH,0,0> f(data,atoms_per_bin * 5 * sizeof(X_FLOAT) * factor);
 #ifdef KOKKOS_HAVE_CUDA
-    Kokkos::parallel_for(config, f);
+    if (ExecutionSpaceFromDevice<DeviceType>::space == Device)
+      Kokkos::parallel_for(config, f);
+    else
+      Kokkos::parallel_for(nall, f);
 #else
     Kokkos::parallel_for(nall, f);
 #endif
diff --git a/src/KOKKOS/npair_kokkos.h b/src/KOKKOS/npair_kokkos.h
index 53183114d3..a28b5ff978 100644
--- a/src/KOKKOS/npair_kokkos.h
+++ b/src/KOKKOS/npair_kokkos.h
@@ -409,7 +409,7 @@ struct NPairKokkosBuildFunctor<LMPHostType,HALF_NEIGH,GHOST_NEWTON,TRI> {
     c.template build_Item<HALF_NEIGH,GHOST_NEWTON,TRI>(i);
   }
 
-  void operator() (typename Kokkos::TeamPolicy<LMPHostType>::member_type dev) const {}
+  void operator() (typename Kokkos::TeamPolicy<LMPHostType>::member_type dev) const {} // Should error out
 };
 
 template<class DeviceType,int HALF_NEIGH>