cuda_device_atomic_ref.hpp

/*
 * VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2022-2024 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// Local include(s).
#include "vecmem/memory/device_address_space.hpp"
#include "vecmem/memory/memory_order.hpp"
#include "vecmem/utils/types.hpp"
 
// System include(s).
#include <type_traits>
 
namespace vecmem {
namespace cuda {
 
template <typename T,
          device_address_space address = device_address_space::global>
class device_atomic_ref {
 
public:
 
    typedef T value_type;
    typedef value_type difference_type;
    typedef value_type* pointer;
    typedef value_type& reference;
 
 
 
    static_assert(
        std::is_integral<value_type>::value,
        "vecmem::cuda::atomic_ref only accepts built-in integral types");
 
 
    VECMEM_HOST_AND_DEVICE
    explicit device_atomic_ref(reference ref);
    VECMEM_HOST_AND_DEVICE
    device_atomic_ref(const device_atomic_ref& parent);
 
    device_atomic_ref& operator=(const device_atomic_ref&) = delete;
 
 
    VECMEM_HOST_AND_DEVICE
    value_type operator=(value_type data) const;
 
    VECMEM_HOST_AND_DEVICE
    void store(value_type data,
               memory_order order = memory_order::seq_cst) const;
    VECMEM_HOST_AND_DEVICE
    value_type load(memory_order order = memory_order::seq_cst) const;
 
    VECMEM_HOST_AND_DEVICE
    value_type exchange(value_type data,
                        memory_order order = memory_order::seq_cst) const;
 
    VECMEM_HOST_AND_DEVICE
    bool compare_exchange_strong(reference expected, value_type desired,
                                 memory_order success,
                                 memory_order failure) const;
    VECMEM_HOST_AND_DEVICE
    bool compare_exchange_strong(
        reference expected, value_type desired,
        memory_order order = memory_order::seq_cst) const;
 
 
 
    VECMEM_HOST_AND_DEVICE
    value_type fetch_add(value_type data,
                         memory_order order = memory_order::seq_cst) const;
    VECMEM_HOST_AND_DEVICE
    value_type fetch_sub(value_type data,
                         memory_order order = memory_order::seq_cst) const;
 
    VECMEM_HOST_AND_DEVICE
    value_type fetch_and(value_type data,
                         memory_order order = memory_order::seq_cst) const;
    VECMEM_HOST_AND_DEVICE
    value_type fetch_or(value_type data,
                        memory_order order = memory_order::seq_cst) const;
    VECMEM_HOST_AND_DEVICE
    value_type fetch_xor(value_type data,
                         memory_order order = memory_order::seq_cst) const;
 
 
private:
    pointer m_ptr;
 
};  // class device_atomic_ref
 
}  // namespace cuda
}  // namespace vecmem
 
// Include the implementation.
#include "vecmem/memory/impl/cuda_device_atomic_ref.ipp"