[IE][VPU]: TopK: Relax requirement for supporting Sort::None mode if K==1 (#2653)

* Update firmware
* Add TopK tests
* Disable some cases
This commit is contained in:
Andrew Bakalin 2020-10-14 13:35:17 +03:00 committed by GitHub
parent 0eb176c815
commit 6bfe41875e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 71 additions and 1 deletions

View File

@ -19,7 +19,7 @@ set(VPU_SUPPORTED_FIRMWARES usb-ma2450 usb-ma2x8x pcie-ma248x)
# Default packages # Default packages
# #
set(FIRMWARE_PACKAGE_VERSION 1397) set(FIRMWARE_PACKAGE_VERSION 1404)
set(VPU_CLC_MA2X8X_VERSION "movi-cltools-20.09.1") set(VPU_CLC_MA2X8X_VERSION "movi-cltools-20.09.1")
# #

View File

@ -0,0 +1,68 @@
// Copyright (C) 2020 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include "single_layer_tests/topk.hpp"
#include <vector>
using namespace LayerTestsDefinitions;
namespace {
const std::vector<InferenceEngine::Precision> netPrecisions = {
InferenceEngine::Precision::FP32,
InferenceEngine::Precision::FP16
};
const std::vector<int64_t> axes = {
0,
1,
2,
};
const std::vector<int64_t> k = {
1,
5,
10,
};
const std::vector<ngraph::opset5::TopK::Mode> modes = {
ngraph::opset5::TopK::Mode::MIN,
ngraph::opset5::TopK::Mode::MAX
};
const std::vector<ngraph::opset5::TopK::SortType> sortTypes = {
ngraph::opset5::TopK::SortType::SORT_INDICES,
ngraph::opset5::TopK::SortType::SORT_VALUES,
};
INSTANTIATE_TEST_CASE_P(smoke_TopK_IndicesValuesSort, TopKLayerTest,
::testing::Combine(
::testing::ValuesIn(k),
::testing::ValuesIn(axes),
::testing::ValuesIn(modes),
::testing::ValuesIn(sortTypes),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(std::vector<size_t>({10, 10, 10})),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD)),
TopKLayerTest::getTestCaseName);
INSTANTIATE_TEST_CASE_P(smoke_TopK_NoneSort, TopKLayerTest,
::testing::Combine(
::testing::Values(1),
::testing::ValuesIn(axes),
::testing::ValuesIn(modes),
::testing::Values(ngraph::opset5::TopK::SortType::NONE),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Layout::ANY),
::testing::Values(std::vector<size_t>({10, 10, 10})),
::testing::Values(CommonTestUtils::DEVICE_MYRIAD)),
TopKLayerTest::getTestCaseName);
} // namespace

View File

@ -23,5 +23,7 @@ std::vector<std::string> disabledTestPatterns() {
R"(.*(IEClassLoadNetwork).*(QueryNetworkMULTIWithHETERONoThrow_V10|QueryNetworkHETEROWithMULTINoThrow_V10).*)", R"(.*(IEClassLoadNetwork).*(QueryNetworkMULTIWithHETERONoThrow_V10|QueryNetworkHETEROWithMULTINoThrow_V10).*)",
// TODO: Issue: 34348 // TODO: Issue: 34348
R"(.*IEClassGetAvailableDevices.*)", R"(.*IEClassGetAvailableDevices.*)",
// TODO: Issue: 40473
R"(.*TopKLayerTest.*mode=min.*sort=index.*)",
}; };
} }