mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8da04e023e
This reverts commit 4146be192ead477360a2763c5005e46a9485c3bf. Reverted https://github.com/pytorch/pytorch/pull/108480 on behalf of https://github.com/huydhn due to Sorry for reverting this, but this is needed to keep trunk green after https://github.com/pytorch/pytorch/pull/108479 was reverted. Both will need to be relanded ([comment](https://github.com/pytorch/pytorch/pull/108480#issuecomment-1707067595))
35 lines
1.1 KiB
C++
35 lines
1.1 KiB
C++
#include <torch/csrc/distributed/rpc/python_resp.h>
|
|
|
|
#include <c10/util/C++17.h>
|
|
|
|
namespace torch {
|
|
namespace distributed {
|
|
namespace rpc {
|
|
|
|
PythonResp::PythonResp(SerializedPyObj&& serializedPyObj)
|
|
: serializedPyObj_(std::move(serializedPyObj)) {}
|
|
|
|
c10::intrusive_ptr<Message> PythonResp::toMessageImpl() && {
|
|
auto payload = std::vector<char>(
|
|
serializedPyObj_.payload_.begin(), serializedPyObj_.payload_.end());
|
|
return c10::make_intrusive<Message>(
|
|
std::move(payload),
|
|
std::move(serializedPyObj_.tensors_),
|
|
MessageType::PYTHON_RET);
|
|
}
|
|
|
|
std::unique_ptr<PythonResp> PythonResp::fromMessage(const Message& message) {
|
|
std::string payload(message.payload().begin(), message.payload().end());
|
|
std::vector<Tensor> tensors = message.tensors();
|
|
SerializedPyObj serializedPyObj(std::move(payload), std::move(tensors));
|
|
return std::make_unique<PythonResp>(std::move(serializedPyObj));
|
|
}
|
|
|
|
const SerializedPyObj& PythonResp::serializedPyObj() const {
|
|
return serializedPyObj_;
|
|
}
|
|
|
|
} // namespace rpc
|
|
} // namespace distributed
|
|
} // namespace torch
|