fix

Author: aobolensk

src/plugins/intel_cpu/src/emitters/snippets/x64/kernel_executors/brgemm_batched.cpp

@@ -224,10 +224,7 @@ void BrgemmBatchedKernelExecutor::execute(const BrgemmBatchedKernelExecutor* exe
     auto iter_count = config.get_iter_count();
     auto iter_size = config.get_K();
 
-    if (config.get_K() % iter_count) {  // test
-        --iter_count;
-    }
-
+    OPENVINO_ASSERT(iter_size % iter_count == 0, "Incompatible iterations count in brgemm");
     iter_size /= iter_count;
 
     size_t stride_A = iter_size * dnnl_data_type_size(config.get_dt_in0());

src/plugins/intel_cpu/src/transformations/snippets/x64/pass/lowered/build_brgemm.cpp

@@ -62,6 +62,9 @@ bool pass::BuildBrgemm::run(snippets::lowered::LinearIR& linear_ir,
             continue;
         }
 
+        if (inner_loop_info->get_work_amount() % inner_loop_info->get_increment() != 0) {
+            continue;
+        }
         auto iter_count = inner_loop_info->get_work_amount() / inner_loop_info->get_increment();
 
         std::shared_ptr<BrgemmCPU> brgemm_node;

src/plugins/intel_cpu/tests/functional/shared_tests_instances/snippets/matmul.cpp

@@ -34,6 +34,13 @@ std::vector<std::vector<ov::test::InputShape>> input_shapes{
     { {{}, {{1, 1, 100, 1024}}},   {{}, {{1, 1, 1024, 100}}} },
     { {{}, {{1, 1, 100, 2500}}},   {{}, {{1, 1, 2500, 100}}} },
     { {{}, {{1, 1, 100, 4500}}},   {{}, {{1, 1, 4500, 100}}} },
+    { {{}, {{1, 1, 100, 2048}}},   {{}, {{1, 1, 2048, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 32}}},   {{}, {{1, 1, 1024 + 32, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 16}}},   {{}, {{1, 1, 1024 + 16, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 8}}},   {{}, {{1, 1, 1024 + 8, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 4}}},   {{}, {{1, 1, 1024 + 4, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 2}}},   {{}, {{1, 1, 1024 + 2, 100}}} },
+    { {{}, {{1, 1, 100, 1024 + 1}}},   {{}, {{1, 1, 1024 + 1, 100}}} },
     // Only M dimension is dynamic + one one loop by M
     {
         {PartialShape{-1, 2, -1, 64}, {{2, 2, 64, 64}, {2, 2, 64, 64}, {2, 2, 35, 64},