From 48158483d1e2870718376b5ad68456fee75b1bba Mon Sep 17 00:00:00 2001
From: Srinivas Yadav Singanaboina <ssinganaboin@nvidia.com>
Date: Fri, 2 Aug 2024 17:46:36 +0000
Subject: [PATCH] deallocate temp storage

---
 include/cuco/detail/open_addressing/open_addressing_impl.cuh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/include/cuco/detail/open_addressing/open_addressing_impl.cuh b/include/cuco/detail/open_addressing/open_addressing_impl.cuh
index 23f4e4c3e..8ba3724cc 100644
--- 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);
   }
 
   /**