[CPU] Move to oneDNN v2.7 (#12281)
This commit is contained in:
1 parent
4ec5499c5d
commit
7e2cefab8a
30 files changed
+46
-36
No files matched your search
@@ -33,7 +33,7 @@ public:
|
||||
|
||||
~jit_snippet() = default;
|
||||
|
||||
jit_snippet() : jit_generator() {
|
||||
jit_snippet() : jit_generator(jit_name()) {
|
||||
}
|
||||
|
||||
void generate() override {
|
||||
|
||||
@@ -50,7 +50,7 @@ struct jit_uni_bin_conv_kernel_f32 : public jit_uni_bin_conv_kernel, public jit_
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_bin_conv_kernel_f32)
|
||||
|
||||
explicit jit_uni_bin_conv_kernel_f32(jit_bin_conv_params jcp, jit_dw_conv_params jcp_dw_conv, const dnnl_primitive_attr &attr) :
|
||||
jit_uni_bin_conv_kernel(jcp, jcp_dw_conv, attr), jit_generator() {}
|
||||
jit_uni_bin_conv_kernel(jcp, jcp_dw_conv, attr), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -127,7 +127,8 @@ protected:
|
||||
};
|
||||
|
||||
jit_uni_converter::jit_uni_converter()
|
||||
: _consts(*this) {
|
||||
: jit_kernel(jit_name()),
|
||||
_consts(*this) {
|
||||
}
|
||||
|
||||
void jit_uni_converter::init() {
|
||||
|
||||
@@ -118,7 +118,8 @@ public:
|
||||
jit_convert_array(convert_vec_t convert_vec,
|
||||
size_t src_size,
|
||||
size_t dst_size)
|
||||
: _convert_vec(convert_vec)
|
||||
: jit_kernel(jit_name())
|
||||
, _convert_vec(convert_vec)
|
||||
, _src_size(src_size)
|
||||
, _dst_size(dst_size) {}
|
||||
|
||||
|
||||
@@ -30,7 +30,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_permute_kernel_f32 : public jit_uni_permute_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_permute_kernel_f32)
|
||||
|
||||
explicit jit_uni_permute_kernel_f32(jit_permute_config_params jcp_) : jit_uni_permute_kernel(jcp_), jit_generator() {}
|
||||
explicit jit_uni_permute_kernel_f32(jit_permute_config_params jcp_) : jit_uni_permute_kernel(jcp_), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -55,7 +55,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_softmax_kernel_f32 : public jit_uni_softmax_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_softmax_kernel_f32)
|
||||
|
||||
jit_uni_softmax_kernel_f32(jit_softmax_config_params jcp) : jcp_(jcp), jit_uni_softmax_kernel(), jit_generator() {}
|
||||
jit_uni_softmax_kernel_f32(jit_softmax_config_params jcp) : jcp_(jcp), jit_uni_softmax_kernel(), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -32,7 +32,7 @@ struct jit_uni_def_conv_kernel_f32 : public jit_uni_def_conv_kernel, public jit_
|
||||
|
||||
constexpr static int sampledPointsPerPixel = DeformableConvolution::sampledPointsPerPixel;
|
||||
|
||||
explicit jit_uni_def_conv_kernel_f32(const jit_def_conv_params& jcp) : jit_uni_def_conv_kernel(jcp), jit_generator() {}
|
||||
explicit jit_uni_def_conv_kernel_f32(const jit_def_conv_params& jcp) : jit_uni_def_conv_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -104,7 +104,7 @@ struct jit_uni_eltwise_generic : public jit_uni_eltwise_kernel, public jit_gener
|
||||
const std::vector<Eltwise::EltwiseData>& eltwise_data,
|
||||
const std::vector<ov::intel_cpu::Type>& ops_list,
|
||||
const dnnl::post_ops& post_ops)
|
||||
: jit_uni_eltwise_kernel(jep), jit_generator(), eltwise_data_(eltwise_data), ops_list_(ops_list), post_ops_(post_ops) {}
|
||||
: jit_uni_eltwise_kernel(jep), jit_generator(jit_name()), eltwise_data_(eltwise_data), ops_list_(ops_list), post_ops_(post_ops) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -32,7 +32,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_extract_image_patches_kernel : public jit_uni_extract_image_patches_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_extract_image_patches_kernel)
|
||||
|
||||
explicit jit_extract_image_patches_kernel(jit_extract_image_patches_params jpp) : jit_uni_extract_image_patches_kernel(jpp), jit_generator() {}
|
||||
explicit jit_extract_image_patches_kernel(jit_extract_image_patches_params jpp) : jit_uni_extract_image_patches_kernel(jpp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -49,7 +49,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_binarization_kernel : public jit_uni_quantize_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_binarization_kernel)
|
||||
|
||||
explicit jit_uni_binarization_kernel(const jit_quantize_params& jqp) : jit_uni_quantize_kernel(jqp), jit_generator() {}
|
||||
explicit jit_uni_binarization_kernel(const jit_quantize_params& jqp) : jit_uni_quantize_kernel(jqp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
@@ -217,7 +217,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_quantization_kernel : public jit_uni_quantize_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_quantization_kernel)
|
||||
|
||||
explicit jit_uni_quantization_kernel(const jit_quantize_params& jqp) : jit_uni_quantize_kernel(jqp), jit_generator() {}
|
||||
explicit jit_uni_quantization_kernel(const jit_quantize_params& jqp) : jit_uni_quantize_kernel(jqp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -45,7 +45,7 @@ struct jit_has_subnormals_base : public jit_generator {
|
||||
|
||||
typedef void (*fn_t)(const args_t*);
|
||||
|
||||
jit_has_subnormals_base() : jit_generator() {
|
||||
jit_has_subnormals_base() : jit_generator(jit_name()) {
|
||||
jit_ker_ = nullptr;
|
||||
}
|
||||
|
||||
|
||||
@@ -50,7 +50,7 @@ struct jit_uni_interpolate_kernel_f32 : public jit_uni_interpolate_kernel, publi
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_interpolate_kernel_f32)
|
||||
|
||||
explicit jit_uni_interpolate_kernel_f32(jit_interpolate_config_params jcp, const dnnl_primitive_attr &attr)
|
||||
: jit_uni_interpolate_kernel(jcp, attr), jit_generator() {}
|
||||
: jit_uni_interpolate_kernel(jcp, attr), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -26,7 +26,7 @@ const unsigned jitGatherKernelBase::incVec[16] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9,
|
||||
|
||||
template <x64::cpu_isa_t isa>
|
||||
jitUniGatherKernel<isa>::jitUniGatherKernel(const jGatherConfParams& jcp) :
|
||||
jitGatherKernelBase(jcp), x64::jit_generator() {
|
||||
jitGatherKernelBase(jcp), x64::jit_generator(jit_name()) {
|
||||
vlen = x64::cpu_isa_traits<isa>::vlen;
|
||||
dataElPerVec = vlen / jcp.dataTypeSize;
|
||||
idxElPerVec = vlen / indicesTypeSize;
|
||||
|
||||
@@ -60,7 +60,7 @@ struct jit_dft_kernel_f32 : public jit_dft_kernel, public jit_generator {
|
||||
public:
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_dft_kernel_f32)
|
||||
|
||||
jit_dft_kernel_f32(bool is_inverse, enum dft_type type) : jit_dft_kernel(is_inverse, type), jit_generator() {}
|
||||
jit_dft_kernel_f32(bool is_inverse, enum dft_type type) : jit_dft_kernel(is_inverse, type), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -328,6 +328,7 @@ RegistersPool::Ptr RegistersPool::create(x64::cpu_isa_t isa, std::initializer_li
|
||||
ISA_SWITCH_CASE(x64::avx512_core)
|
||||
ISA_SWITCH_CASE(x64::avx512_core_vnni)
|
||||
ISA_SWITCH_CASE(x64::avx512_core_bf16)
|
||||
ISA_SWITCH_CASE(x64::avx512_core_fp16)
|
||||
case x64::avx_vnni: return std::make_shared<IsaRegistersPool<x64::avx>>(regsToExclude);
|
||||
case x64::avx512_core_bf16_ymm: return std::make_shared<IsaRegistersPool<x64::avx512_core>>(regsToExclude);
|
||||
case x64::avx512_core_bf16_amx_int8: return std::make_shared<IsaRegistersPool<x64::avx512_core>>(regsToExclude);
|
||||
|
||||
@@ -34,7 +34,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_mul_add_softmax_kernel : public jit_uni_mul_add_softmax_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_mul_add_softmax_kernel)
|
||||
|
||||
explicit jit_mul_add_softmax_kernel(const jit_mul_add_softmax_compile_params& jcp) : jit_uni_mul_add_softmax_kernel(jcp), jit_generator() {
|
||||
explicit jit_mul_add_softmax_kernel(const jit_mul_add_softmax_compile_params& jcp) : jit_uni_mul_add_softmax_kernel(jcp), jit_generator(jit_name()) {
|
||||
exp_emitter = std::make_shared<jit_dnnl_aux_emitter>(this, isa, dnnl_eltwise_exp, 0.f, 0.f);
|
||||
|
||||
vec_size = dnnl::impl::cpu::x64::cpu_isa_traits<isa>::vlen / sizeof(float);
|
||||
@@ -357,7 +357,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_convert_reorder_kernel : public jit_uni_convert_reorder_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_convert_reorder_kernel)
|
||||
|
||||
explicit jit_convert_reorder_kernel(const jit_convert_reorder_compile_params& jcp) : jit_uni_convert_reorder_kernel(jcp), jit_generator() {
|
||||
explicit jit_convert_reorder_kernel(const jit_convert_reorder_compile_params& jcp) : jit_uni_convert_reorder_kernel(jcp), jit_generator(jit_name()) {
|
||||
vec_size = dnnl::impl::cpu::x64::cpu_isa_traits<isa>::vlen / sizeof(float);
|
||||
}
|
||||
virtual ~jit_convert_reorder_kernel() {}
|
||||
@@ -503,7 +503,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_convert_transpose_kernel : public jit_uni_convert_transpose_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_convert_transpose_kernel)
|
||||
|
||||
explicit jit_convert_transpose_kernel(const jit_convert_transpose_compile_params& jcp) : jit_uni_convert_transpose_kernel(jcp), jit_generator() {
|
||||
explicit jit_convert_transpose_kernel(const jit_convert_transpose_compile_params& jcp) : jit_uni_convert_transpose_kernel(jcp), jit_generator(jit_name()) {
|
||||
interm_prc = jcp_.with_scales ? Precision(Precision::FP32) : jcp_.src_prc;
|
||||
vec_size = dnnl::impl::cpu::x64::cpu_isa_traits<isa>::vlen / interm_prc.size();
|
||||
}
|
||||
@@ -792,8 +792,9 @@ void MHA::init_brgemm(brgemmCtx& ctx, std::unique_ptr<brgemm_kernel_t>& brgKerne
|
||||
brgemm_t brgDesc;
|
||||
brgemm_strides_t strides {static_cast<dnnl_dim_t>(ctx.M * ctx.K), static_cast<dnnl_dim_t>(ctx.K * ctx.N)};
|
||||
|
||||
const bool is_int8 = utils::one_of(ctx.dt_in0, data_type::u8, data_type::s8) && utils::one_of(ctx.dt_in1, data_type::u8, data_type::s8);
|
||||
auto isa = use_amx ? isa_any
|
||||
: ctx.dt_in0 == dnnl_data_type_t::dnnl_bf16 ? avx512_core_bf16 : avx512_core_vnni;
|
||||
: ctx.dt_in0 == dnnl_data_type_t::dnnl_bf16 ? avx512_core_bf16 : (is_int8 ? avx512_core_vnni : avx512_core);
|
||||
auto status = brgemm_desc_init(&brgDesc, isa, brgemm_strd, ctx.dt_in0, ctx.dt_in1,
|
||||
false, false, brgemm_row_major, 1.f, ctx.beta, ctx.LDA, ctx.LDB, ctx.LDC, ctx.M, ctx.N, ctx.K, &strides);
|
||||
if (status != dnnl_success) {
|
||||
@@ -851,6 +852,9 @@ void MHA::init_brgemm_copy_b(std::unique_ptr<jit_brgemm_matmul_copy_b_t>& brgCop
|
||||
brgCopyKernelConf.K = K;
|
||||
brgCopyKernelConf.K_blk = K;
|
||||
brgCopyKernelConf.N_chunk_elems = brgCopyKernelConf.N_blk;
|
||||
brgCopyKernelConf.b_dt_sz = DnnlExtensionUtils::sizeOfDataType(static_cast<dnnl::memory::data_type>(brgCopyKernelConf.src_dt));
|
||||
brgCopyKernelConf.tr_b_dt_sz = DnnlExtensionUtils::sizeOfDataType(static_cast<dnnl::memory::data_type>(brgCopyKernelConf.src_dt));
|
||||
brgCopyKernelConf.req_wei_vnni_downconvert = false;
|
||||
|
||||
if (is_with_amx) {
|
||||
brgCopyKernelConf.isa = dt_in0 == dnnl_data_type_t::dnnl_bf16 ? avx512_core_bf16_amx_bf16 : avx512_core_bf16_amx_int8;
|
||||
|
||||
@@ -102,7 +102,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_mvn_mean_variance_kernel_f32 : public jit_uni_mvn_mean_variance_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_mvn_mean_kernel_f32)
|
||||
|
||||
explicit jit_uni_mvn_mean_variance_kernel_f32(jit_mvn_config_params jcp) : jit_uni_mvn_mean_variance_kernel(jcp), jit_generator() {}
|
||||
explicit jit_uni_mvn_mean_variance_kernel_f32(jit_mvn_config_params jcp) : jit_uni_mvn_mean_variance_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
@@ -413,7 +413,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_mvn_kernel_f32 : public jit_uni_mvn_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_mvn_kernel_f32)
|
||||
|
||||
explicit jit_uni_mvn_kernel_f32(jit_mvn_config_params jcp, const dnnl_primitive_attr &attr) : jit_uni_mvn_kernel(jcp, attr), jit_generator() {}
|
||||
explicit jit_uni_mvn_kernel_f32(jit_mvn_config_params jcp, const dnnl_primitive_attr &attr) : jit_uni_mvn_kernel(jcp, attr), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -36,7 +36,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_nms_kernel_f32 : public jit_uni_nms_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_nms_kernel_f32)
|
||||
|
||||
explicit jit_uni_nms_kernel_f32(jit_nms_config_params jcp_) : jit_uni_nms_kernel(jcp_), jit_generator() {}
|
||||
explicit jit_uni_nms_kernel_f32(jit_nms_config_params jcp_) : jit_uni_nms_kernel(jcp_), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -85,7 +85,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_normalize_modulo_kernel_f32 : public jit_uni_normalize_modulo_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_normalize_modulo_kernel_f32)
|
||||
|
||||
jit_uni_normalize_modulo_kernel_f32(jit_normalize_config_params jcp) : jit_uni_normalize_modulo_kernel(jcp), jit_generator() {}
|
||||
jit_uni_normalize_modulo_kernel_f32(jit_normalize_config_params jcp) : jit_uni_normalize_modulo_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
@@ -206,7 +206,7 @@ struct jit_uni_normalize_kernel_f32 : public jit_uni_normalize_kernel, public ji
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_normalize_kernel_f32)
|
||||
|
||||
explicit jit_uni_normalize_kernel_f32(jit_normalize_config_params jcp, const dnnl_primitive_attr &attr)
|
||||
: jit_uni_normalize_kernel(jcp, attr), jit_generator() {}
|
||||
: jit_uni_normalize_kernel(jcp, attr), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -112,7 +112,7 @@ struct jit_uni_reduce_kernel_f32 : public jit_uni_reduce_kernel, public jit_gene
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_reduce_kernel_f32)
|
||||
|
||||
explicit jit_uni_reduce_kernel_f32(jit_reduce_config_params jcp)
|
||||
: jit_uni_reduce_kernel(jcp), jit_generator() {}
|
||||
: jit_uni_reduce_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
@@ -1079,7 +1079,7 @@ struct jit_uni_reduce_post_kernel_f32 : public jit_uni_reduce_post_kernel, publi
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_reduce_post_kernel_f32)
|
||||
|
||||
explicit jit_uni_reduce_post_kernel_f32(jit_reduce_config_params jcp, const dnnl_primitive_attr &attr)
|
||||
: jit_uni_reduce_post_kernel(jcp, attr), jit_generator() {}
|
||||
: jit_uni_reduce_post_kernel(jcp, attr), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -31,7 +31,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_logistic_kernel_f32 : public jit_uni_logistic_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_logistic_kernel_f32)
|
||||
|
||||
jit_uni_logistic_kernel_f32(jit_logistic_config_params jcp) : jcp_(jcp), jit_uni_logistic_kernel(), jit_generator() {}
|
||||
jit_uni_logistic_kernel_f32(jit_logistic_config_params jcp) : jcp_(jcp), jit_uni_logistic_kernel(), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -38,7 +38,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_roi_align_kernel_f32 : public jit_uni_roi_align_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_roi_align_kernel_f32);
|
||||
|
||||
explicit jit_uni_roi_align_kernel_f32(jit_roi_align_params jcp) : jit_uni_roi_align_kernel(jcp), jit_generator() {}
|
||||
explicit jit_uni_roi_align_kernel_f32(jit_roi_align_params jcp) : jit_uni_roi_align_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -40,7 +40,7 @@ template <cpu_isa_t isa>
|
||||
struct jit_uni_roi_pooling_kernel_f32 : public jit_uni_roi_pooling_kernel, public jit_generator {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_roi_pooling_kernel_f32);
|
||||
|
||||
explicit jit_uni_roi_pooling_kernel_f32(jit_roi_pooling_params jcp) : jit_uni_roi_pooling_kernel(jcp), jit_generator() {}
|
||||
explicit jit_uni_roi_pooling_kernel_f32(jit_roi_pooling_params jcp) : jit_uni_roi_pooling_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -74,7 +74,7 @@ struct jit_uni_topk_kernel_f32 : public jit_uni_topk_kernel, public jit_generato
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_uni_topk_kernel_f32)
|
||||
|
||||
explicit jit_uni_topk_kernel_f32(jit_topk_config_params jcp)
|
||||
: jit_uni_topk_kernel(jcp), jit_generator() {}
|
||||
: jit_uni_topk_kernel(jcp), jit_generator(jit_name()) {}
|
||||
|
||||
void create_ker() override {
|
||||
jit_generator::create_kernel();
|
||||
|
||||
@@ -226,8 +226,8 @@ const void * consts_table::store(const void *data, size_t size) {
|
||||
|
||||
} // namespace internal
|
||||
|
||||
jit_kernel::jit_kernel()
|
||||
: jit_generator() {
|
||||
jit_kernel::jit_kernel(const char* name)
|
||||
: jit_generator(name) {
|
||||
_free_rmmregs.reserve(16);
|
||||
_free_rmmregs.reserve(16);
|
||||
|
||||
|
||||
@@ -634,7 +634,7 @@ struct jit_kernel : public dnnl::impl::cpu::x64::jit_generator {
|
||||
return { *this, internal::make_shared(res, *this) };
|
||||
}
|
||||
|
||||
jit_kernel();
|
||||
jit_kernel(const char *name);
|
||||
|
||||
template<typename RegType>
|
||||
const RegType & reserve();
|
||||
|
||||
@@ -215,7 +215,7 @@ void CPUTestsBase::CheckPluginRelatedResultsImpl(const std::shared_ptr<const ov:
|
||||
|
||||
auto primType = getExecValue(ExecGraphInfoSerialization::IMPL_TYPE);
|
||||
|
||||
ASSERT_TRUE(primTypeCheck(primType)) << "primType is unexpected: " << primType;
|
||||
ASSERT_TRUE(primTypeCheck(primType)) << "primType is unexpected: " << primType << " Expected: " << selectedType;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -18,6 +18,9 @@ template<typename Params>
|
||||
struct jit_test_kernel : public jit_kernel {
|
||||
DECLARE_CPU_JIT_AUX_FUNCTIONS(jit_test_kernel)
|
||||
|
||||
jit_test_kernel()
|
||||
: jit_kernel(jit_name()) {}
|
||||
|
||||
typedef void (*function_t)(const Params *);
|
||||
|
||||
void init() {
|
||||
|
||||
+1
-1
Submodule src/plugins/intel_cpu/thirdparty/onednn updated: 10c74ec753...8545afe3e1.
Vendored
+1
-1
Submodule thirdparty/xbyak updated: 757e4063f6...9357732aa2.
Reference in new issue
Block a user