[CAPI][TEST] fix cancel api test with infer finished before cancel call (#22175)
* [CAPI][TEST] fix cancel api test with infer finished before cancel call Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com> * fix error Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com> --------- Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com>
This commit is contained in:
parent
1ef753a590
commit
7b782956ab
|
|
@ -298,8 +298,12 @@ TEST_P(ov_infer_request_test, infer) {
|
|||
|
||||
TEST_P(ov_infer_request_test, cancel) {
|
||||
OV_EXPECT_OK(ov_infer_request_set_tensor(infer_request, in_tensor_name, input_tensor));
|
||||
|
||||
OV_EXPECT_OK(ov_infer_request_cancel(infer_request));
|
||||
OV_ASSERT_OK(ov_infer_request_start_async(infer_request));
|
||||
ov_status_e return_status = ov_infer_request_cancel(infer_request);
|
||||
if (return_status == ov_status_e::OK || return_status == ov_status_e::INFER_CANCELLED)
|
||||
GTEST_SUCCEED();
|
||||
else
|
||||
GTEST_FAIL();
|
||||
}
|
||||
|
||||
TEST_P(ov_infer_request_ppp, infer_ppp) {
|
||||
|
|
|
|||
Loading…
Reference in New Issue