fix funcitional test

mryzhov · Feb 21, 2023 · 4cb2b0a · 4cb2b0a
1 parent 981d086
commit 4cb2b0a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/src/plugins/intel_gna/src/gna_plugin.cpp b/src/plugins/intel_gna/src/gna_plugin.cpp
@@ -1290,20 +1290,11 @@ RequestStatus GNAPlugin::WaitFor(uint32_t request_idx, int64_t millisTimeout) {
                 const ngraph::Shape & model_input_shape = model->get_parameters()[0]->get_output_shape(0);
                 const auto model_input_type = InferenceEngine::details::convertPrecision(outputDesc.tensor_precision);
 
-                std::cout << "[EMUTEX DEBUG] model_input_type " << model_input_type << std::endl;
-
                 for (auto param : model->get_parameters()) {
                     param->set_element_type(model_input_type);
                 }
                 model->validate_nodes_and_infer_types();
 
-                {
-                    ngraph::pass::Manager manager;
-                    manager.register_pass<ngraph::pass::InitNodeInfo>();
-                    manager.register_pass<ngraph::pass::VisualizeTree>("./output_model.png"); // DEBUG
-                    manager.run_passes(model);
-                }
-
                 ov::Tensor input_tensor(model_input_type, model_input_shape, output_ptr);
 
                 if (model->evaluate(model_result, ov::TensorVector{input_tensor})) {

diff --git a/src/plugins/intel_gna/tests/functional/pass_tests/gather_remove.cpp b/src/plugins/intel_gna/tests/functional/pass_tests/gather_remove.cpp
@@ -145,14 +145,13 @@ class RemoveOutputGather : public testing::WithParamInterface<GatherRemoveConvsP
 
         auto input_params = ngraph::builder::makeParams(ngPrc, { input_shape });
 
-        auto tanh = std::make_shared<ngraph::opset9::Tanh>(input_params[0]);
-        auto tanh1 = std::make_shared<ngraph::opset9::Tanh>(tanh);
+        auto abs = std::make_shared<ngraph::opset9::Abs>(input_params[0]);
 
         const std::vector<size_t> indexes = MakeGatherIndexes(input_shape_product);
         auto gather_indexes_node = ngraph::opset9::Constant::create(ngraph::element::i64, ov::Shape{indexes.size()}, indexes);
         const size_t gather_axis = 1;
         auto gather_axis_node = ngraph::opset9::Constant::create(ngraph::element::i64, ngraph::Shape{}, {gather_axis});
-        auto gather_node = std::make_shared<ngraph::opset9::Gather>(tanh1,
+        auto gather_node = std::make_shared<ngraph::opset9::Gather>(abs,
                                                                 gather_indexes_node,
                                                                 gather_axis_node);