#1983: location: bypass complex logic when entity is local

DARMA-tasking · Sep 28, 2022 · 4c4bb4a · 4c4bb4a
1 parent 452df8a
commit 4c4bb4a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/vt/topos/location/location.h b/src/vt/topos/location/location.h
@@ -237,6 +237,14 @@ struct EntityLocationCoord : LocationCoord {
     MsgSharedPtr<MessageT> const& msg
   );
 
+  /**
+   * \brief Route a message with a custom handler where the element is local
+   *
+   * \param[in] m message shared pointer
+   */
+  template <typename MessageT>
+  void routeMsgHandlerLocal(MsgSharedPtr<MessageT> const& msg);
+
   /**
    * \brief Route a message to the default handler
    *

diff --git a/src/vt/topos/location/location.impl.h b/src/vt/topos/location/location.impl.h
@@ -668,7 +668,22 @@ void EntityLocationCoord<EntityID>::routeMsgHandler(
 # endif
 
   msg->setHandler(handler);
-  return routeMsg<MessageT>(id,home_node,msg);
+
+  if (local_registered_.find(id) == local_registered_.end()) {
+    return routeMsg<MessageT>(id,home_node,msg);
+  } else {
+    return routeMsgHandlerLocal(msg);
+  }
+}
+
+template <typename EntityID>
+template <typename MessageT>
+void EntityLocationCoord<EntityID>::routeMsgHandlerLocal(
+  MsgSharedPtr<MessageT> const& msg
+) {
+  runnable::makeRunnable(msg, true, msg->getHandler(), theContext()->getNode())
+    .withTDEpochFromMsg()
+    .run();
 }
 
 template <typename EntityID>