From 4b35d33d25abe661f143d25d542620c7289955f7 Mon Sep 17 00:00:00 2001
From: Carl Pearson <cwpears@sandia.gov>
Date: Wed, 15 May 2024 12:42:11 -0600
Subject: [PATCH] mpi: send profile region

---
 src/impl/KokkosComm_send.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/impl/KokkosComm_send.hpp b/src/impl/KokkosComm_send.hpp
index f614afde..09186305 100644
--- a/src/impl/KokkosComm_send.hpp
+++ b/src/impl/KokkosComm_send.hpp
@@ -28,6 +28,7 @@ namespace KokkosComm::Impl {
 
 template <KokkosView SendView>
 void send(const SendView &sv, int dest, int tag, MPI_Comm comm) {
+  Kokkos::Tools::pushRegion("KokkosComm::Impl::send");
   using KCT = typename KokkosComm::Traits<SendView>;
 
   if (KCT::is_contiguous(sv)) {
@@ -36,6 +37,7 @@ void send(const SendView &sv, int dest, int tag, MPI_Comm comm) {
   } else {
     throw std::runtime_error("only contiguous views supported for low-level send");
   }
+  Kokkos::Tools::popRegion();
 }
 
 template <CommMode SendMode = CommMode::Default, KokkosExecutionSpace ExecSpace, KokkosView SendView>