Update Kokkos_Profiling_Interface.hpp after kokkos pr 7394

kokkos · Oct 6, 2024 · 0bbba87 · 0bbba87
1 parent a9453b2
commit 0bbba87
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/profiling/all/impl/Kokkos_Profiling_Interface.hpp b/profiling/all/impl/Kokkos_Profiling_Interface.hpp
@@ -101,6 +101,15 @@ inline uint32_t device_id(ExecutionSpace const& space) noexcept {
           << num_instance_bits) +
          space.impl_instance_id();
 }
+
+inline uint32_t int_for_synchronization_reason(
+    Kokkos::Tools::Experimental::SpecialSynchronizationCases reason) {
+  switch (reason) {
+    case GlobalDeviceSynchronization: return 0;
+    case DeepCopyResourceSynchronization: return 0x00ffffff;
+  }
+  return 0;
+}
 }  // namespace Experimental
 }  // namespace Tools
 }  // end namespace Kokkos