Skip to content

Commit

Permalink
[VPU][TEST] Move some Interpolate tests to nightly, reduce number of …
Browse files Browse the repository at this point in the history
…tests (#5785)

* Move some tests to nightly, reduce number of tests

* small change - failures fix
  • Loading branch information
DariaMityagina authored May 25, 2021
1 parent 74293c5 commit 0772f7b
Showing 1 changed file with 54 additions and 43 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -19,47 +19,45 @@ const std::vector<InferenceEngine::Precision> netPrecisions = {
typedef std::map<std::string, std::string> Config;

const std::vector<std::vector<size_t>> inShapes = {
{3, 8, 38, 38},
{3, 8, 36, 36},
{3, 8, 35, 35},
{3, 8, 6, 6},

{3, 8, 3, 3},
{3, 8, 4, 4},
{3, 8, 16, 16},
{3, 8, 31, 31},
{3, 8, 26, 26},
{2, 8, 35, 35},
};

const std::vector<std::vector<size_t>> targetShapes = {
{3, 8, 38 * 2, 38 * 2},
{3, 8, 70, 70}, // * 1.94
{3, 8, 46, 46}, // * 1.3
{3, 8, 9, 9},

{3, 8, 6, 6},
{3, 8, 3, 3},
{3, 8, 36, 36},
{3, 8, 72, 72},
{3, 8, 30, 30},
{2, 8, 46, 46},
};

const std::vector<std::vector<size_t>> inShapes2x = {
{5, 19, 37, 37},
const std::vector<std::vector<size_t>> inShapesNoBatch = {
{1, 8, 6, 6},
};

const std::vector<std::vector<size_t>> targetShapes2x = {
{5, 19, 37 * 2, 37 * 2},
const std::vector<std::vector<size_t>> targetShapesNoBatch = {
{1, 8, 12, 12},
};

const std::vector<std::vector<size_t>> inShapesNoBatch = {
{1, 8, 38, 38},
{1, 8, 36, 36},
const std::vector<std::vector<size_t>> inShapesNightly = {
{2, 8, 38, 38},
{2, 8, 3, 3},
{2, 8, 4, 4},
{2, 8, 16, 16},
{2, 8, 31, 31},
{2, 8, 26, 26},
};

const std::vector<std::vector<size_t>> targetShapesNoBatch = {
{1, 8, 38 * 2, 38 * 2},
{1, 8, 70, 70}, // * 1.94
const std::vector<std::vector<size_t>> targetShapesNightly = {
{2, 8, 38 * 2, 38 * 2},
{2, 8, 6, 6},
{2, 8, 3, 3},
{2, 8, 36, 36},
{2, 8, 72, 72},
{2, 8, 30, 30},
};

const std::vector<std::vector<size_t>> inShapes2x = {
{2, 19, 37, 37},
};

const std::vector<std::vector<size_t>> targetShapes2x = {
{2, 19, 37 * 2, 37 * 2},
};

const std::vector<ngraph::op::v4::Interpolate::InterpolateMode> modesWithoutNearest = {
Expand Down Expand Up @@ -181,19 +179,6 @@ const auto interpolateCasesWithoutNearestMode = ::testing::Combine(
::testing::ValuesIn(defaultAxes),
::testing::ValuesIn(defaultScales));

INSTANTIATE_TEST_CASE_P(smoke_Interpolate_nearest_mode_2x, InterpolateLayerTest, ::testing::Combine(
interpolateCasesNearestMode2x,
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::ValuesIn(inShapes2x),
::testing::ValuesIn(targetShapes2x),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD),
::testing::Values(Config{{InferenceEngine::MYRIAD_DETECT_NETWORK_BATCH, CONFIG_VALUE(NO)}})),
InterpolateLayerTest::getTestCaseName);

INSTANTIATE_TEST_CASE_P(smoke_Interpolate_nearest_mode_no_batch, InterpolateLayerTest, ::testing::Combine(
interpolateCasesNearestMode,
::testing::ValuesIn(netPrecisions),
Expand Down Expand Up @@ -259,4 +244,30 @@ INSTANTIATE_TEST_CASE_P(smoke_Interpolate_without_nearest, InterpolateLayerTest,
::testing::Values(Config{{InferenceEngine::MYRIAD_DETECT_NETWORK_BATCH, CONFIG_VALUE(NO)}})),
InterpolateLayerTest::getTestCaseName);

INSTANTIATE_TEST_CASE_P(nightly_Interpolate_without_nearest, InterpolateLayerTest, ::testing::Combine(
interpolateCasesWithoutNearestMode,
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::ValuesIn(inShapesNightly),
::testing::ValuesIn(targetShapesNightly),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD),
::testing::Values(Config{{InferenceEngine::MYRIAD_DETECT_NETWORK_BATCH, CONFIG_VALUE(NO)}})),
InterpolateLayerTest::getTestCaseName);

INSTANTIATE_TEST_CASE_P(nightly_Interpolate_nearest_mode_2x, InterpolateLayerTest, ::testing::Combine(
interpolateCasesNearestMode2x,
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::ValuesIn(inShapes2x),
::testing::ValuesIn(targetShapes2x),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD),
::testing::Values(Config{{InferenceEngine::MYRIAD_DETECT_NETWORK_BATCH, CONFIG_VALUE(NO)}})),
InterpolateLayerTest::getTestCaseName);

} // namespace

0 comments on commit 0772f7b

Please sign in to comment.