mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
95dbbf713e
Follows #125102 Pull Request resolved: https://github.com/pytorch/pytorch/pull/130109 Approved by: https://github.com/ezyang
24 lines
621 B
C++
24 lines
621 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/distributed/rpc/rpc_command_base.h>
|
|
#include <torch/csrc/distributed/rpc/types.h>
|
|
|
|
namespace torch::distributed::rpc {
|
|
|
|
// RPC call representing the response of a Python UDF over RPC.
|
|
class TORCH_API PythonResp final : public RpcCommandBase {
|
|
public:
|
|
explicit PythonResp(SerializedPyObj&& serializedPyObj);
|
|
|
|
c10::intrusive_ptr<Message> toMessageImpl() && override;
|
|
|
|
static std::unique_ptr<PythonResp> fromMessage(const Message& message);
|
|
|
|
const SerializedPyObj& serializedPyObj() const;
|
|
|
|
private:
|
|
SerializedPyObj serializedPyObj_;
|
|
};
|
|
|
|
} // namespace torch::distributed::rpc
|