fix cinn_compile_test not pass problem (#37190)

PaddlePaddle · Nov 15, 2021 · 83eef6d · 83eef6d
1 parent 59fdf4d
commit 83eef6d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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 = "";