From ee514eac21bb49d3a41c65117b21d7830347d600 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Sun, 8 Apr 2018 17:55:13 -0700
Subject: [PATCH] [RPC] callback option rpc server starts (#1092)
---
src/runtime/rpc/rpc_session.cc | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/runtime/rpc/rpc_session.cc b/src/runtime/rpc/rpc_session.cc
index 2c38294b9..923194cf4 100644
--- a/src/runtime/rpc/rpc_session.cc
+++ b/src/runtime/rpc/rpc_session.cc
@@ -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...";
--
GitLab