kernel_task.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/async/async_task.hpp>
#include <uni20/common/trace.hpp>
#include <uni20/linalg/kernel_attempt.hpp>
 
#include <concepts>
#include <optional>
#include <type_traits>
#include <utility>
 
namespace uni20::linalg
{
 
template <class Task>
  requires std::derived_from<Task, async::BasicTask>
class KernelTaskAttempt {
  public:
    using task_type = Task;
 
    explicit KernelTaskAttempt(KernelAttempt attempt) noexcept : attempt_(attempt) {}
 
    explicit KernelTaskAttempt(Task task) noexcept : task_(std::move(task)) {}
 
    KernelTaskAttempt(KernelTaskAttempt const&) = delete;
    KernelTaskAttempt& operator=(KernelTaskAttempt const&) = delete;
    KernelTaskAttempt(KernelTaskAttempt&&) noexcept = default;
    KernelTaskAttempt& operator=(KernelTaskAttempt&&) noexcept = default;
 
    [[nodiscard]] KernelAttempt attempt() const noexcept { return attempt_; }
 
    [[nodiscard]] bool has_task() const noexcept { return task_.has_value(); }
 
    [[nodiscard]] Task take_task() &&
    {
      CHECK(kernel_attempt_succeeded(attempt_));
      CHECK(task_.has_value());
      return std::move(*task_);
    }
 
  private:
    KernelAttempt attempt_ = KernelAttempt::success;
    std::optional<Task> task_{};
};
 
namespace detail
{
template <class T> struct IsKernelTaskAttempt : std::false_type
{};
 
template <class Task> struct IsKernelTaskAttempt<KernelTaskAttempt<Task>> : std::true_type
{};
 
template <class T> inline constexpr bool is_kernel_task_attempt = IsKernelTaskAttempt<std::remove_cvref_t<T>>::value;
} // namespace detail
 
} // namespace uni20::linalg