fix a bug with heap buffer overflow (openvinotoolkit#5793)

rnugmanx · Aug 26, 2021 · 5ad4dc6 · 5ad4dc6
1 parent 25a5e64
commit 5ad4dc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/gna_plugin/layers/gna_layer_info.hpp b/inference-engine/src/gna_plugin/layers/gna_layer_info.hpp
@@ -264,7 +264,7 @@ class LayerInfo {
                 return false;
             }
             // check dims in between
-            for (int j = permute.first + 1; j != permute.second; j++) {
+            for (int j = std::min(permute.first, permute.second) + 1; j < std::max(permute.first, permute.second); j++) {
                 if (inputsOrderTransformed[j] != 1) {
                     return false;
                 }