Skip to content

T
tic
Commit 19381b51 by Tianqi Chen Committed by GitHub
Options

[RPC] Enable shutdown hook (#308)

parent 48fd9e46
Showing with 3 additions and 0 deletions
src/runtime/rpc/rpc_session.cc
...@@ -749,6 +749,9 @@ void RPCSession::ServerLoop() { ...@@ -749,6 +749,9 @@ void RPCSession::ServerLoop() {
TVMRetValue rv; TVMRetValue rv;
CHECK(HandleUntilReturnEvent(&rv, false, nullptr) == RPCCode::kShutdown); CHECK(HandleUntilReturnEvent(&rv, false, nullptr) == RPCCode::kShutdown);
LOG(INFO) << "Shutdown..."; LOG(INFO) << "Shutdown...";
if (const auto* f = Registry::Get("tvm.contrib.rpc.server.shutdown")) {
(*f)();
}
channel_.reset(nullptr); channel_.reset(nullptr);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment