Fixed address sanitizer (#9364)

2021-12-23 14:02:44 +03:00
parent a94a6a774e
commit d0cdf14c47
1 changed files with 4 additions and 1 deletions
--- a/docs/template_plugin/tests/functional/op_reference/base_reference_test.hpp
+++ b/docs/template_plugin/tests/functional/op_reference/base_reference_test.hpp
@@ -58,7 +58,10 @@ ov::runtime::Tensor CreateTensor(const ov::Shape& shape,
                                 const ov::element::Type& element_type,
                                 const std::vector<T>& values) {
    ov::runtime::Tensor tensor{element_type, shape};
-    std::memcpy(tensor.data(), values.data(), sizeof(T) * values.size());
+    size_t size = sizeof(T) * values.size();
+    if (tensor.get_byte_size() < size)
+        size = tensor.get_byte_size();
+    std::memcpy(tensor.data(), values.data(), size);

    return tensor;
 }