#138: Tighten up types

DARMA-tasking · Oct 26, 2020 · fe6baf8 · fe6baf8
1 parent f5d6c77
commit fe6baf8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/checkpoint/dispatch/vrt/object_registry.h b/src/checkpoint/dispatch/vrt/object_registry.h
@@ -77,7 +77,7 @@ struct ObjectEntry {
   std::size_t size_ = 0;                   /**< The registered object size */
   std::function<void*(void)> allocator_;   /**< Do standard allocation for object */
   std::function<T*(void*)> constructor_;   /**< Construct object on memory */
-  std::function<T*(void*)> caster_;        /**< Try to dynamic_cast<ObjT> */
+  std::function<T*(T*)> caster_;           /**< Try to dynamic_cast<ObjT> */
 };
 
 template <typename T>
@@ -115,7 +115,7 @@ Registrar<ObjT>::Registrar() {
       sizeof(ObjT),
       []()          -> void*       { return std::allocator<ObjT>{}.allocate(1); },
       [](void* buf) -> BaseType*   { return dispatch::Reconstructor<ObjT>::constructAllowFail(buf); },
-      [](void* buf) -> ObjT*       { return dynamic_cast<ObjT>(buf); }
+      [](BaseType* buf) -> ObjT*   { return dynamic_cast<ObjT>(buf); }
     }
   );
 }