free buffer

ZiyueXu77 · Jun 11, 2024 · 454f69d · 454f69d
1 parent 71e578c
commit 454f69d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/collective/aggregator.h b/src/collective/aggregator.h
@@ -132,6 +132,9 @@ void ApplyWithLabels(Context const*, MetaInfo const& info, HostDeviceVector<T>*
       // call HandleGHPairs
       size_t size;
       processor_instance->HandleGHPairs(&size, buffer, buffer_size);
+
+      // free the buffer
+      free(buffer);
     } else {
       // clear text mode, broadcast the data directly
       result->Resize(size);