context.h

 
#pragma once
 
#include <cstdint>
#include <cstring>
#include <memory>
#include <string>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/component.h>
#include <ucxx/config.h>
#include <ucxx/context_builder.h>
#include <ucxx/memory_handle_builder.h>
#include <ucxx/worker_builder.h>
 
namespace ucxx {
 
class ContextBuilder;
 
class MemoryHandle;
class Worker;
 
class Context : public Component {
 private:
  ucp_context_h _handle{nullptr};  
  Config _config{{}};              
  uint64_t _featureFlags{0};       
  bool _cudaSupport{false};        
 
  Context(const ConfigMap ucxConfig, const uint64_t featureFlags);
 
 public:
  static constexpr uint64_t defaultFeatureFlags =
    UCP_FEATURE_TAG | UCP_FEATURE_WAKEUP | UCP_FEATURE_STREAM | UCP_FEATURE_AM |
    UCP_FEATURE_RMA;  
 
  Context()                          = delete;
  Context(const Context&)            = delete;
  Context& operator=(Context const&) = delete;
  Context(Context&& o)               = delete;
  Context& operator=(Context&& o)    = delete;
 
  // Allow internal construction without exposing factory functions.
  friend class detail::ConstructorFactory;
 
  friend class ContextBuilder;
 
  ~Context();
 
  [[nodiscard]] ConfigMap getConfig();
 
  [[nodiscard]] ucp_context_h getHandle();
 
  [[nodiscard]] std::string getInfo();
 
  [[nodiscard]] uint64_t getFeatureFlags() const;
 
  [[nodiscard]] bool hasCudaSupport() const;
 
  [[nodiscard]] WorkerBuilder workerBuilder();
 
  [[nodiscard]] MemoryHandleBuilder memoryHandleBuilder(size_t size);
};
 
}  // namespace ucxx