store_thread_id.hpp

Go to the documentation of this file.

#ifndef KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_STORE_THREAD_ID_HPP
#define KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_STORE_THREAD_ID_HPP
 
#include <thread>
 
#include "kokkos-execution/stdexec.hpp"
 
namespace Tests::Utils::Functors {
 
struct StoreThreadID {
    template <typename... Args>
    void operator()(Args&&...) const noexcept {
        *tid = std::this_thread::get_id();
    }
 
    std::thread::id* tid;
};

#define THEN_STORE_THREAD_ID(__id__)       stdexec::then(Tests::Utils::Functors::StoreThreadID{__id__})

#define UPON_ERROR_STORE_THREAD_ID(__id__) stdexec::upon_error(Tests::Utils::Functors::StoreThreadID{__id__})
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_STORE_THREAD_ID_HPP