memory_handle_builder.h

 
#pragma once
 
#include <cstddef>
#include <memory>
#include <utility>
 
#include <ucs/memory/memory_type.h>
 
namespace ucxx {
 
class Context;
class MemoryHandle;
 
class MemoryHandleBuilder final {
 public:
  MemoryHandleBuilder(std::shared_ptr<Context> context, size_t size);
 
  ~MemoryHandleBuilder();
 
  MemoryHandleBuilder(const MemoryHandleBuilder& other);
  MemoryHandleBuilder& operator=(const MemoryHandleBuilder& other);
  MemoryHandleBuilder(MemoryHandleBuilder&&) noexcept;
  MemoryHandleBuilder& operator=(MemoryHandleBuilder&&) noexcept;
 
  operator std::shared_ptr<MemoryHandle>();
 
  MemoryHandleBuilder& buffer(void* buffer);
 
  MemoryHandleBuilder& memoryType(ucs_memory_type_t memoryType);
 
  [[nodiscard]] std::shared_ptr<MemoryHandle> build();
 
 private:
  struct Impl;
  std::unique_ptr<Impl> _impl;
};
 
}  // namespace ucxx