Merge pull request #823 from stephenswat/fix/kokkos_global_index_cvt

Fix global counter conversion issue in Kokkos
acts-project · Jan 24, 2025 · 51568d3 · 51568d3
2 parents 2b95e84 + 2b8ca72
commit 51568d3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/device/kokkos/src/seeding/spacepoint_binning.cpp b/device/kokkos/src/seeding/spacepoint_binning.cpp
@@ -57,9 +57,10 @@ spacepoint_binning::output_type spacepoint_binning::operator()(
                 Kokkos::TeamThreadRange(team_member, num_threads),
                 [&](const int& thr) {
                     device::count_grid_capacities(
-                        static_cast<std::size_t>(team_member.league_rank() *
-                                                     team_member.team_size() +
-                                                 thr),
+                        static_cast<device::global_index_t>(
+                            team_member.league_rank() *
+                                team_member.team_size() +
+                            thr),
                         config, axes.first, axes.second, spacepoints_view,
                         grid_capacities_view);
                 });