cuda_task.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "async_task_promise.hpp"
 
namespace uni20::async
{
 
class CudaTask final : public BasicTask {
  public:
    using base_type = BasicTask;
    using promise_type = CudaTaskPromise;
 
    CudaTask() = default;
    CudaTask(CudaTask const&) = delete;
    CudaTask& operator=(CudaTask const&) = delete;
    CudaTask(CudaTask&&) noexcept = default;
    CudaTask& operator=(CudaTask&&) noexcept = default;
 
    CudaTask& debug_name(std::string const& label)
    {
      this->BasicTask::debug_name(label);
      return *this;
    }
 
  private:
    explicit CudaTask(TaskHandle handle) noexcept : BasicTask(handle, construction_key{}) {}
 
    friend class CudaTaskPromise;
};
 
inline CudaTask CudaTaskPromise::get_return_object() noexcept
{
  auto handle = std::coroutine_handle<CudaTaskPromise>::from_promise(*this);
  return CudaTask(this->initialize_task(handle));
}
 
class ICudaScheduler : public virtual IScheduler {
  public:
    ~ICudaScheduler() override = default;
 
    virtual void schedule(CudaTask&& task) = 0;
 
    virtual void schedule(CudaTask&& task, int device) = 0;
};
 
} // namespace uni20::async