increment.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_INCREMENT_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_INCREMENT_HPP
 
#include <type_traits>
 
#include "kokkos-execution/stdexec.hpp"
 
#include "Kokkos_Core.hpp"
 
#include "kokkos-utils/concepts/View.hpp"
 
#include "tests/utils/atomic.hpp"
 
namespace Tests::Utils::Functors {

template <Kokkos::utils::concepts::ViewOfRank<0> ViewType, bool MayThrow = true, bool Atomic = false>
struct Increment {
    typename ViewType::non_const_type data;
 
    KOKKOS_FUNCTION
    void operator()() const noexcept(!MayThrow) requires(Atomic == false)
    {
        ++data();
    }
 
    KOKKOS_FUNCTION
    void operator()() const noexcept(!MayThrow) requires(Atomic == true)
    {
        Tests::Utils::atomic_add(data.data(), 1);
    }
};

#define THEN_INCREMENT(_data_)                                                                                         \
    stdexec::then(Tests::Utils::Functors::Increment<std::remove_cvref_t<decltype(_data_)>, true, false>{.data = _data_})

#define THEN_INCREMENT_ATOMIC(_data_)                                                                                  \
    stdexec::then(Tests::Utils::Functors::Increment<std::remove_cvref_t<decltype(_data_)>, true, true>{.data = _data_})
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_INCREMENT_HPP