[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:
Xuejun Zhai 2024-01-16 21:17:13 +08:00 committed by GitHub
parent 1ef753a590
commit 7b782956ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions

View File

@ -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) {