fix compilation on Windows

ROCm · Apr 22, 2024 · 6915c5d · 6915c5d
1 parent e4013bb
commit 6915c5d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/include/migraphx/argument.hpp b/src/include/migraphx/argument.hpp
@@ -117,7 +117,7 @@ struct MIGRAPHX_EXPORT argument : raw_data<argument>
     data_t m_data{};
 };
 
-std::vector<argument> flatten(const std::vector<argument>& args);
+MIGRAPHX_EXPORT std::vector<argument> flatten(const std::vector<argument>& args);
 
 MIGRAPHX_EXPORT std::vector<shape> to_shapes(const std::vector<argument>& args);
 MIGRAPHX_EXPORT void migraphx_to_value(value& v, const argument& a);

diff --git a/src/include/migraphx/shape.hpp b/src/include/migraphx/shape.hpp
@@ -429,7 +429,7 @@ struct MIGRAPHX_EXPORT shape
 };
 
 /// Flatten subshapes to a single vector of non-tuple type of shapes
-std::vector<shape> flatten(const std::vector<shape>& shapes);
+MIGRAPHX_EXPORT std::vector<shape> flatten(const std::vector<shape>& shapes);
 
 MIGRAPHX_EXPORT void migraphx_to_value(value& v, const shape& s);
 MIGRAPHX_EXPORT void migraphx_from_value(const value& v, shape& s);