#2092: location: call function pointer directly

DARMA-tasking · Feb 14, 2023 · 31dc65b · 31dc65b
1 parent a4ea3a8
commit 31dc65b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/vt/topos/location/location.h b/src/vt/topos/location/location.h
@@ -248,7 +248,7 @@ struct EntityLocationCoord : LocationCoord {
    *
    * \param[in] m message shared pointer
    */
-  template <typename MessageT>
+  template <typename MessageT, ActiveTypedFnType<MessageT> *f>
   void routePreparedMsgHandler(MsgSharedPtr<MessageT> const& msg);
 
   /**
@@ -264,7 +264,7 @@ struct EntityLocationCoord : LocationCoord {
    *
    * \param[in] m message shared pointer
    */
-  template <typename MessageT>
+  template <typename MessageT, ActiveTypedFnType<MessageT> *f>
   void routeMsgHandlerLocal(MsgSharedPtr<MessageT> const& msg);
 
   /**

diff --git a/src/vt/topos/location/location.impl.h b/src/vt/topos/location/location.impl.h
@@ -663,14 +663,14 @@ void EntityLocationCoord<EntityID>::routeMsgHandler(
 }
 
 template <typename EntityID>
-template <typename MessageT>
+template <typename MessageT, ActiveTypedFnType<MessageT> *f>
 void EntityLocationCoord<EntityID>::routePreparedMsgHandler(
   MsgSharedPtr<MessageT> const& msg
 ) {
   if (local_registered_.find(msg->getEntity()) == local_registered_.end()) {
     return routePreparedMsg(msg);
   } else {
-    return routeMsgHandlerLocal(msg);
+    return routeMsgHandlerLocal<MessageT, f>(msg);
   }
 }
 
@@ -698,7 +698,7 @@ void EntityLocationCoord<EntityID>::setupMessageForRouting(
 }
 
 template <typename EntityID>
-template <typename MessageT>
+template <typename MessageT, ActiveTypedFnType<MessageT> *f>
 void EntityLocationCoord<EntityID>::routeMsgHandlerLocal(
   MsgSharedPtr<MessageT> const& msg
 ) {

diff --git a/src/vt/vrt/collection/manager.impl.h b/src/vt/vrt/collection/manager.impl.h
@@ -1168,7 +1168,7 @@ messaging::PendingSend CollectionManager::sendMsgUntypedHandler(
     msg, [](MsgSharedPtr<BaseMsgType>& inner_msg){
       auto typed_msg = inner_msg.template to<MsgT>();
       auto lm2 = theLocMan()->getCollectionLM<IdxT>(typed_msg->getLocInst());
-      lm2->template routePreparedMsgHandler<MsgT>(typed_msg);
+      lm2->template routePreparedMsgHandler<MsgT, collectionMsgTypedHandler<ColT,IdxT,MsgT>>(typed_msg);
     }
   };
 }