Merge pull request #22 from maartenarnst/type_list_contains

Add predicate to know whether a type is contained in a type list
uliegecsm · Oct 21, 2024 · 961c6ff · 961c6ff
2 parents fb6841e + 4fdec8e
commit 961c6ff
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/include/kokkos-utils/impl/type_list.hpp b/include/kokkos-utils/impl/type_list.hpp
@@ -24,6 +24,18 @@ template <typename T>
 inline constexpr size_t type_list_size_v = TypeListSize<T>::value;
 ///@}
 
+//! @name Whether a type is contained in a @c Kokkos::Impl::type_list.
+///@{
+template <typename, typename>
+struct TypeListContains;
+
+template <typename T, typename... Ts>
+struct TypeListContains<T, Kokkos::Impl::type_list<Ts...>> : std::disjunction<std::is_same<T, Ts>...> {};
+
+template <typename T, typename S>
+inline constexpr bool type_list_contains_v = TypeListContains<T, S>::value;
+///@}
+
 //! @name Get the @p I th type in a @c Kokkos::Impl::type_list.
 ///@{
 template <size_t, typename>

diff --git a/tests/impl/test_type_list.cpp b/tests/impl/test_type_list.cpp
@@ -28,6 +28,15 @@ TEST(impl, type_list_size_v)
     static_assert(type_list_size_v<type_list_t> == 3);
 }
 
+//! @test Check @ref Kokkos::utils::impl::type_list_contains_v.
+TEST(impl, type_list_contains_v)
+{
+    using Kokkos::utils::impl::type_list_contains_v;
+
+    static_assert(   type_list_contains_v<char,   type_list_t>);
+    static_assert( ! type_list_contains_v<double, type_list_t>);
+}
+
 //! @test Check @ref Kokkos::utils::impl::type_list_at_t.
 TEST(impl, type_list_at_t)
 {