incorporate review feedback

ROCm · May 27, 2024 · 31459c7 · 31459c7
1 parent 6ec5a5e
commit 31459c7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/include/migraphx/register_target.hpp b/src/include/migraphx/register_target.hpp
@@ -65,6 +65,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));