From 1bf03cbe777457a6dc8b9a5bd34da5e8b23d900d Mon Sep 17 00:00:00 2001
From: Andrey Prokopenko <prokopenkoav@ornl.gov>
Date: Tue, 11 Jun 2024 15:20:25 -0400
Subject: [PATCH] Use MPI_BYTE for unknown types

---
 src/impl/KokkosComm_types.hpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/impl/KokkosComm_types.hpp b/src/impl/KokkosComm_types.hpp
index 548f21de..897816c2 100644
--- a/src/impl/KokkosComm_types.hpp
+++ b/src/impl/KokkosComm_types.hpp
@@ -25,10 +25,7 @@ template <typename Scalar>
 MPI_Datatype mpi_type() {
   using T = std::decay_t<Scalar>;
 
-  if constexpr (std::is_same_v<T, std::byte>)
-    return MPI_BYTE;
-
-  else if constexpr (std::is_same_v<T, char>)
+  if constexpr (std::is_same_v<T, char>)
     return MPI_CHAR;
   else if constexpr (std::is_same_v<T, unsigned char>)
     return MPI_UNSIGNED_CHAR;
@@ -99,6 +96,9 @@ MPI_Datatype mpi_type() {
   else if constexpr (std::is_same_v<T, Kokkos::complex<double>>)
     return MPI_DOUBLE_COMPLEX;
 
+  else if constexpr (std::is_trivially_copyable_v<T>)
+    return MPI_BYTE;
+
   else {
     static_assert(std::is_void_v<T>, "mpi_type not implemented");
     return MPI_CHAR;  // unreachable