This should be fixed... but I get a segfault on GPU

parthenon-hpc-lab · Nov 7, 2024 · 42a9356 · 42a9356
1 parent 8c6f4d0
commit 42a9356
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/utils/sort.hpp b/src/utils/sort.hpp
@@ -59,6 +59,8 @@ KOKKOS_INLINE_FUNCTION int upper_bound(const T &arr, Real val) {
 template <class Key, class KeyComparator>
 void sort(ParArray1D<Key> data, KeyComparator comparator, size_t min_idx,
           size_t max_idx) {
+  Kokkos::fence();
+  printf("%s:%i\n", __FILE__, __LINE__);
   PARTHENON_DEBUG_REQUIRE(min_idx < data.extent(0), "Invalid minimum sort index!");
   PARTHENON_DEBUG_REQUIRE(max_idx < data.extent(0), "Invalid maximum sort index!");
 #if defined(KOKKOS_ENABLE_CUDA)
@@ -90,6 +92,8 @@ void sort(ParArray1D<Key> data, KeyComparator comparator, size_t min_idx,
                    "touch by opening an issue on the Parthenon GitHub.");
   }
 #endif // KOKKOS_ENABLE_CUDA
+  Kokkos::fence();
+  printf("%s:%i\n", __FILE__, __LINE__);
 }
 
 template <class Key>