diff --git a/profiling/all/impl/Kokkos_Profiling_Interface.hpp b/profiling/all/impl/Kokkos_Profiling_Interface.hpp index b66886d9f..ddd6223be 100644 --- 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