From 5b9cdd9bc5c37932035a0b12b14f45ca8ae0d7e0 Mon Sep 17 00:00:00 2001
From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com>
Date: Thu, 21 Apr 2022 10:09:57 +0800
Subject: [PATCH] fix adaptive pool pass bug (#42022)

---
 .../framework/ir/adaptive_pool2d_convert_global_pass.cc     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddle/fluid/framework/ir/adaptive_pool2d_convert_global_pass.cc b/paddle/fluid/framework/ir/adaptive_pool2d_convert_global_pass.cc
index 7846016d7e7..8eb8274168c 100644
--- a/paddle/fluid/framework/ir/adaptive_pool2d_convert_global_pass.cc
+++ b/paddle/fluid/framework/ir/adaptive_pool2d_convert_global_pass.cc
@@ -77,13 +77,13 @@ void AdaptivePool2dConvertGlobalPass::ApplyImpl(ir::Graph* graph) const {
         if (op->HasAttr("global_pooling")) {
           bool global_pooling =
               BOOST_GET_CONST(bool, op->GetAttr("global_pooling"));
-          if (global_pooling) return;
+          if (global_pooling) continue;
         }
-        if (!op->HasAttr("pooling_type")) return;
+        if (!op->HasAttr("pooling_type")) continue;
         std::string type =
             BOOST_GET_CONST(std::string, op->GetAttr("pooling_type"));
         // adaptive has no effect on max pooling
-        if (type == "max") return;
+        if (type == "max") continue;
         bool adaptive = BOOST_GET_CONST(bool, op->GetAttr("adaptive"));
         std::vector<int> ksize =
             BOOST_GET_CONST(std::vector<int>, op->GetAttr("ksize"));
-- 
GitLab