Skip to content

Commit

Permalink
[snippets][CPU] Fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
a-sidorova committed Nov 25, 2024
1 parent 196f792 commit 93a9f81
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,7 @@ struct BrgemmBaseKernelConfig {
protected:
struct StaticBaseParams {
StaticBaseParams(const element::Type& in0_dtype, const element::Type& in1_dtype, dnnl::impl::cpu::x64::cpu_isa_t primitive_isa);
virtual ~StaticBaseParams() = default;

const dnnl_data_type_t dt_in0 {dnnl_f32}, dt_in1 {dnnl_f32};
const dnnl::impl::cpu::x64::cpu_isa_t isa {dnnl::impl::cpu::x64::isa_undef};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
//

#include "openvino/opsets/opset.hpp"
#include "openvino/runtime/system_conf.hpp"
#include "snippets/snippets_isa.hpp"
#include "snippets/lowered/linear_ir.hpp"
#include "snippets/lowered/pass/mark_loops.hpp"
Expand Down Expand Up @@ -255,6 +256,11 @@ TEST_P(MHAFP32BufferAllocationTest, BufferAllocationCPU) {
}

TEST_P(MHABF16AMXBufferAllocationTest, BufferAllocationCPU) {
// Scratchpad memory for AMX with CopyA (dynamic case) has allocation size which depends on element count in vector register.
// So the current `expected_allocation_size` in the test is targeted on real AVX512 platforms with vector registers with 512 bits.
// If the test infrastructure has AVX2, the allocation size will not be matched.
if (m_linear_ir.is_dynamic() && !with_cpu_x86_avx512_core())
GTEST_SKIP();
Validate();
}

Expand Down

0 comments on commit 93a9f81

Please sign in to comment.