[IE TESTS] Add example for QueryNetwork in SLT (#7628)

* [IE TESTS] Add example for QueryNetwork in SLT

* Update mul_conv_fusion.cpp

* Skip

* skip[
This commit is contained in:
Irina Efode 2021-09-27 11:22:38 +03:00 committed by GitHub
parent 3bf34b1166
commit 818f385398
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 38 additions and 1 deletions

View File

@ -50,8 +50,10 @@ std::vector<std::string> disabledTestPatterns() {
R"(.*smoke_MemoryTest.*iteration_count=3.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*iteration_count=4.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*iteration_count=10.*IS=\(1.10\).*)",
R"(.*smoke_MemoryTest.*LOW_LATENCY.*iteration_count=10.*IS=\(1.2\).*)",
R"(.*smoke_MemoryTest.*LOW_LATENCY.*iteration_count=10.*IS=\(1.2\).*)",/**/
// CVS-58963: Not implemented yet
R"(.*Behavior.*InferRequest.*OutOfFirstOutIsInputForSecondNetwork.*)",
// TODO: Issue: 29577
R"(.*QueryNetwork.*)",
};
}

View File

@ -20,4 +20,16 @@ TEST_P(ActivationDynamicLayerTest, CompareWithRefs) {
Run();
}
TEST_P(ActivationLayerTest, QueryNetwork) {
QueryNetwork();
}
TEST_P(ActivationParamLayerTest, QueryNetwork) {
QueryNetwork();
}
TEST_P(ActivationDynamicLayerTest, QueryNetwork) {
QueryNetwork();
}
} // namespace LayerTestsDefinitions

View File

@ -12,4 +12,8 @@ TEST_P(SplitConvConcat, CompareWithRefImpl) {
Run();
};
TEST_P(SplitConvConcat, QueryNetwork) {
QueryNetwork();
}
} // namespace SubgraphTestsDefinitions

View File

@ -57,6 +57,8 @@ public:
virtual void Serialize();
virtual void QueryNetwork();
static void Compare(const std::vector<std::pair<ngraph::element::Type, std::vector<std::uint8_t>>> &expected,
const std::vector<InferenceEngine::Blob::Ptr> &actual,
float threshold);

View File

@ -90,6 +90,23 @@ void LayerTestsCommon::Serialize() {
CommonTestUtils::removeIRFiles(out_xml_path, out_bin_path);
}
void LayerTestsCommon::QueryNetwork() {
SKIP_IF_CURRENT_TEST_IS_DISABLED();
cnnNetwork = InferenceEngine::CNNNetwork(function);
auto queryNetworkResult = PluginCache::get().ie()->QueryNetwork(cnnNetwork, targetDevice);
std::set<std::string> expected;
for (auto&& node : function->get_ops()) {
expected.insert(node->get_friendly_name());
}
std::set<std::string> actual;
for (auto&& res : queryNetworkResult.supportedLayersMap) {
actual.insert(res.first);
}
ASSERT_EQ(expected, actual);
}
InferenceEngine::Blob::Ptr LayerTestsCommon::GenerateInput(const InferenceEngine::InputInfo &info) const {
return FuncTestUtils::createAndFillBlob(info.getTensorDesc());
}