From 8051c2d53546c3f37a1b337e4fe0eaf883f0afdc Mon Sep 17 00:00:00 2001
From: Maxim Vafin <maxim.vafin@intel.com>
Date: Wed, 1 Feb 2023 21:26:34 +0100
Subject: [PATCH] Remove second call for UnrollIf (#15441)

---
 src/frontends/pytorch/src/frontend.cpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/frontends/pytorch/src/frontend.cpp b/src/frontends/pytorch/src/frontend.cpp
index 0b56350fc51e5e..b4b53e6ce3ac44 100644
--- a/src/frontends/pytorch/src/frontend.cpp
+++ b/src/frontends/pytorch/src/frontend.cpp
@@ -9,6 +9,7 @@
 #include "openvino/pass/constant_folding.hpp"
 #include "openvino/util/log.hpp"
 #include "pt_framework_node.hpp"
+#include "transformations/common_optimizations/push_constant_to_subgraph.hpp"
 #include "transformations/control_flow/unroll_if.hpp"
 #include "transforms.hpp"
 #include "transforms/append_list_unpack_replacer.hpp"
@@ -81,8 +82,7 @@ void FrontEnd::normalize(const std::shared_ptr<ov::Model>& model) const {
     ov::pass::Manager manager;
 
     manager.register_pass<ov::pass::ConstantFolding>();
-    manager.register_pass<ov::pass::UnrollIf>();
-    // Have to run UnrollIf second time, because conditions are defined outside of nested If (ticket 98155)
+    manager.register_pass<ov::pass::PushConstantToSubgraph>();
     manager.register_pass<ov::pass::UnrollIf>();
     manager.register_pass<ov::frontend::pytorch::pass::AtenCatToConcat>();
     manager.register_pass<ov::frontend::pytorch::pass::AppendListUnpackReplacer>();
@@ -92,7 +92,6 @@ void FrontEnd::normalize(const std::shared_ptr<ov::Model>& model) const {
     manager.register_pass<ov::frontend::pytorch::pass::ListConstructReshapeReplacer>();
     manager.register_pass<ov::frontend::pytorch::pass::PrimListConstructPadReplacer>();
     manager.register_pass<ov::frontend::pytorch::pass::DecomposeTupleResults>();
-    manager.register_pass<ov::pass::ConstantFolding>();
 
     manager.run_passes(model);