Skip to content

Commit

Permalink
[AUTO Plugin] add memory_states tests (openvinotoolkit#22455)
Browse files Browse the repository at this point in the history
* [AUTO Plugin] add memory_states tests

Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com>

* Fix warning as error

Signed-off-by: xuejun <xuejun.zhai@intel.com>

---------

Signed-off-by: Zhai, Xuejun <xuejun.zhai@intel.com>
Signed-off-by: xuejun <xuejun.zhai@intel.com>
  • Loading branch information
zhaixuejun1993 authored Jan 29, 2024
1 parent ec49f61 commit 1a55740
Showing 1 changed file with 24 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
// Copyright (C) 2018-2023 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#include "behavior/ov_infer_request/memory_states.hpp"

using namespace ov::test::behavior;

namespace {
std::vector<memoryStateParams> memoryStateTestCases = {
memoryStateParams(OVInferRequestVariableStateTest::get_network(),
{"c_1-3", "r_1-3"},
ov::test::utils::DEVICE_AUTO,
{{ov::device::priorities.name(), ov::test::utils::DEVICE_TEMPLATE}}),
memoryStateParams(OVInferRequestVariableStateTest::get_network(),
{"c_1-3", "r_1-3"},
ov::test::utils::DEVICE_MULTI,
{{ov::device::priorities.name(), ov::test::utils::DEVICE_TEMPLATE}})};

INSTANTIATE_TEST_SUITE_P(smoke_Auto_BehaviorTests,
OVInferRequestVariableStateTest,
::testing::ValuesIn(memoryStateTestCases),
OVInferRequestVariableStateTest::getTestCaseName);
} // namespace

0 comments on commit 1a55740

Please sign in to comment.