This commit is contained in:
Egor Duplenskii 2024-06-12 03:27:45 +02:00 committed by GitHub
commit 9336e5e2dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 0 deletions

View File

@ -219,6 +219,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}},
@ -744,6 +752,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,
};