Skip to content

Commit 69607ef

Browse files
committed
Remove usage of deprecated functions
1 parent 61d3193 commit 69607ef

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

modules/nvidia_plugin/tests/unit/plugin.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,13 +40,13 @@ TEST_F(PluginTest, CompileModel_Success) {
4040
}
4141

4242
TEST_F(PluginTest, CompileModel_NegativeId_Failed) {
43-
auto dummyFunction = std::make_shared<ov::Model>(ov::NodeVector{}, ov::ParameterVector{});
43+
auto dummyFunction = std::make_shared<ov::Model>(ov::OutputVector{}, ov::ParameterVector{});
4444
auto plugin = std::make_shared<Plugin>();
4545
ASSERT_THROW(plugin->compile_model(dummyFunction, {{ov::device::id.name(), "-1"}}), ov::Exception);
4646
}
4747

4848
TEST_F(PluginTest, CompileModel_OutRangeId_Failed) {
49-
auto dummyFunction = std::make_shared<ov::Model>(ov::NodeVector{}, ov::ParameterVector{});
49+
auto dummyFunction = std::make_shared<ov::Model>(ov::OutputVector{}, ov::ParameterVector{});
5050
auto plugin = std::make_shared<Plugin>();
5151
ASSERT_THROW(plugin->compile_model(dummyFunction, {{ov::device::id.name(), std::to_string(CUDA::Device::count())}}),
5252
ov::Exception);

modules/nvidia_plugin/tests/unit/transformations/cuda_transformations_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ TEST(TransformationTests, cuda_transformations_f16) {
2525
auto divide_constant = ov::op::v0::Constant::create(ov::element::f32, ov::Shape{1}, {2});
2626
auto divide = std::make_shared<ov::op::v1::Divide>(data, divide_constant);
2727

28-
model = std::make_shared<ov::Model>(ov::NodeVector{divide}, ov::ParameterVector{data});
28+
model = std::make_shared<ov::Model>(ov::OutputVector{divide}, ov::ParameterVector{data});
2929

3030
// Run transformation
3131
const CUDA::Device device{};
@@ -48,7 +48,7 @@ TEST(TransformationTests, cuda_transformations_f16) {
4848
auto mul = std::make_shared<ov::op::v1::Multiply>(convert_f16, mul_constant);
4949
auto convert_f32 = std::make_shared<ov::op::v0::Convert>(mul, ov::element::f32);
5050

51-
model_ref = std::make_shared<ov::Model>(ov::NodeVector{convert_f32}, ov::ParameterVector{data});
51+
model_ref = std::make_shared<ov::Model>(ov::OutputVector{convert_f32}, ov::ParameterVector{data});
5252
}
5353
auto res = compare_functions(model, model_ref);
5454
ASSERT_TRUE(res.first) << res.second;

0 commit comments

Comments
 (0)