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 {
    void operator()() 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__})
 
} // namespace Tests::Utils::Functors
 
#endif // KOKKOS_EXECUTION_TESTS_UTILS_FUNCTORS_STORE_THREAD_ID_HPP