Expose key_eq member accessors (#584)

Currently, `key_eq` is only accessible in ref classes, not in container
classes. This PR addresses the issue by exposing `key_eq` in all
classes.

include/cuco/detail/static_map/static_map.inl

@@ -709,6 +709,21 @@ static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::
   return impl_->erased_key_sentinel();
 }
 
+template <class Key,
+          class T,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+constexpr static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_equal
+static_map<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_eq()
+  const noexcept
+{
+  return impl_->key_eq();
+}
+
 template <class Key,
           class T,
           class Extent,

include/cuco/detail/static_multimap/static_multimap.inl

@@ -312,6 +312,22 @@ constexpr static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Alloca
   return impl_->erased_key_sentinel();
 }
 
+template <class Key,
+          class T,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+constexpr static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::
+  key_equal
+  static_multimap<Key, T, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_eq()
+    const noexcept
+{
+  return impl_->key_eq();
+}
+
 template <class Key,
           class T,
           class Extent,

include/cuco/detail/static_multiset/static_multiset.inl

@@ -393,6 +393,21 @@ static_multiset<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>
   return impl_->erased_key_sentinel();
 }
 
+template <class Key,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+constexpr static_multiset<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::
+  key_equal
+  static_multiset<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_eq()
+    const noexcept
+{
+  return impl_->key_eq();
+}
+
 template <class Key,
           class Extent,
           cuda::thread_scope Scope,

include/cuco/detail/static_set/static_set.inl

@@ -514,6 +514,19 @@ static_set<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::era
   return impl_->erased_key_sentinel();
 }
 
+template <class Key,
+          class Extent,
+          cuda::thread_scope Scope,
+          class KeyEqual,
+          class ProbingScheme,
+          class Allocator,
+          class Storage>
+constexpr static_set<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_equal
+static_set<Key, Extent, Scope, KeyEqual, ProbingScheme, Allocator, Storage>::key_eq() const noexcept
+{
+  return impl_->key_eq();
+}
+
 template <class Key,
           class Extent,
           cuda::thread_scope Scope,

include/cuco/static_map.cuh

@@ -952,6 +952,13 @@ class static_map {
    */
   [[nodiscard]] constexpr key_type erased_key_sentinel() const noexcept;
 
+  /**
+   * @brief Gets the function used to compare keys for equality
+   *
+   * @return The function used to compare keys for equality
+   */
+  [[nodiscard]] constexpr key_equal key_eq() const noexcept;
+
   /**
    * @brief Get device ref with operators.
    *

include/cuco/static_multimap.cuh

@@ -447,6 +447,13 @@ class static_multimap {
    */
   [[nodiscard]] constexpr key_type erased_key_sentinel() const noexcept;
 
+  /**
+   * @brief Gets the function used to compare keys for equality
+   *
+   * @return The function used to compare keys for equality
+   */
+  [[nodiscard]] constexpr key_equal key_eq() const noexcept;
+
   /**
    * @brief Get device ref with operators.
    *

include/cuco/static_multiset.cuh

@@ -570,6 +570,13 @@ class static_multiset {
    */
   [[nodiscard]] constexpr key_type erased_key_sentinel() const noexcept;
 
+  /**
+   * @brief Gets the function used to compare keys for equality
+   *
+   * @return The function used to compare keys for equality
+   */
+  [[nodiscard]] constexpr key_equal key_eq() const noexcept;
+
   /**
    * @brief Get device ref with operators.
    *

include/cuco/static_set.cuh

@@ -766,6 +766,13 @@ class static_set {
    */
   [[nodiscard]] constexpr key_type erased_key_sentinel() const noexcept;
 
+  /**
+   * @brief Gets the function used to compare keys for equality
+   *
+   * @return The function used to compare keys for equality
+   */
+  [[nodiscard]] constexpr key_equal key_eq() const noexcept;
+
   /**
    * @brief Get device ref with operators.
    *