fix reallocation issue

parthenon-hpc-lab · Oct 24, 2024 · dc54426 · dc54426
1 parent 59c8680
commit dc54426
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/utils/communication_buffer.hpp b/src/utils/communication_buffer.hpp
@@ -101,9 +101,11 @@ class CommBuffer {
   T &buffer() { return buf_; }
   const T &buffer() const { return buf_; }
 
-  void Allocate(int size = 0) {
-    buf_ = get_resource_(size);
-    active_ = true;
+  void Allocate(int size = -1) {
+    if (!active_ || (size > 0 && buf_.size() != size)) {
+      buf_ = get_resource_(size);
+      active_ = true;
+    }
   }
 
   template <class... Args>