sparse_sort_crs: Fix row begin/end offset in entry shuffle

kokkos · Sep 26, 2024 · 129c837 · 129c837
1 parent 3d6bbb2
commit 129c837
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/perf_test/sparse/KokkosSparse_sort_crs.cpp b/perf_test/sparse/KokkosSparse_sort_crs.cpp
@@ -82,7 +82,9 @@ void run_experiment(int argc, char** argv, const CommonInputParams& common_param
   std::random_device rd;
   std::mt19937 g(rd());
   for (lno_t i = 0; i < m; i++) {
-    std::shuffle(entriesHost.data() + i, entriesHost.data() + i + 1, g);
+    const size_type rowBegin = rowmapHost(i);
+    const size_type rowEnd = rowmapHost(i+1);
+    std::shuffle(entriesHost.data() + rowBegin, entriesHost.data() + rowEnd, g);
   }
   Kokkos::deep_copy(shuffledEntries, entriesHost);
   exec_space exec;