natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add Mod for Snippets tokenization

This commit is contained in:
xuchen-intel 2024-05-24 09:40:05 +02:00
parent e43fd019c7
commit 3cccec91c0
5 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/plugins/intel_cpu/src/emitters/snippets/aarch64/cpu_generator.cpp
View File

@ -88,6 +88,7 @@ CPUTargetMachine::CPUTargetMachine(dnnl::impl::cpu::aarch64::cpu_isa_t host_isa)
// binary
jitters[op::v1::Add::get_type_info_static()] = CREATE_CPU_EMITTER(jit_add_emitter);
jitters[op::v1::Divide::get_type_info_static()] = CREATE_CPU_EMITTER(jit_divide_emitter);
jitters[op::v1::Mod::get_type_info_static()] = CREATE_CPU_EMITTER(jit_mod_emitter);
jitters[op::v1::Multiply::get_type_info_static()] = CREATE_CPU_EMITTER(jit_multiply_emitter);
// unary

1
src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp
View File

@ -952,6 +952,7 @@ void Transformations::MainSnippets(void) {
ov::is_type<ov::op::v0::Elu>(n) ||
ov::is_type<ov::op::v0::Exp>(n) ||
ov::is_type<ov::op::v4::HSwish>(n) ||
ov::is_type<ov::op::v1::Mod>(n) ||
ov::is_type<ov::op::v1::Multiply>(n) ||
ov::is_type<ov::op::v0::Relu>(n) ||
ov::is_type<ov::op::v0::Sigmoid>(n) ||

9
src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/classes/eltwise.cpp
View File

@ -314,6 +314,15 @@ const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinInp() {
return eltwiseOpTypesBinInp;
}
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinInpSnippets() {
static const std::vector<utils::EltwiseTypes> eltwiseOpTypesBinInp = {
utils::EltwiseTypes::ADD,
utils::EltwiseTypes::MULTIPLY,
utils::EltwiseTypes::MOD,
};
return eltwiseOpTypesBinInp;
}
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesDiffInp() {
static const std::vector<utils::EltwiseTypes> eltwiseOpTypesDiffInp = { // Different number of input nodes depending on optimizations
utils::EltwiseTypes::POWER,

1
src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/classes/eltwise.hpp
View File

@ -51,6 +51,7 @@ const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinInp();
const std::vector<utils::InputLayerType>& secondaryInputTypes();
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinInp();
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinInpSnippets();
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesDiffInp();
const std::vector<utils::EltwiseTypes>& eltwiseOpTypesBinDyn();

17
src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/instances/common/eltwise.cpp
View File

@ -29,6 +29,23 @@ const auto params_4D = ::testing::Combine(
INSTANTIATE_TEST_SUITE_P(smoke_CompareWithRefs_4D_MemOrder, EltwiseLayerCPUTest, params_4D, EltwiseLayerCPUTest::getTestCaseName);
const auto params_4D_Snippets = ::testing::Combine(
::testing::Combine(
::testing::ValuesIn(static_shapes_to_test_representation(inShapes_4D())),
::testing::ValuesIn(eltwiseOpTypesBinInpSnippets()),
::testing::ValuesIn(secondaryInputTypes()),
::testing::ValuesIn(opTypes()),
::testing::ValuesIn(netType()),
::testing::Values(ov::element::undefined),
::testing::Values(ov::element::undefined),
::testing::Values(ov::test::utils::DEVICE_CPU),
::testing::Values(additional_config()[0])),
::testing::ValuesIn(filterCPUSpecificParams(cpuParams_4D())),
::testing::Values(emptyFusingSpec),
::testing::Values(true));
INSTANTIATE_TEST_SUITE_P(smoke_CompareWithRefs_4D_MemOrder_Snippets, EltwiseLayerCPUTest, params_4D_Snippets, EltwiseLayerCPUTest::getTestCaseName);
const auto params_4D_emptyCPUSpec = ::testing::Combine(
::testing::Combine(
::testing::ValuesIn(static_shapes_to_test_representation(inShapes_4D())),