From 7e75cd0d1e2aa5db613b683639db5ad48feab741 Mon Sep 17 00:00:00 2001
From: Yunsong Wang <yunsongw@nvidia.com>
Date: Mon, 19 Aug 2024 14:35:50 -0700
Subject: [PATCH] Add is_double_hashing trait

---
 include/cuco/probing_scheme.cuh | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/include/cuco/probing_scheme.cuh b/include/cuco/probing_scheme.cuh
index 5591cb74a..e8243c2a8 100644
--- a/include/cuco/probing_scheme.cuh
+++ b/include/cuco/probing_scheme.cuh
@@ -19,6 +19,8 @@
 #include <cuco/detail/probing_scheme/probing_scheme_base.cuh>
 #include <cuco/pair.cuh>
 
+#include <cuda/std/type_traits>
+
 #include <cooperative_groups.h>
 
 namespace cuco {
@@ -198,6 +200,24 @@ class double_hashing : private detail::probing_scheme_base<CGSize> {
   Hash2 hash2_;
 };
 
+/**
+ * @brief Trait indicating whether the given probing scheme is of `double_hashing` type or not
+ *
+ * @tparam T Input probing scheme type
+ */
+template <typename T>
+struct is_double_hashing : cuda::std::false_type {};
+
+/**
+ * @brief Trait indicating whether the given probing scheme is of `double_hashing` type or not
+ *
+ * @tparam CGSize Size of CUDA Cooperative Groups
+ * @tparam Hash1 Unary callable type
+ * @tparam Hash2 Unary callable type
+ */
+template <int32_t CGSize, typename Hash1, typename Hash2>
+struct is_double_hashing<cuco::double_hashing<CGSize, Hash1, Hash2>> : cuda::std::true_type {};
+
 }  // namespace cuco
 
 #include <cuco/detail/probing_scheme/probing_scheme_impl.inl>