Add multimap size

include/cuco/detail/static_multimap/static_multimap.inl

@@ -456,6 +456,21 @@ void static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator,
   this->impl_->rehash_async(extent, *this, stream);
 }
 
+template <class Key,
+          class T,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::size_type
+static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::size(
+  cuda::stream_ref stream) const
+{
+  return impl_->size(stream);
+}
+
 template <class Key,
           class T,
           class Extent,

include/cuco/static_multimap.cuh

@@ -657,6 +657,16 @@ class static_multimap {
    */
   void rehash_async(size_type capacity, cuda::stream_ref stream = {});
 
+  /**
+   * @brief Gets the number of elements in the container.
+   *
+   * @note This function synchronizes the given stream.
+   *
+   * @param stream CUDA stream used to get the number of inserted elements
+   * @return The number of elements in the container
+   */
+  [[nodiscard]] size_type size(cuda::stream_ref stream = {}) const;
+
   /**
    * @brief Gets the maximum number of elements the hash map can hold.
    *