request_mem.h

 
#pragma once
#include <memory>
#include <string>
#include <utility>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/delayed_submission.h>
#include <ucxx/request.h>
#include <ucxx/typedefs.h>
 
namespace ucxx {
 
class Buffer;
 
namespace internal {
class RecvMemMessage;
}  // namespace internal
 
class RequestMem : public Request {
 private:
  friend class internal::RecvMemMessage;
 
  uint64_t _remote_addr{};
  ucp_rkey_h _rkey{};
 
  RequestMem(std::shared_ptr<Endpoint> endpoint,
             const std::variant<data::MemPut, data::MemGet> requestData,
             const std::string operationName,
             const bool enablePythonFuture                = false,
             RequestCallbackUserFunction callbackFunction = nullptr,
             RequestCallbackUserData callbackData         = nullptr);
 
 public:
  friend std::shared_ptr<RequestMem> createRequestMem(
    std::shared_ptr<Endpoint> endpoint,
    const std::variant<data::MemPut, data::MemGet> requestData,
    const bool enablePythonFuture,
    RequestCallbackUserFunction callbackFunction,
    RequestCallbackUserData callbackData);
 
  virtual void populateDelayedSubmission();
 
  static void memPutCallback(void* request, ucs_status_t status, void* arg);
 
  static void memGetCallback(void* request, ucs_status_t status, void* arg);
 
  void request();
};
 
}  // namespace ucxx