Expose pointer to underlying storage

Author: sleeepyjack

include/cuco/detail/open_addressing/open_addressing_impl.cuh

@@ -1086,6 +1086,13 @@ class open_addressing_impl {
    */
   [[nodiscard]] constexpr auto capacity() const noexcept { return storage_.capacity(); }
 
+  /**
+   * @brief Gets a pointer to the underlying slot storage.
+   *
+   * @return Pointer to the underlying slot storage
+   */
+  [[nodiscard]] __host__ value_type* data() const { return storage_.data(); }
+
   /**
    * @brief Gets the sentinel value used to represent an empty key slot.
    *

include/cuco/detail/static_map/static_map.inl

@@ -798,6 +798,20 @@ static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::
   return impl_->capacity();
 }
 
+template <class Key,
+          class T,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+__host__ auto static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::data()
+  const -> value_type*
+{
+  return impl_->data();
+}
+
 template <class Key,
           class T,
           class Extent,

include/cuco/detail/static_multimap/static_multimap.inl

@@ -618,6 +618,21 @@ static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Stora
   return impl_->capacity();
 }
 
+template <class Key,
+          class T,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+__host__ auto
+static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::data() const
+  -> value_type*
+{
+  return impl_->data();
+}
+
 template <class Key,
           class T,
           class Extent,

include/cuco/detail/static_multiset/static_multiset.inl

@@ -657,6 +657,20 @@ static_multiset<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>
   return impl_->capacity();
 }
 
+template <class Key,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+__host__ auto
+static_multiset<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::data() const
+  -> value_type*
+{
+  return impl_->data();
+}
+
 template <class Key,
           class Extent,
           cuda::thread_scope Scope,

include/cuco/detail/static_set/static_set.inl

@@ -621,6 +621,19 @@ static_set<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::cap
   return impl_->capacity();
 }
 
+template <class Key,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+__host__ auto static_set<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::data()
+  const -> value_type*
+{
+  return impl_->data();
+}
+
 template <class Key,
           class Extent,
           cuda::thread_scope Scope,

include/cuco/static_map.cuh

@@ -1116,6 +1116,13 @@ class static_map {
    */
   [[nodiscard]] constexpr auto capacity() const noexcept;
 
+  /**
+   * @brief Gets a pointer to the underlying slot storage.
+   *
+   * @return Pointer to the underlying slot storage
+   */
+  [[nodiscard]] __host__ value_type* data() const;
+
   /**
    * @brief Gets the sentinel value used to represent an empty key slot.
    *

include/cuco/static_multimap.cuh

@@ -863,6 +863,13 @@ class static_multimap {
    */
   [[nodiscard]] constexpr auto capacity() const noexcept;
 
+  /**
+   * @brief Gets a pointer to the underlying slot storage.
+   *
+   * @return Pointer to the underlying slot storage
+   */
+  [[nodiscard]] __host__ value_type* data() const;
+
   /**
    * @brief Gets the sentinel value used to represent an empty key slot.
    *

include/cuco/static_multiset.cuh

@@ -101,7 +101,6 @@ class static_multiset {
   using storage_ref_type    = typename impl_type::storage_ref_type;
   using probing_scheme_type = typename impl_type::probing_scheme_type;  ///< Probing scheme type
   using hasher              = typename probing_scheme_type::hasher;     ///< Hash function type
-
   template <typename... Operators>
   using ref_type = cuco::static_multiset_ref<key_type,
                                              thread_scope,
@@ -962,6 +961,13 @@ class static_multiset {
    */
   [[nodiscard]] constexpr auto capacity() const noexcept;
 
+  /**
+   * @brief Gets a pointer to the underlying slot storage.
+   *
+   * @return Pointer to the underlying slot storage
+   */
+  [[nodiscard]] __host__ value_type* data() const;
+
   /**
    * @brief Gets the sentinel value used to represent an empty key slot.
    *

include/cuco/static_set.cuh

@@ -106,7 +106,6 @@ class static_set {
   using storage_ref_type    = typename impl_type::storage_ref_type;
   using probing_scheme_type = typename impl_type::probing_scheme_type;  ///< Probing scheme type
   using hasher              = typename probing_scheme_type::hasher;     ///< Hash function type
-
   template <typename... Operators>
   using ref_type = cuco::static_set_ref<key_type,
                                         thread_scope,
@@ -934,6 +933,13 @@ class static_set {
    */
   [[nodiscard]] constexpr auto capacity() const noexcept;
 
+  /**
+   * @brief Gets a pointer to the underlying slot storage.
+   *
+   * @return Pointer to the underlying slot storage
+   */
+  [[nodiscard]] __host__ value_type* data() const;
+
   /**
    * @brief Gets the sentinel value used to represent an empty key slot.
    *