Skip to content
Snippets Groups Projects
Commit ee514eac authored by Tianqi Chen's avatar Tianqi Chen Committed by GitHub
Browse files

[RPC] callback option rpc server starts (#1092)

parent 9082a93b
No related branches found
No related tags found
No related merge requests found
......@@ -783,6 +783,9 @@ void RPCSession::Shutdown() {
void RPCSession::ServerLoop() {
std::lock_guard<std::recursive_mutex> lock(mutex_);
if (const auto* f = Registry::Get("tvm.contrib.rpc.server.start")) {
(*f)();
}
TVMRetValue rv;
CHECK(HandleUntilReturnEvent(&rv, false, nullptr) == RPCCode::kShutdown);
LOG(INFO) << "Shutdown...";
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment