deallocate temp storage

NVIDIA · Aug 2, 2024 · 4815848 · 4815848
1 parent c500e5a
commit 4815848
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/include/cuco/detail/open_addressing/open_addressing_impl.cuh b/include/cuco/detail/open_addressing/open_addressing_impl.cuh
@@ -728,6 +728,8 @@ class open_addressing_impl {
                                            this->capacity(),
                                            op,
                                            stream.get()));
+
+    temp_allocator.deallocate(d_temp_storage, temp_storage_bytes);
   }
 
   /**