Skip to content

Commit

Permalink
[CPU][TESTS] Add FC test cases to coverfusing binary post ops
Browse files Browse the repository at this point in the history
  • Loading branch information
EgorDuplensky committed Mar 21, 2024
1 parent f81902d commit 35e7759
Showing 1 changed file with 16 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,14 @@ const auto testParams2D_Brgemm_smoke = ::testing::Combine(fullyConnectedParams2D

INSTANTIATE_TEST_SUITE_P(smoke_FC_2D_Brgemm, MatMulLayerCPUTest, testParams2D_Brgemm_smoke, MatMulLayerCPUTest::getTestCaseName);

const auto testParams2D_Brgemm_binary_po =
::testing::Combine(fullyConnectedParams2D_Brgemm_smoke,
::testing::Values(MatMulNodeType::FullyConnected),
::testing::Values(fusingReluScaleShift), // eltwise_relu (to prevent Multiply opt out), binary_mul, binary_add
::testing::ValuesIn(filterSpecificParams_Brgemm(true)));

INSTANTIATE_TEST_SUITE_P(smoke_FC_2D_Brgemm_binary_po, MatMulLayerCPUTest, testParams2D_Brgemm_binary_po, MatMulLayerCPUTest::getTestCaseName);

const std::vector<ShapeRelatedParams> IS_brgemm_smoke = {
{static_shapes_to_test_representation({{1, 2, 32, 120}, {120, 5}}), {false, false}},
{static_shapes_to_test_representation({{1, 2, 32, 120}, {120, 5}}), {true, false}},
Expand Down Expand Up @@ -733,6 +741,14 @@ const auto testParams3D_smoke = ::testing::Combine(fullyConnectedParams3D_smoke,

INSTANTIATE_TEST_SUITE_P(smoke_FC_3D, MatMulLayerCPUTest, testParams3D_smoke, MatMulLayerCPUTest::getTestCaseName);

const auto testParams3D_binary_po =
::testing::Combine(fullyConnectedParams3D_smoke,
::testing::Values(MatMulNodeType::FullyConnected),
::testing::Values(fusingReluScaleShift), // eltwise_relu (to prevent Multiply opt out), binary_mul, binary_add
::testing::ValuesIn(filterCPUInfo(filterSpecificParams())));

INSTANTIATE_TEST_SUITE_P(smoke_FC_3D_binary_po, MatMulLayerCPUTest, testParams3D_binary_po, MatMulLayerCPUTest::getTestCaseName);

std::vector<fusingSpecificParams> fusingParamsSet3D_nightly {
fusingFakeQuantizePerTensorRelu,
};
Expand Down

0 comments on commit 35e7759

Please sign in to comment.