Adapt to buffer pool to fix the new introduced apps. (#164)

alibaba · Jul 26, 2024 · 16b38c0 · 16b38c0
1 parent f8f50fb
commit 16b38c0
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/examples/analytical_apps/kclique/kclique_utils.h b/examples/analytical_apps/kclique/kclique_utils.h
@@ -66,6 +66,13 @@ struct KCliqueMsg {
   VID_T* data;
 };
 
+template <typename VID_T>
+struct SerializedSize<KCliqueMsg<VID_T>> {
+  static size_t size(const KCliqueMsg<VID_T>& msg) {
+    return sizeof(int) + sizeof(int) + msg.size * sizeof(VID_T);
+  }
+};
+
 template <typename VID_T>
 InArchive& operator<<(InArchive& arc, const KCliqueMsg<VID_T>& msg) {
   arc << msg.prefix_size;
@@ -82,6 +89,14 @@ OutArchive& operator>>(OutArchive& arc, KCliqueMsg<VID_T>& msg) {
   return arc;
 }
 
+template <typename VID_T>
+FixedInArchive& operator<<(FixedInArchive& arc, const KCliqueMsg<VID_T>& msg) {
+  arc << msg.prefix_size;
+  arc << msg.size;
+  arc.add_bytes(msg.data, msg.size * sizeof(VID_T));
+  return arc;
+}
+
 template <typename FRAG_T>
 struct KCliqueUtils {
   using fragment_t = FRAG_T;

diff --git a/grape/worker/worker.h b/grape/worker/worker.h
@@ -86,7 +86,9 @@ class Worker {
 
     comm_spec_ = comm_spec;
     MPI_Barrier(comm_spec_.comm());
-    context_ = std::make_shared<context_t>(graph);
+    if (context_ == nullptr) {
+      context_ = std::make_shared<context_t>(graph);
+    }
 
     initPool(pe_spec);
     messages_.Init(comm_spec_.comm());