add doc for CallbackChannel

apache · Oct 23, 2019 · 097ee8f · 097ee8f
1 parent cedabad
commit 097ee8f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/src/runtime/rpc/rpc_event_impl.cc b/src/runtime/rpc/rpc_event_impl.cc
@@ -32,7 +32,11 @@ namespace runtime {
 PackedFunc CreateEventDrivenServer(PackedFunc fsend,
                                    std::string name,
                                    std::string remote_key) {
-  std::unique_ptr<CallbackChannel> ch(new CallbackChannel(fsend));
+  static PackedFunc frecv([](TVMArgs args, TVMRetValue* rv) {
+    LOG(FATAL) << "Do not allow explicit receive";
+    return 0;
+  });
+  std::unique_ptr<CallbackChannel> ch(new CallbackChannel(fsend, frecv));
   std::shared_ptr<RPCSession> sess =
       RPCSession::Create(std::move(ch), name, remote_key);
   return PackedFunc([sess](TVMArgs args, TVMRetValue* rv) {

diff --git a/src/runtime/rpc/rpc_session.h b/src/runtime/rpc/rpc_session.h
@@ -253,19 +253,15 @@ class RPCSession {
   std::string remote_key_;
 };
 
+/*!
+ * \brief RPC channel which callback
+ * frontend (Python/Java/etc.)'s send & recv function
+ */
 class CallbackChannel final : public RPCChannel {
  public:
   explicit CallbackChannel(PackedFunc fsend, PackedFunc frecv)
       : fsend_(std::move(fsend)), frecv_(std::move(frecv)) {}
 
-  explicit CallbackChannel(PackedFunc fsend)
-      : fsend_(std::move(fsend)) {
-    PackedFunc frecv([](TVMArgs args, TVMRetValue* rv) {
-      LOG(FATAL) << "Do not allow explicit receive";
-      return 0;
-    });
-    frecv_ = frecv;
-  }
   ~CallbackChannel() {}
   /*!
    * \brief Send data over to the channel.