RdmaGpuBuffer.hpp

Go to the documentation of this file.

#pragma once
 
#include <score_plugin_gfx_export.h>
 
#include <cstddef>
#include <cstdint>
#include <memory>
 
namespace score::gfx
{
struct CudaFunctions;
}
 
namespace score::gfx::interop
{
 
enum class RdmaGpuApi : std::uint8_t
{
  None = 0,
  Cuda,         
  Vulkan,       
  D3D12,        
  HostFallback, 
};
 
struct RdmaGpuSlot
{
  void* gpuVA{};       
  std::size_t size{};  
  void* opaque{};      
};
 
struct RdmaGpuBufferConfig
{
  RdmaGpuApi api{RdmaGpuApi::None};
  std::uint32_t slotCount{};
  std::size_t frameBytes{};
 
  // Backend context — only the field matching `api` is read; the caller owns
  // the lifetime of whatever it points to.
  score::gfx::CudaFunctions* cuda{}; 
  int cudaDevice{0};                 
  const void* vulkanCtx{};           
  void* d3d12Device{};               
};
 
class SCORE_PLUGIN_GFX_EXPORT RdmaGpuBuffer
{
public:
  RdmaGpuBuffer() noexcept;
  ~RdmaGpuBuffer();
 
  RdmaGpuBuffer(RdmaGpuBuffer&&) noexcept;
  RdmaGpuBuffer& operator=(RdmaGpuBuffer&&) noexcept;
  RdmaGpuBuffer(const RdmaGpuBuffer&) = delete;
  RdmaGpuBuffer& operator=(const RdmaGpuBuffer&) = delete;
 
  bool create(const RdmaGpuBufferConfig& cfg);
 
  bool valid() const noexcept;
  RdmaGpuApi backend() const noexcept;
  std::size_t slotCount() const noexcept;
  const RdmaGpuSlot& slot(std::size_t i) const noexcept;
 
  void destroy() noexcept;
 
private:
  struct Impl;
  std::unique_ptr<Impl> m_impl;
};
 
} // namespace score::gfx::interop