Add missing ReductionIdentity<char> specialization

sandialabs · Jan 23, 2023 · de26b23 · de26b23
1 parent 2bde8fc
commit de26b23
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/core/src/Kokkos_ReductionIdentity.hpp b/core/src/Kokkos_ReductionIdentity.hpp
@@ -53,6 +53,30 @@ type { static_assert( false, "Missing specialization of
 Kokkos::reduction_identity for custom land reduction type"); return T(); }
 };*/
 
+template <>
+struct reduction_identity<char> {
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char sum() {
+    return static_cast<char>(0);
+  }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char prod() {
+    return static_cast<char>(1);
+  }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char max() { return CHAR_MIN; }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char min() { return CHAR_MAX; }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char bor() {
+    return static_cast<char>(0x0);
+  }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char band() {
+    return ~static_cast<char>(0x0);
+  }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char lor() {
+    return static_cast<char>(0);
+  }
+  KOKKOS_FORCEINLINE_FUNCTION constexpr static char land() {
+    return static_cast<char>(1);
+  }
+};
+
 template <>
 struct reduction_identity<signed char> {
   KOKKOS_FORCEINLINE_FUNCTION constexpr static signed char sum() {