1 #pragma once
2 
3 #include <torch/csrc/distributed/rpc/message.h>
4 #include <torch/csrc/distributed/rpc/rpc_command_base.h>
5 
6 namespace torch {
7 namespace distributed {
8 namespace autograd {
9 
10 // Empty response for CleanupAutogradContextReq. Send to acknowledge receipt of
11 // a CleanupAutogradContextReq.
12 class TORCH_API CleanupAutogradContextResp : public rpc::RpcCommandBase {
13  public:
14   CleanupAutogradContextResp() = default;
15   // Serialization and deserialization methods.
16   c10::intrusive_ptr<rpc::Message> toMessageImpl() && override;
17   static std::unique_ptr<CleanupAutogradContextResp> fromMessage(
18       const rpc::Message& message);
19 };
20 
21 } // namespace autograd
22 } // namespace distributed
23 } // namespace torch
24