labeled.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LABELED_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LABELED_HPP
 
#include "kokkos-execution/stdexec.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 <char ID>
struct Labeled {
    template <typename... Args>
    KOKKOS_FUNCTION void operator()(Args...) const noexcept {
    }
};

#define THEN_LABELED(_id_) stdexec::then(Tests::Utils::Functors::Labeled<_id_>{})

#define THEN_LABELED_PFOR(_exec_, _id_)                                                                                \
    Kokkos::Execution::parallel_for(#_id_, Kokkos::RangePolicy<_exec_>(0, 1), Tests::Utils::Functors::Labeled<_id_>{})
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_LABELED_HPP