Skip to content

Commit bec3062

Browse files
committed
Fix tests
1 parent 215a629 commit bec3062

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/plugins/intel_cpu/tests/unit/snippets_transformations/x64/lowered/buffer_allocation.cpp

+7-7
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ class MHABF16AMXBufferAllocationTest : public BufferAllocationCPUTest {
191191
std::shared_ptr<ov::Model> GetModel(const std::vector<ov::PartialShape>& shapes) const override {
192192
const auto subtensor_scalar = std::vector<size_t>{1};
193193
const auto subtensor_power = std::vector<size_t>{1, ov::snippets::utils::get_full_dim_value()};
194-
const auto subtensor_full = std::vector<size_t>(2, ov::snippets::utils::get_full_dim_value());
194+
const auto subtensor_flat = std::vector<size_t>(1, ov::snippets::utils::get_full_dim_value());
195195

196196
OPENVINO_ASSERT(shapes.size() == 3, "Incorrect count of input shapes");
197197
const auto parameter0 = std::make_shared<ov::op::v0::Parameter>(ov::element::bf16, shapes[0]);
@@ -238,12 +238,12 @@ class MHABF16AMXBufferAllocationTest : public BufferAllocationCPUTest {
238238
MarkOp(store, subtensor_scalar);
239239
MarkOp(power, subtensor_power);
240240

241-
MarkOp(brgemm_cpu0, subtensor_full);
242-
MarkOp(brgemm_cpu1, subtensor_full);
243-
MarkOp(brgemm_copyb0, subtensor_full);
244-
MarkOp(brgemm_copyb1, subtensor_full);
245-
MarkOp(scratch0, subtensor_full);
246-
MarkOp(scratch1, subtensor_full);
241+
MarkOp(brgemm_cpu0, subtensor_flat);
242+
MarkOp(brgemm_cpu1, subtensor_flat);
243+
MarkOp(brgemm_copyb0, subtensor_flat);
244+
MarkOp(brgemm_copyb1, subtensor_flat);
245+
MarkOp(scratch0, subtensor_flat);
246+
MarkOp(scratch1, subtensor_flat);
247247

248248
ov::snippets::lowered::PortDescriptorUtils::get_port_descriptor_ptr(load_reshape->input(0))->set_layout(order);
249249

0 commit comments

Comments
 (0)