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

Avoid capi infer wait timeout (#23973) 

### Details:
 - Avoid capi infer wait timeout
 - *...*

### Tickets:
 - CVS-128963
 - CVS-136569

---------

Co-authored-by: vurusovs <vitaliy.urusovskij@intel.com>
This commit is contained in:
River Li 2024-04-16 14:51:07 +08:00 committed by GitHub
parent 778e26ed2d
commit d145b32d2f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/bindings/c/tests/ov_infer_request_test.cpp
View File

@ -1,7 +1,9 @@
// Copyright (C) 2018-2024 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#include <chrono>
#include <mutex>
#include <thread>
#include "ov_test.hpp"
@ -338,13 +340,10 @@ TEST_P(ov_infer_request_test, infer_async_wait_for) {
if (!HasFatalFailure()) {
ov_status_e ret = ov_status_e::OK;
EXPECT_NO_THROW(ret = ov_infer_request_wait_for(infer_request, 10));
size_t max_times = 10;
// Random timeout in some platform, increase wait() times if timeout occurr.
while (ret != ov_status_e::OK && max_times-- > 0) {
EXPECT_NO_THROW(ret = ov_infer_request_wait_for(infer_request, 10));
}
// Set enough timeout duration to avoid timeout due to CPU is busying sometimes
EXPECT_NO_THROW(ret = ov_infer_request_wait_for(infer_request, 3000));
EXPECT_EQ(ret, ov_status_e::OK);
OV_EXPECT_OK(ov_infer_request_get_output_tensor_by_index(infer_request, 0, &output_tensor));
EXPECT_NE(nullptr, output_tensor);
}
@ -353,6 +352,12 @@ TEST_P(ov_infer_request_test, infer_async_wait_for) {
TEST_P(ov_infer_request_test, infer_async_wait_for_return_busy) {
OV_EXPECT_OK(ov_infer_request_set_input_tensor_by_index(infer_request, 0, input_tensor));
ov_callback_t callback;
callback.callback_func = [](void* args) {
std::this_thread::sleep_for(std::chrono::seconds(3));
};
OV_EXPECT_OK(ov_infer_request_set_callback(infer_request, &callback));
OV_ASSERT_OK(ov_infer_request_start_async(infer_request));
if (!HasFatalFailure()) {