revert some more changes

ROCm · Jun 6, 2024 · 165d0ec · 165d0ec
1 parent bcd8ad1
commit 165d0ec
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/include/migraphx/register_target.hpp b/src/include/migraphx/register_target.hpp
@@ -34,6 +34,7 @@
 namespace migraphx {
 inline namespace MIGRAPHX_INLINE_NS {
 
+MIGRAPHX_EXPORT void register_target_init();
 MIGRAPHX_EXPORT void register_target(const target& t);
 MIGRAPHX_EXPORT void unregister_target(const std::string& name);
 MIGRAPHX_EXPORT target make_target(const std::string& name);
@@ -52,6 +53,7 @@ struct target_handler
 template <class T>
 void register_target()
 {
+    register_target_init();
     static auto t_h = detail::target_handler(T{});
     register_target(t_h.t);
 }
@@ -65,6 +67,9 @@ struct register_target_action
     }
 };
 
+template <class T>
+using auto_register_target = auto_register<register_target_action, T>;
+
 #define MIGRAPHX_REGISTER_TARGET(...) MIGRAPHX_AUTO_REGISTER(register_target_action, __VA_ARGS__)
 
 } // namespace MIGRAPHX_INLINE_NS

diff --git a/src/register_target.cpp b/src/register_target.cpp
@@ -44,6 +44,8 @@ std::unordered_map<std::string, target>& target_map()
     return m;
 }
 
+void register_target_init() { (void)target_map(); }
+
 void unregister_target(const std::string& name)
 {
     assert(target_map().count(name));