counter.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_COUNTER_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_COUNTER_HPP
 
#include "Kokkos_Core.hpp"
 
namespace Tests::Utils::Functors {

struct Counter {
    static inline std::atomic<unsigned int> default_constructions{0};
    static inline std::atomic<unsigned int> destructions{0};
    static inline std::atomic<unsigned int> copy_constructions{0};
    static inline std::atomic<unsigned int> copy_assignments{0};
    static inline std::atomic<unsigned int> move_constructions{0};
    static inline std::atomic<unsigned int> move_assignments{0};
 
    static inline std::atomic<unsigned int> next_id{0};
 
    unsigned int id{0};
 
    Counter() noexcept
        : id(next_id.fetch_add(1, std::memory_order_relaxed)) {
        default_constructions.fetch_add(1, std::memory_order_relaxed);
    }
 
    ~Counter() noexcept {
        destructions.fetch_add(1, std::memory_order_relaxed);
    }
 
    Counter(const Counter&) noexcept
        : id(next_id.fetch_add(1, std::memory_order_relaxed)) {
        copy_constructions.fetch_add(1, std::memory_order_relaxed);
    }
 
    Counter& operator=(const Counter&) noexcept {
        copy_assignments.fetch_add(1, std::memory_order_relaxed);
        return *this;
    }
 
    Counter(Counter&&) noexcept
        : id(next_id.fetch_add(1, std::memory_order_relaxed)) {
        move_constructions.fetch_add(1, std::memory_order_relaxed);
    }
 
    Counter& operator=(Counter&&) noexcept {
        move_assignments.fetch_add(1, std::memory_order_relaxed);
        return *this;
    }
 
    template <typename... Args>
    KOKKOS_FUNCTION void operator()(Args&&...) const noexcept {
    }
 
    static void reset() {
        default_constructions = 0;
        destructions = 0;
        copy_constructions = 0;
        copy_assignments = 0;
        move_constructions = 0;
        move_assignments = 0;
    }
};
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_COUNTER_HPP