fixup: halo capture for cuda

ORNL · Oct 16, 2023 · 14005e1 · 14005e1
1 parent e68f0d6
commit 14005e1
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/src/CabanaPD_Comm.hpp b/src/CabanaPD_Comm.hpp
@@ -80,9 +80,7 @@ struct HaloIds
         // balancing).
         neighborBounds( local_grid );
 
-        build(
-            positions,
-            KOKKOS_LAMBDA( const int, const double[3] ) { return true; } );
+        build( positions );
     }
 
     // Find the bounds of each neighbor rank and store for determining which
@@ -198,6 +196,16 @@ struct HaloIds
         Kokkos::fence();
     }
 
+    template <class PositionSliceType>
+    void build( const PositionSliceType& positions )
+    {
+        auto empty_functor = KOKKOS_LAMBDA( const int, const double[3] )
+        {
+            return true;
+        };
+        build( positions, empty_functor );
+    }
+
     template <class PositionSliceType>
     void rebuild( const PositionSliceType& positions )
     {
@@ -216,9 +224,7 @@ struct HaloIds
         if ( dest_count > dest_size )
         {
             Kokkos::deep_copy( _send_count, 0 );
-            build(
-                positions,
-                KOKKOS_LAMBDA( const int, const double[3] ) { return true; } );
+            build( positions );
         }
     }
 };