fix rc

Signed-off-by: fishbell <[email protected]>
songbell · Dec 25, 2023 · 49b7d7e · 49b7d7e
1 parent bc121c0
commit 49b7d7e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/core/include/openvino/core/layout.hpp b/src/core/include/openvino/core/layout.hpp
@@ -216,6 +216,9 @@ class OPENVINO_API LayoutAttribute : public ov::RuntimeAttribute {
     std::string to_string() const override;
 
     Layout value;
+
+private:
+    std::shared_ptr<std::mutex> m_mutex = std::make_shared<std::mutex>();
 };
 
 }  // namespace ov
diff --git a/src/core/src/layout.cpp b/src/core/src/layout.cpp
@@ -638,6 +638,7 @@ void AttributeAdapter<ov::Layout>::set(const std::string& value) {
 }
 
 bool LayoutAttribute::visit_attributes(AttributeVisitor& visitor) {
+    std::lock_guard<std::mutex> lock{*m_mutex.get()};
     std::string layout_str = value.to_string();
     visitor.on_attribute("layout", layout_str);
     value = Layout(layout_str);