diff --git a/modules/nvidia_plugin/tests/unit/transformations/bidirectional_lstm_sequence_composition.cpp b/modules/nvidia_plugin/tests/unit/transformations/bidirectional_lstm_sequence_composition.cpp index 549f57197..e83a74596 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/bidirectional_lstm_sequence_composition.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/bidirectional_lstm_sequence_composition.cpp @@ -10,8 +10,14 @@ #include "common_test_utils/ov_test_utils.hpp" #include "openvino/core/model.hpp" +#include "openvino/op/concat.hpp" #include "openvino/op/constant.hpp" +#include "openvino/op/parameter.hpp" #include "openvino/op/split.hpp" +#include "openvino/op/transpose.hpp" +#include "openvino/op/reshape.hpp" +#include "openvino/op/matmul.hpp" +#include "openvino/op/lstm_sequence.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" #include "transformations/utils/utils.hpp" diff --git a/modules/nvidia_plugin/tests/unit/transformations/concat_transformation.cpp b/modules/nvidia_plugin/tests/unit/transformations/concat_transformation.cpp index 6ab8ee4c9..409c8ee46 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/concat_transformation.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/concat_transformation.cpp @@ -10,7 +10,9 @@ #include "common_test_utils/ov_test_utils.hpp" #include "openvino/core/model.hpp" +#include "openvino/op/concat.hpp" #include "openvino/op/constant.hpp" +#include "openvino/op/parameter.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" #include "transformations/utils/utils.hpp" diff --git a/modules/nvidia_plugin/tests/unit/transformations/cuda_transformations_test.cpp b/modules/nvidia_plugin/tests/unit/transformations/cuda_transformations_test.cpp index d79476f21..d43915902 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/cuda_transformations_test.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/cuda_transformations_test.cpp @@ -8,7 +8,11 @@ #include #include "common_test_utils/ov_test_utils.hpp" -#include "openvino/opsets/opset10.hpp" +#include "openvino/op/constant.hpp" +#include "openvino/op/convert.hpp" +#include "openvino/op/divide.hpp" +#include "openvino/op/multiply.hpp" +#include "openvino/op/parameter.hpp" #include "transformer/cuda_graph_transformer.hpp" using namespace testing; @@ -17,9 +21,9 @@ TEST(TransformationTests, cuda_transformations_f16) { std::shared_ptr model, model_ref; { // Example model - auto data = std::make_shared(ov::element::f32, ov::Shape{3, 1, 2}); - auto divide_constant = ov::opset10::Constant::create(ov::element::f32, ov::Shape{1}, {2}); - auto divide = std::make_shared(data, divide_constant); + auto data = std::make_shared(ov::element::f32, ov::Shape{3, 1, 2}); + auto divide_constant = ov::op::v0::Constant::create(ov::element::f32, ov::Shape{1}, {2}); + auto divide = std::make_shared(data, divide_constant); model = std::make_shared(ov::NodeVector{divide}, ov::ParameterVector{data}); @@ -39,10 +43,10 @@ TEST(TransformationTests, cuda_transformations_f16) { { // Example reference model auto data = std::make_shared(ov::element::f32, ov::Shape{3, 1, 2}); - auto convert_f16 = std::make_shared(data, ov::element::f16); - auto mul_constant = ov::opset10::Constant::create(ov::element::f16, ov::Shape{1, 1, 1}, {0.5}); - auto mul = std::make_shared(convert_f16, mul_constant); - auto convert_f32 = std::make_shared(mul, ov::element::f32); + auto convert_f16 = std::make_shared(data, ov::element::f16); + auto mul_constant = ov::op::v0::Constant::create(ov::element::f16, ov::Shape{1, 1, 1}, {0.5}); + auto mul = std::make_shared(convert_f16, mul_constant); + auto convert_f32 = std::make_shared(mul, ov::element::f32); model_ref = std::make_shared(ov::NodeVector{convert_f32}, ov::ParameterVector{data}); } diff --git a/modules/nvidia_plugin/tests/unit/transformations/detection_output_fix_input_types.cpp b/modules/nvidia_plugin/tests/unit/transformations/detection_output_fix_input_types.cpp index 2074575ec..6b0405960 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/detection_output_fix_input_types.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/detection_output_fix_input_types.cpp @@ -10,6 +10,7 @@ #include "cuda_operation_registry.hpp" #include "openvino/core/model.hpp" #include "openvino/op/convert.hpp" +#include "openvino/op/parameter.hpp" #include "openvino/op/detection_output.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" diff --git a/modules/nvidia_plugin/tests/unit/transformations/fuse_matmul_add.cpp b/modules/nvidia_plugin/tests/unit/transformations/fuse_matmul_add.cpp index 22fc969d9..3c665ae3f 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/fuse_matmul_add.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/fuse_matmul_add.cpp @@ -10,7 +10,12 @@ #include "common_test_utils/ov_test_utils.hpp" #include "openvino/core/model.hpp" +#include "openvino/op/concat.hpp" #include "openvino/op/constant.hpp" +#include "openvino/op/parameter.hpp" +#include "openvino/op/add.hpp" +#include "openvino/op/matmul.hpp" +#include "openvino/op/variadic_split.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" #include "transformations/utils/utils.hpp" diff --git a/modules/nvidia_plugin/tests/unit/transformations/reduce_transformation.cpp b/modules/nvidia_plugin/tests/unit/transformations/reduce_transformation.cpp index fc6b22603..f95647bd8 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/reduce_transformation.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/reduce_transformation.cpp @@ -9,6 +9,13 @@ #include "common_test_utils/ov_test_utils.hpp" #include "openvino/core/model.hpp" #include "openvino/op/parameter.hpp" +#include "openvino/op/constant.hpp" +#include "openvino/op/reshape.hpp" +#include "openvino/op/reduce_max.hpp" +#include "openvino/op/reduce_mean.hpp" +#include "openvino/op/reduce_min.hpp" +#include "openvino/op/reduce_sum.hpp" +#include "openvino/op/reduce_prod.hpp" #include "openvino/opsets/opset.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" diff --git a/modules/nvidia_plugin/tests/unit/transformations/remove_redundant_convert_transformation.cpp b/modules/nvidia_plugin/tests/unit/transformations/remove_redundant_convert_transformation.cpp index c51347f62..df9832ad5 100644 --- a/modules/nvidia_plugin/tests/unit/transformations/remove_redundant_convert_transformation.cpp +++ b/modules/nvidia_plugin/tests/unit/transformations/remove_redundant_convert_transformation.cpp @@ -9,7 +9,8 @@ #include "common_test_utils/ov_test_utils.hpp" #include "openvino/core/model.hpp" #include "openvino/op/parameter.hpp" -#include "openvino/opsets/opset10.hpp" +#include "openvino/op/convert.hpp" +#include "openvino/op/abs.hpp" #include "openvino/pass/manager.hpp" #include "transformations/init_node_info.hpp" #include "transformations/utils/utils.hpp"