atomic.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_ATOMIC_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_ATOMIC_HPP
 
#include "Kokkos_Core.hpp"
 
#if defined(KOKKOS_ENABLE_CUDA)
#    include <cuda/atomic>
#endif

 
namespace Tests::Utils {

template <typename T>
KOKKOS_FUNCTION void atomic_add(T* ptr, const T val) {
#if defined(KOKKOS_ENABLE_CUDA)
    cuda::atomic_ref<T, cuda::thread_scope_system>(*ptr).fetch_add(val, cuda::memory_order_relaxed);
#elif defined(KOKKOS_ENABLE_HIP)
    __hip_atomic_fetch_add(ptr, val, __ATOMIC_RELAXED, __HIP_MEMORY_SCOPE_SYSTEM);
#else
    std::atomic_ref<T>(*ptr).fetch_add(val, std::memory_order_relaxed);
#endif
}
 
} // namespace Tests::Utils
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_ATOMIC_HPP