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

[GNA]Fix crash in gna_pluing when using POT (#16003)

Browse Source
This commit is contained in:
Marcin Kusmierski 2023-04-03 17:22:56 +02:00 committed by GitHub
parent dec425c408
commit 0a56927671
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/plugins/intel_gna/src/frontend/model_quantizer.hpp
View File

@ -148,15 +148,30 @@ private:
inf_loop_count++; inf_loop_count++;
} else { } else {
if (inf_loop_count > 0 && if (inf_loop_count > 0 &&
(inf_loop_history.size() % inf_loop_pattern.size() == 0 || sf.allLayersProcessed()) && (inf_loop_pattern.size() > 0 && inf_loop_history.size() % inf_loop_pattern.size() == 0 ||
!std::equal(inf_loop_history.begin() + (inf_loop_history.size() - inf_loop_pattern.size()), sf.allLayersProcessed())) {
int32_t history_shift = 0;
int32_t pattern_shift = 0;
if (inf_loop_history.size() > inf_loop_pattern.size()) {
history_shift = inf_loop_history.size() - inf_loop_pattern.size();
} else {
pattern_shift = inf_loop_pattern.size() - inf_loop_history.size();
}
if (!std::equal(inf_loop_history.begin() + history_shift,
inf_loop_history.end(), inf_loop_history.end(),
inf_loop_pattern.begin())) { inf_loop_pattern.begin() + pattern_shift)) {
inf_loop_count = 0; inf_loop_count = 0;
log::debug() << "inf_loop_pattern:\n";
for (const auto& s : inf_loop_pattern) {
log::debug() << "\t " << s << '\n';
}
inf_loop_pattern.clear(); inf_loop_pattern.clear();
log::debug() << "infinite loop fixed\n"; log::debug() << "infinite loop fixed\n";
} }
} }
}
sf.SetInfiniteLoopCount(inf_loop_count); sf.SetInfiniteLoopCount(inf_loop_count);
} }