From 83eef6d2443dfbab5c1821e681764145c3f2d1cb Mon Sep 17 00:00:00 2001
From: jiangcheng <thisjiang@qq.com>
Date: Mon, 15 Nov 2021 14:18:16 +0800
Subject: [PATCH] fix cinn_compile_test not pass problem (#37190)

---
 paddle/fluid/framework/paddle2cinn/cinn_compiler_test.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/framework/paddle2cinn/cinn_compiler_test.cc b/paddle/fluid/framework/paddle2cinn/cinn_compiler_test.cc
index 145d3d83d4..923607fe5a 100644
--- a/paddle/fluid/framework/paddle2cinn/cinn_compiler_test.cc
+++ b/paddle/fluid/framework/paddle2cinn/cinn_compiler_test.cc
@@ -213,9 +213,10 @@ TEST(CinnCompilerTest, FlagController) {
     ASSERT_EQ(compilation_keys.size(), 1);
     const auto& compiling_graph = cinn_compiler->FindGraph(compilation_keys[0]);
     auto op_types = ExtractOpTypes(compiling_graph);
-    ASSERT_EQ(op_types.size(), 2);
+    ASSERT_EQ(op_types.size(), 3);
     ASSERT_EQ(op_types.count("feed"), 1);
     ASSERT_EQ(op_types.count("mul"), 1);
+    ASSERT_EQ(op_types.count("fetch"), 1);
   }
   // recover flags
   FLAGS_allow_cinn_ops = "";
-- 
GitLab