#1672: lb: move CommMsg to baselb messages

DARMA-tasking · May 31, 2022 · f787829 · f787829
1 parent e68ed98
commit f787829
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 16 deletions.
diff --git a/src/vt/vrt/collection/balance/baselb/baselb.h b/src/vt/vrt/collection/balance/baselb/baselb.h
@@ -179,22 +179,6 @@ struct BaseLB {
   std::shared_ptr<balance::Reassignment> pending_reassignment_ = nullptr;
 };
 
-struct CommMsg : vt::Message {
-  using MessageParentType = vt::Message;
-  vt_msg_serialize_required();
-
-  CommMsg() = default;
-  explicit CommMsg(lb::BaseLB::ElementCommType in_comm) : comm_(in_comm) { }
-
-  lb::BaseLB::ElementCommType comm_;
-
-  template <typename SerializerT>
-  void serialize(SerializerT& s) {
-    MessageParentType::serialize(s);
-    s | comm_;
-  }
-};
-
 }}}} // namespace vt::vrt::collection::lb
 
 #endif /*INCLUDED_VT_VRT_COLLECTION_BALANCE_BASELB_BASELB_H*/
diff --git a/src/vt/vrt/collection/balance/baselb/baselb_msgs.h b/src/vt/vrt/collection/balance/baselb/baselb_msgs.h
@@ -82,6 +82,22 @@ struct CountMsg : vt::collective::ReduceTMsg<int32_t> {
   {}
 };
 
+struct CommMsg : vt::Message {
+  using MessageParentType = vt::Message;
+  vt_msg_serialize_required();
+
+  CommMsg() = default;
+  explicit CommMsg(elm::CommMapType const& in_comm) : comm_(in_comm) { }
+
+  elm::CommMapType comm_;
+
+  template <typename SerializerT>
+  void serialize(SerializerT& s) {
+    MessageParentType::serialize(s);
+    s | comm_;
+  }
+};
+
 }}}} /* end namespace vt::vrt::collection::lb */
 
 #endif /*INCLUDED_VT_VRT_COLLECTION_BALANCE_BASELB_BASELB_MSGS_H*/