IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix array out-of-bounds error (#8966)

Browse Source
This commit is contained in:
Mikhail Nosov 2021-12-02 00:07:37 +03:00 committed by GitHub
parent 4e6eeea6ff
commit 3ab533a89c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/src/layout.cpp
View File

@ -277,7 +277,7 @@ Layout apply_permutation(const Layout& src_layout, const std::vector<uint64_t>&
std::vector<int64_t> find_permutation(const Layout& src_layout, const Rank& rank, const Layout& dst) {
auto check_trivial = [](std::vector<int64_t>& res) -> std::vector<int64_t>& {
size_t i = 0;
while (res[i] == i && i < res.size()) {
while (i < res.size() && res[i] == i) {
i++;
}
if (i == res.size()) {

17
src/core/tests/preprocess.cpp
View File

@ -675,6 +675,23 @@ TEST(pre_post_process, preprocess_convert_layout_default) {
EXPECT_EQ(f->get_parameters()[0]->get_output_tensor(0).get_partial_shape(), (PartialShape{1, 2, 2, 3}));
}
TEST(pre_post_process, preprocess_convert_layout_same_various) {
for (size_t i = 1; i < 100; i++) {
auto f = create_simple_function(element::f32, PartialShape::dynamic(static_cast<int64_t>(i)));
auto p = PrePostProcessor(f);
std::stringstream stream;
stream << "[0";
for (auto j = 1; j < i; j++) {
stream << "," << std::to_string(j);
}
stream << "]";
auto l = stream.str();
p.input().tensor().set_layout(ov::Layout(l));
p.input().network().set_layout(ov::Layout(std::string(i, '?')));
EXPECT_NO_THROW(p.build());
}
}
TEST(pre_post_process, preprocess_convert_layout_same) {
auto f = create_simple_function(element::f32, Shape{1, 3, 2, 2});
auto size_old = f->get_ordered_ops().size();