declare network interface

oneflow/core/actor/actor_message.h

@@ -47,6 +47,7 @@ class ActorMsg final {
     CHECK_EQ(msg_type_, ActorMsgType::kPieceModelIdMsg);
     return model_version_id_;
   }
+
   // Setters
   void set_dst_actor_id(int64_t val) { dst_actor_id_ = val; }
   void set_regst_wrapper(std::shared_ptr<RegstWrapper> val) {
@@ -66,6 +67,16 @@ class ActorMsg final {
     model_version_id_ = val;
   }
 
+  // Serialize
+  template<typename StreamT>
+  void Serialize(StreamT& out_stream) const {
+    TODO();
+  }
+  template<typename StreamT>
+  void Deserialize(StreamT& in_stream) {
+    TODO();
+  }
+
  private:
   int64_t dst_actor_id_;
   ActorMsgType msg_type_;

oneflow/core/actor/actor_message_bus.cpp

@@ -13,7 +13,7 @@ void ActorMsgBus::SendMsg(const ActorMsg& msg) {
         IDMgr::Singleton()->ThrdLocId4ActorId(msg.dst_actor_id());
     ThreadMgr::Singleton()->GetThrd(thrd_loc_id)->GetMsgChannelPtr()->Send(msg);
   } else {
-    TODO();
+    CommNetwork::Singleton()->SendActorMsg(dst_machine_id, msg);
   }
 }
 

oneflow/core/actor/actor_message_bus.h

@@ -3,6 +3,7 @@
 
 #include <stdint.h>
 #include "oneflow/core/actor/actor_message.h"
+#include "oneflow/core/comm_network/comm_network.h"
 #include "oneflow/core/common/util.h"
 
 namespace oneflow {
@@ -17,7 +18,8 @@ class ActorMsgBus final {
   void SendMsg(const ActorMsg& msg);
 
   void Init() {
-    // TODO
+    CommNetwork::Singleton()->SetCallbackForReceivedActorMsg(
+        [](const ActorMsg& msg) { ActorMsgBus::Singleton()->SendMsg(msg); });
   }
 
  private:

oneflow/core/comm_network/comm_network.h

@@ -16,6 +16,19 @@ class CommNetwork final {
     // TODO
   }
 
+  void Read(int64_t src_machine_id, void* src_dptr, void* dst_dptr, size_t sz,
+            std::function<void()> callback) {
+    // TODO
+  }
+
+  void SendActorMsg(int64_t dst_machine_id, const ActorMsg& msg) {
+    // TODO
+  }
+  void SetCallbackForReceivedActorMsg(
+      std::function<void(const ActorMsg&)> callback) {
+    // TODO
+  }
+
  private:
   CommNetwork() = default;
 };