[Snippets] Support decreasing output shapes (#9446)

2021-12-28 16:22:45 +03:00
parent ecbeff460a
commit 0803684f9e
1 changed files with 2 additions and 2 deletions
--- a/src/common/snippets/src/op/subgraph.cpp
+++ b/src/common/snippets/src/op/subgraph.cpp
@@ -264,8 +264,8 @@ snippets::Schedule snippets::op::Subgraph::generate(const BlockedShapeVector& ou

        for (size_t i = 0; i < work_size.size(); i++) {
            if (work_size[i] != shape[i]) {
-                if (work_size[i] == 1) {
-                    work_size[i] = shape[i];
+                if (work_size[i] == 1 || shape[i] == 1) {
+                    work_size[i] = max(work_size[i], shape[i]);
                } else {
                    throw ngraph_error("incompatible shapes for output graphs");
                }