From 71c9c2e61e66588692944fa7f3b0799eba085116 Mon Sep 17 00:00:00 2001
From: Yunsong Wang <yunsongw@nvidia.com>
Date: Fri, 5 Jul 2024 12:34:22 -0700
Subject: [PATCH] Fix a buffer flushing bug with zip iterator

---
 include/cuco/detail/static_multimap/device_view_impl.inl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/include/cuco/detail/static_multimap/device_view_impl.inl b/include/cuco/detail/static_multimap/device_view_impl.inl
index 13e62d709..25aa0a63b 100644
--- a/include/cuco/detail/static_multimap/device_view_impl.inl
+++ b/include/cuco/detail/static_multimap/device_view_impl.inl
@@ -512,7 +512,8 @@ class static_multimap<Key, Value, Scope, Allocator, ProbeSequence>::device_view_
 #endif  // end CUCO_HAS_CUDA_BARRIER
     } else {
       for (auto index = lane_id; index < num_outputs; index += g.size()) {
-        *(output_begin + offset + index) = output_buffer[index];
+        thrust::get<0>(*(output_begin + offset + index)) = output_buffer[index].first;
+        thrust::get<1>(*(output_begin + offset + index)) = output_buffer[index].second;
       }
     }
   }