load_check_add.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LOAD_CHECK_ADD_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LOAD_CHECK_ADD_HPP
 
#include "Kokkos_Macros.hpp"
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#    include "Kokkos_Core.hpp"
#endif
 
namespace Tests::Utils::Functors {

template <typename ValueType, bool OnDevice>
struct LoadCheckAdd {
    ValueType prev;
    ValueType value;
    ValueType* data;
 
    KOKKOS_FUNCTION
    void operator()() const {
        if constexpr (OnDevice) {
            KOKKOS_IF_ON_HOST(Kokkos::abort("You should not be running on host.");)
        } else {
            KOKKOS_IF_ON_DEVICE(Kokkos::abort("You should not be running on device.");)
        }
 
        if (*data != prev) {
            KOKKOS_IF_ON_HOST(Kokkos::abort("Unexpected value on host.");)
            KOKKOS_IF_ON_DEVICE(Kokkos::abort("Unexpected value on device.");)
        }
        *data += value;
    }
};
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LOAD_CHECK_ADD_HPP