typedefs.h

 
#pragma once
 
#include <functional>
#include <limits>
#include <memory>
#include <string>
#include <unordered_map>
 
#include <ucp/api/ucp.h>
 
namespace ucxx {
 
class Buffer;
class Request;
class RequestAm;
 
typedef enum {
  UCXX_LOG_LEVEL_FATAL,       /* Immediate termination */
  UCXX_LOG_LEVEL_ERROR,       /* Error is returned to the user */
  UCXX_LOG_LEVEL_WARN,        /* Something's wrong, but we continue */
  UCXX_LOG_LEVEL_DIAG,        /* Diagnostics, silent adjustments or internal error handling */
  UCXX_LOG_LEVEL_INFO,        /* Information */
  UCXX_LOG_LEVEL_DEBUG,       /* Low-volume debugging */
  UCXX_LOG_LEVEL_TRACE,       /* High-volume debugging */
  UCXX_LOG_LEVEL_TRACE_REQ,   /* Every send/receive request */
  UCXX_LOG_LEVEL_TRACE_DATA,  /* Data sent/received on the transport */
  UCXX_LOG_LEVEL_TRACE_ASYNC, /* Asynchronous progress engine */
  UCXX_LOG_LEVEL_TRACE_FUNC,  /* Function calls */
  UCXX_LOG_LEVEL_TRACE_POLL,  /* Polling functions */
  UCXX_LOG_LEVEL_LAST,        /* Last level barrier, not an actual level */
  UCXX_LOG_LEVEL_PRINT        /* Temporary output */
} ucxx_log_level_t;
 
enum class TransferDirection { Send = 0, Receive };
 
enum Tag : ucp_tag_t {};
 
enum TagMask : ucp_tag_t {};
 
static constexpr TagMask TagMaskFull{std::numeric_limits<std::underlying_type_t<TagMask>>::max()};
 
class TagRecvInfo {
 public:
  Tag senderTag;  
  size_t length;  
 
  explicit TagRecvInfo(const ucp_tag_recv_info_t&);
};
 
typedef std::unordered_map<std::string, std::string> ConfigMap;
 
typedef std::function<void(ucs_status_t, std::shared_ptr<void>)> RequestCallbackUserFunction;
 
typedef std::shared_ptr<void> RequestCallbackUserData;
 
typedef RequestCallbackUserFunction EndpointCloseCallbackUserFunction;
 
typedef RequestCallbackUserData EndpointCloseCallbackUserData;
 
typedef std::function<std::shared_ptr<Buffer>(size_t)> AmAllocatorType;
 
typedef std::function<void(std::shared_ptr<Request>, ucp_ep_h)> AmReceiverCallbackType;
 
typedef std::string AmReceiverCallbackOwnerType;
 
typedef uint64_t AmReceiverCallbackIdType;
 
typedef const std::string AmReceiverCallbackInfoSerialized;
 
class AmReceiverCallbackInfo {
 public:
  const AmReceiverCallbackOwnerType owner;
  const AmReceiverCallbackIdType id;
 
  AmReceiverCallbackInfo() = delete;
  AmReceiverCallbackInfo(const AmReceiverCallbackOwnerType owner, AmReceiverCallbackIdType id);
};
 
typedef const std::string SerializedRemoteKey;
 
}  // namespace ucxx