未验证 提交 83eef6d2 编写于 作者: J jiangcheng 提交者: GitHub

fix cinn_compile_test not pass problem (#37190)

上级 59fdf4da
...@@ -213,9 +213,10 @@ TEST(CinnCompilerTest, FlagController) { ...@@ -213,9 +213,10 @@ TEST(CinnCompilerTest, FlagController) {
ASSERT_EQ(compilation_keys.size(), 1); ASSERT_EQ(compilation_keys.size(), 1);
const auto& compiling_graph = cinn_compiler->FindGraph(compilation_keys[0]); const auto& compiling_graph = cinn_compiler->FindGraph(compilation_keys[0]);
auto op_types = ExtractOpTypes(compiling_graph); 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("feed"), 1);
ASSERT_EQ(op_types.count("mul"), 1); ASSERT_EQ(op_types.count("mul"), 1);
ASSERT_EQ(op_types.count("fetch"), 1);
} }
// recover flags // recover flags
FLAGS_allow_cinn_ops = ""; FLAGS_allow_cinn_ops = "";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册