diff --git a/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_offsets.cpp b/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_offsets.cpp index 258384f07e0e38..6b2d3d6e538377 100644 --- a/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_offsets.cpp +++ b/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_offsets.cpp @@ -128,23 +128,35 @@ const std::vector input_shapes = { const std::vector> indices = {{0, 1, 2, 2, 3}, {4, 4, 3, 1, 0}, {1, 2, 1, 2, 1, 2, 1, 2, 1, 2}}; const std::vector> offsets = {{0, 2}, {0, 0, 2, 2}, {2, 4}}; const std::vector default_index = {0, 4}; -const std::vector with_weights = {false}; const std::vector with_default_index = {false, true}; const std::vector reduction = { ov::op::util::EmbeddingBagOffsetsBase::Reduction::SUM, ov::op::util::EmbeddingBagOffsetsBase::Reduction::MEAN}; -const auto embBagOffsetArgSet = ::testing::Combine(::testing::ValuesIn(input_shapes), +const auto embBagOffsetArgSetWthWeights = ::testing::Combine(::testing::ValuesIn(input_shapes), ::testing::ValuesIn(indices), ::testing::ValuesIn(offsets), ::testing::ValuesIn(default_index), - ::testing::ValuesIn(with_weights), + ::testing::Values(true), + ::testing::ValuesIn(with_default_index), + ::testing::Values(ov::op::util::EmbeddingBagOffsetsBase::Reduction::SUM)); +const auto embBagOffsetArgSetNoWeights = ::testing::Combine(::testing::ValuesIn(input_shapes), + ::testing::ValuesIn(indices), + ::testing::ValuesIn(offsets), + ::testing::ValuesIn(default_index), + ::testing::Values(false), ::testing::ValuesIn(with_default_index), ::testing::ValuesIn(reduction)); - -INSTANTIATE_TEST_SUITE_P(smoke, +INSTANTIATE_TEST_SUITE_P(smoke_EmbeddingBagOffsets_With_Weights, + EmbeddingBagOffsetsLayerCPUTest, + ::testing::Combine(embBagOffsetArgSetWthWeights, + ::testing::ValuesIn(netPrecisions), + ::testing::ValuesIn(indPrecisions), + ::testing::Values(ov::test::utils::DEVICE_CPU)), + EmbeddingBagOffsetsLayerCPUTest::getTestCaseName); +INSTANTIATE_TEST_SUITE_P(smoke_EmbeddingBagOffsets_No_Weights, EmbeddingBagOffsetsLayerCPUTest, - ::testing::Combine(embBagOffsetArgSet, + ::testing::Combine(embBagOffsetArgSetNoWeights, ::testing::ValuesIn(netPrecisions), ::testing::ValuesIn(indPrecisions), ::testing::Values(ov::test::utils::DEVICE_CPU)), diff --git a/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_packed.cpp b/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_packed.cpp index 3ae301da0cb685..8b6cd5e5a30291 100644 --- a/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_packed.cpp +++ b/src/plugins/intel_cpu/tests/functional/custom/single_layer_tests/embedding_bag_packed.cpp @@ -119,20 +119,31 @@ const std::vector input_shapes = { const std::vector>> indices = {{{0, 1}, {2, 2}, {3, 4}}, {{4, 4, 3}, {1, 0, 2}}, {{1, 2, 1, 2}, {1, 2, 1, 2}}}; -const std::vector with_weights = {false}; const std::vector reduction = { ov::op::util::EmbeddingBagPackedBase::Reduction::SUM, ov::op::util::EmbeddingBagPackedBase::Reduction::MEAN}; -const auto embBagPackedArgSet = ::testing::Combine(::testing::ValuesIn(input_shapes), +const auto embBagPackedArgSetWthWeights = ::testing::Combine(::testing::ValuesIn(input_shapes), ::testing::ValuesIn(indices), - ::testing::ValuesIn(with_weights), + ::testing::Values(true), + ::testing::Values(ov::op::util::EmbeddingBagPackedBase::Reduction::SUM)); + +const auto embBagPackedArgSetNoWeights = ::testing::Combine(::testing::ValuesIn(input_shapes), + ::testing::ValuesIn(indices), + ::testing::Values(false), ::testing::ValuesIn(reduction)); -INSTANTIATE_TEST_SUITE_P(smoke, +INSTANTIATE_TEST_SUITE_P(smoke_EmbeddingBagPacked_With_Weights, + EmbeddingBagPackedLayerCPUTest, + ::testing::Combine(embBagPackedArgSetWthWeights, + ::testing::ValuesIn(netPrecisions), + ::testing::ValuesIn(indPrecisions), + ::testing::Values(ov::test::utils::DEVICE_CPU)), + EmbeddingBagPackedLayerCPUTest::getTestCaseName); +INSTANTIATE_TEST_SUITE_P(smoke_EmbeddingBagPacked_No_Weights, EmbeddingBagPackedLayerCPUTest, - ::testing::Combine(embBagPackedArgSet, + ::testing::Combine(embBagPackedArgSetNoWeights, ::testing::ValuesIn(netPrecisions), ::testing::ValuesIn(indPrecisions), ::testing::Values(ov::test::utils::DEVICE_CPU)),