command_queue.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_BACKENDS_OPENCL_DETAIL_COMMAND_QUEUE_HPP_
#define PLSSVM_BACKENDS_OPENCL_DETAIL_COMMAND_QUEUE_HPP_
#pragma once
 
#include "plssvm/backends/OpenCL/detail/kernel.hpp"  // plssvm::opencl::detail::kernel
 
#include "CL/cl.h"                                   // cl_context, cl_command_queue, cl_device_id
 
#include <map>                                       // std::map
 
namespace plssvm::opencl::detail {
 
class command_queue {
  public:
    command_queue() = default;
    command_queue(cl_context context, cl_device_id device);
 
    command_queue(const command_queue &) = delete;
    command_queue(command_queue &&other) noexcept;
    command_queue &operator=(const command_queue &) = delete;
    command_queue &operator=(command_queue &&other) noexcept;
 
    ~command_queue();
 
    [[nodiscard]] operator cl_command_queue &() noexcept { return queue; }
    [[nodiscard]] operator const cl_command_queue &() const noexcept { return queue; }
 
    template <typename real_type>
    void add_kernel(compute_kernel_name name, kernel &&compute_kernel);
 
    template <typename real_type>
    [[nodiscard]] const kernel &get_kernel(compute_kernel_name name) const;
 
    cl_command_queue queue{};
    std::map<compute_kernel_name, kernel> float_kernels{};
    std::map<compute_kernel_name, kernel> double_kernels{};
};
 
}  // namespace plssvm::opencl::detail
 
#endif  // PLSSVM_BACKENDS_OPENCL_DETAIL_COMMAND_QUEUE_HPP_