memory_handle.h

 
#pragma once
#include <memory>
#include <string>
#include <utility>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/component.h>
#include <ucxx/context.h>
 
namespace ucxx {
 
class RemoteKey;
 
class MemoryHandle : public Component {
 private:
  ucp_mem_h _handle{};       
  size_t _size{0};           
  uint64_t _baseAddress{0};  
  ucs_memory_type_t _memoryType{
    UCS_MEMORY_TYPE_HOST};  
 
  MemoryHandle(std::shared_ptr<Context> context,
               const size_t size,
               void* buffer,
               const ucs_memory_type_t memoryType);
 
 public:
  MemoryHandle()                               = delete;
  MemoryHandle(const MemoryHandle&)            = delete;
  MemoryHandle& operator=(MemoryHandle const&) = delete;
  MemoryHandle(MemoryHandle&& o)               = delete;
  MemoryHandle& operator=(MemoryHandle&& o)    = delete;
 
  [[nodiscard]] friend std::shared_ptr<MemoryHandle> createMemoryHandle(
    std::shared_ptr<Context> context,
    const size_t size,
    void* buffer,
    const ucs_memory_type_t memoryType);
 
  ~MemoryHandle();
 
  [[nodiscard]] ucp_mem_h getHandle();
 
  [[nodiscard]] size_t getSize() const;
 
  [[nodiscard]] uint64_t getBaseAddress();
 
  [[nodiscard]] ucs_memory_type_t getMemoryType();
 
  [[nodiscard]] std::shared_ptr<RemoteKey> createRemoteKey();
};
 
}  // namespace ucxx