diff --git a/src/KOKKOS/npair_kokkos.cpp b/src/KOKKOS/npair_kokkos.cpp index 852a4a9280..06567cbeb6 100644 --- a/src/KOKKOS/npair_kokkos.cpp +++ b/src/KOKKOS/npair_kokkos.cpp @@ -240,22 +240,25 @@ void NPairKokkos::build(NeighList *list_) // assumes newton off NPairKokkosBuildFunctorGhost f(data,atoms_per_bin * 5 * sizeof(X_FLOAT) * factor); -#ifdef LMP_KOKKOS_GPU - if (ExecutionSpaceFromDevice::space == Device) { - int team_size = atoms_per_bin*factor; - int team_size_max = Kokkos::TeamPolicy(team_size,Kokkos::AUTO).team_size_max(f,Kokkos::ParallelForTag()); - if (team_size <= team_size_max) { - Kokkos::TeamPolicy config((mbins+factor-1)/factor,team_size); - Kokkos::parallel_for(config, f); - } else { // fall back to flat method - f.sharedsize = 0; - Kokkos::parallel_for(nall, f); - } - } else - Kokkos::parallel_for(nall, f); -#else + +// temporarily disable team policy for ghost due to known bug + +//#ifdef LMP_KOKKOS_GPU +// if (ExecutionSpaceFromDevice::space == Device) { +// int team_size = atoms_per_bin*factor; +// int team_size_max = Kokkos::TeamPolicy(team_size,Kokkos::AUTO).team_size_max(f,Kokkos::ParallelForTag()); +// if (team_size <= team_size_max) { +// Kokkos::TeamPolicy config((mbins+factor-1)/factor,team_size); +// Kokkos::parallel_for(config, f); +// } else { // fall back to flat method +// f.sharedsize = 0; +// Kokkos::parallel_for(nall, f); +// } +// } else +// Kokkos::parallel_for(nall, f); +//#else Kokkos::parallel_for(nall, f); -#endif +//#endif } else { if (SIZE) { NPairKokkosBuildFunctorSize f(data,atoms_per_bin * 6 * sizeof(X_FLOAT) * factor);