opencl_manager.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include "sol3/core/shmem_buffer.h"
#include "sol3/cpp/filesystem.h"
#include "sol3/cpp/thread_safe_value.h"
#include "sol3/opencl/opencl_compat.h"
 
#include <map>
#include <string>
#include <vector>
 
namespace sol3::opencl {
 
class OpenClManager {
 public:
  static OpenClManager& instance();
 
  cl::Context const& context() const { return context_; }
 
  cl::Device const& device() const { return device_; }
 
  std::vector<cl::Device> const& devices() const { return devices_; }
 
  // Creates a command queue to use for a kernel instance.
  cl::CommandQueue createCommandQueue() const {
    return cl::CommandQueue(context_, device_);
  }
 
  cl::Kernel createKernel(
      cpp::fs::path const& source_cl_file,
      std::string const& kernel_name,
      std::string const& compiler_opts = "");
 
  cl::Program getOrCompileProgramFile(
      cpp::fs::path const& source_cl_file,
      std::string const& compiler_opts = "");
 
  cl::Program getOrCompileProgramSource(
      std::string const& name,
      std::string const& source,
      std::string const& compiler_opts = "");
 
  cl::Buffer mapBufferToGpu(core::IBufferMutable& buffer);
  cl::Buffer mapBufferToGpu(core::IBufferConst const& buffer);
 
 private:
  OpenClManager();
  ~OpenClManager() = default;
 
  // Non-copyable, non-movable
  OpenClManager(OpenClManager const&) = delete;
  OpenClManager& operator=(OpenClManager const&) = delete;
  OpenClManager(OpenClManager&&) = delete;
  OpenClManager& operator=(OpenClManager&&) = delete;
 
  template <typename TGetSourceFn>
  cl::Program getOrCompileImpl(
      std::string const& key,
      std::string const& compiler_opts,
      TGetSourceFn&& get_source);
 
  std::vector<cl::Platform> platforms_;
  std::vector<cl::Device> devices_;
  cl::Device device_;
  cl::Context context_;
  cpp::ThreadSafeValue<std::map<std::string, cl::Program>> programs_;
};
 
}  // namespace sol3::opencl
 
namespace sol3 {
inline opencl::OpenClManager& openclMgr() {
  return opencl::OpenClManager::instance();
}
}  // namespace sol3