make actor_msg in c++

oneflow/actor/actor.h

@@ -4,7 +4,7 @@
 #include "common/util.h"
 #include "kernel/kernel.h"
 #include "common/task.pb.h"
-#include "actor/actor_message.pb.h"
+#include "actor/actor_message.h"
 #include "register/register.h"
 
 namespace oneflow {

oneflow/actor/actor_message.cpp

+#include "actor/actor_message.h"
+
+namespace oneflow {
+
+ActorMsg::ActorMsg() {
+  TODO();
+}
+
+} // namespace oneflow

oneflow/actor/actor_message.h

+#ifndef ONEFLOW_ACTOR_ACTOR_MESSAGE_H_
+#define ONEFLOW_ACTOR_ACTOR_MESSAGE_H_
+
+#include "common/util.h"
+#include "register/register.h"
+
+namespace oneflow {
+
+class ActorMsg final {
+ public:
+  // OF_DISALLOW_COPY_AND_MOVE(ActorMsg);
+  ActorMsg();
+  ~ActorMsg() = default;
+
+  // Getters
+  uint64_t dst_actor_id() const { return dst_actor_id_; }
+  Regst* regst() const { return regst_; }
+
+  // Setters
+  void set_dst_actor_id(uint64_t val) { dst_actor_id_ = val; }
+  void set_regst(Regst* val) { regst_ = val; }
+
+ private:
+  uint64_t dst_actor_id_;
+  Regst* regst_;
+
+};
+
+} // namespace oneflow
+
+#endif // ONEFLOW_ACTOR_ACTOR_MESSAGE_H_

oneflow/actor/actor_message.proto

-syntax = "proto3";
-package oneflow;
-
-message ActorMsg {
-  uint64 dst_actor_id = 1;
-  uint64 register_id = 2;
-}

oneflow/actor/actor_msg_bus.h

@@ -3,7 +3,7 @@
 
 #include <stdint.h>
 #include "common/util.h"
-#include "actor/actor_message.pb.h"
+#include "actor/actor_message.h"
 
 namespace oneflow {
 

oneflow/register/register.h

@@ -2,8 +2,6 @@
 #define ONEFLOW_REGISTER_REGISTER_H_
 
 #include "register/blob.h"
-#include "actor/actor_message.pb.h"
-#include "actor/actor_msg_bus.h"
 #include "common/util.h"
 #include "register/runtime_register_desc.h"