fixes

Author: sbaldu

include/CLUEstering/core/detail/Clusterer.hpp

@@ -162,7 +162,6 @@ namespace clue {
     alpaka::wait(queue);
 
     const auto n_points = h_points.size();
-    std::cout << "ntiles = " << m_tiles->view().ntiles << std::endl;
     m_tiles->template fill_batch<Acc>(queue, dev_points, n_points, d_event_offsets, max_event_size);
 
     detail::computeLocalDensityBatched<internal::Acc2D>(queue,

include/CLUEstering/core/detail/ClusteringKernels.hpp

@@ -399,7 +399,7 @@ namespace clue::detail {
                                          std::size_t max_event_size,
                                          std::size_t block_size) {
     const auto blocks_per_event = divide_up_by(max_event_size, block_size);
-    const auto batch_size = event_offsets.size() - 1;
+    const auto batch_size = alpaka::getExtents(event_offsets)[0] - 1;
     const Idx grid_size = clue::divide_up_by(max_event_size, block_size);
     const auto work_division =
         make_workdiv<internal::Acc2D>({batch_size, blocks_per_event}, {1, block_size});
@@ -452,7 +452,7 @@ namespace clue::detail {
     alpaka::memset(queue, d_seed_candidates, 0u);
 
     const auto blocks_per_event = divide_up_by(max_event_size, block_size);
-    const auto batch_size = event_offsets.size() - 1;
+    const auto batch_size = alpaka::getExtents(event_offsets)[0] - 1;
     const Idx grid_size = clue::divide_up_by(max_event_size, block_size);
     const auto work_division =
         make_workdiv<internal::Acc2D>({batch_size, blocks_per_event}, {1, block_size});

include/CLUEstering/data_structures/detail/AssociationMap.hpp

@@ -445,7 +445,7 @@ namespace clue {
 
     const auto blocksize = 256;
     const auto blocks_per_event = divide_up_by(max_event_size, blocksize);
-    const auto batch_size = event_offsets.size() - 1;
+    const auto batch_size = alpaka::getExtents(event_offsets)[0] - 1;
     const auto batch_workdiv =
         make_workdiv<internal::Acc2D>({batch_size, blocks_per_event}, {1, blocksize});
     alpaka::exec<internal::Acc2D>(queue,

tests/test_batched_clustering.cpp

@@ -19,6 +19,7 @@ TEST_CASE("Test batched clustering with fixed batch size") {
 
   std::vector<std::size_t> event_sizes(10, batch_size);
   algo.make_clusters(queue, h_points, d_points, event_sizes, clue::FlatKernel{.5f});
+  alpaka::wait(queue);
 
   auto truth = clue::read_output<2>(queue, "../../../data/truth_files/data_1024_truth.csv");
   auto truth_n_clusters = clue::detail::compute_nclusters(truth.clusterIndexes());