IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BATCH as the constants device name in test

Browse Source
This commit is contained in:
myshevts 2021-11-22 15:54:34 +03:00
parent 6e0dfffb36
commit 84e858d636
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
inference-engine/tests/functional/plugin/shared/include/auto_batching/auto_batching_tests.hpp
View File

@ -65,14 +65,16 @@ TEST_P(AutoBatching_Test, compareAutoBatchingToBatch1) {
for (size_t i = 0; i < nets.size(); ++i) {
auto net = nets[i];
net.getInputsInfo().begin()->second->setLayout(Layout::NCHW);
net.getInputsInfo().begin()->second->setPrecision(Precision::FP32);
// we test single inputs networks only
auto inp = net.getInputsInfo().begin()->second;
inp->setLayout(Layout::NCHW);
inp->setPrecision(Precision::FP32);
std::map<std::string, std::string> config;
if (device_name.find("GPU") != std::string::npos)
config[CONFIG_KEY(GPU_THROUGHPUT_STREAMS)] = std::to_string(num_streams);
if (device_name.find("CPU") != std::string::npos)
config[CONFIG_KEY(CPU_THROUGHPUT_STREAMS)] = std::to_string(num_streams);
auto exec_net_ref = ie.LoadNetwork(net, std::string("BATCH:") +
auto exec_net_ref = ie.LoadNetwork(net, std::string(CommonTestUtils::DEVICE_BATCH) +
device_name + "(" + std::to_string(num_batch) + ")",
config);

1
inference-engine/tests/ie_test_utils/common_test_utils/test_constants.hpp
View File

@ -10,6 +10,7 @@ const char DEVICE_AUTO[] = "AUTO";
const char DEVICE_CPU[] = "CPU";
const char DEVICE_GNA[] = "GNA";
const char DEVICE_GPU[] = "GPU";
const char DEVICE_BATCH[] = "BATCH";
const char DEVICE_HDDL[] = "HDDL";
const char DEVICE_MYRIAD[] = "MYRIAD";
const char DEVICE_KEEMBAY[] = "VPUX";