Get rid of LegacyDefaultCallback in benchmarks

arborx · Apr 8, 2024 · b257854 · b257854
1 parent e474b9e
commit b257854
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/benchmarks/brute_force_vs_bvh/brute_force_vs_bvh_timpl.hpp b/benchmarks/brute_force_vs_bvh/brute_force_vs_bvh_timpl.hpp
@@ -93,10 +93,10 @@ static void run_fp(int nprimitives, int nqueries, int nrepeats)
                                       ArborX::PairValueIndex<Point>>
           bvh{space, ArborX::Experimental::attach_indices(primitives)};
 
-      Kokkos::View<int *, ExecutionSpace> indices("Benchmark::indices_ref", 0);
+      Kokkos::View<unsigned *, ExecutionSpace> indices("Benchmark::indices_ref",
+                                                       0);
       Kokkos::View<int *, ExecutionSpace> offset("Benchmark::offset_ref", 0);
-      bvh.query(space, predicates, ArborX::Details::LegacyDefaultCallback{},
-                indices, offset);
+      bvh.query(space, predicates, indices, offset);
 
       space.fence();
       double time = timer.seconds();
@@ -112,10 +112,9 @@ static void run_fp(int nprimitives, int nqueries, int nrepeats)
       ArborX::BruteForce<MemorySpace, ArborX::PairValueIndex<Point>> brute{
           space, ArborX::Experimental::attach_indices(primitives)};
 
-      Kokkos::View<int *, ExecutionSpace> indices("Benchmark::indices", 0);
+      Kokkos::View<unsigned *, ExecutionSpace> indices("Benchmark::indices", 0);
       Kokkos::View<int *, ExecutionSpace> offset("Benchmark::offset", 0);
-      brute.query(space, predicates, ArborX::Details::LegacyDefaultCallback{},
-                  indices, offset);
+      brute.query(space, predicates, indices, offset);
 
       space.fence();
       double time = timer.seconds();

diff --git a/benchmarks/dbscan/ArborX_DBSCANVerification.hpp b/benchmarks/dbscan/ArborX_DBSCANVerification.hpp
@@ -317,10 +317,9 @@ bool verifyDBSCAN(ExecutionSpace exec_space, Primitives const &primitives,
   auto const predicates = ArborX::Experimental::attach_indices(
       ArborX::Experimental::make_intersects(points, eps));
 
-  Kokkos::View<int *, MemorySpace> indices("ArborX::DBSCAN::indices", 0);
+  Kokkos::View<unsigned *, MemorySpace> indices("ArborX::DBSCAN::indices", 0);
   Kokkos::View<int *, MemorySpace> offset("ArborX::DBSCAN::offset", 0);
-  ArborX::query(bvh, exec_space, predicates,
-                ArborX::Details::LegacyDefaultCallback{}, indices, offset);
+  ArborX::query(bvh, exec_space, predicates, indices, offset);
 
   auto passed = Details::verifyClusters(exec_space, indices, offset, labels,
                                         core_min_size);