fix UT of test_topk_no_split

pull/11987/head
shenghong96 4 years ago
parent 2f883fb4c2
commit 49144fde37

@ -113,8 +113,7 @@ TEST_F(TestHWTopKSplit, test_topk_no_split) {
ConstInputToAttr(topk_cnode, attr_index);
EXPECT_EQ(topk_cnode->inputs().size(), 2);
input_names_vec = AnfAlgo::GetNodeAttr<std::vector<std::string>>(topk_cnode, kAttrInputNames);
/*
EXPECT_EQ(input_names_vec.size(), 1);
EXPECT_EQ(input_names_vec.size(), 2);
auto optimizer = std::make_shared<opt::GraphOptimizer>();
auto pm = std::make_shared<opt::PassManager>();
@ -125,7 +124,6 @@ TEST_F(TestHWTopKSplit, test_topk_no_split) {
optimizer->AddPassManager(pm);
FuncGraphPtr new_graph = optimizer->Optimize(kernel_graph);
EXPECT_EQ(topk_cnode, GetTopkCNodeFromKernelGraph(new_graph));
*/
}
} // namespace opt
} // namespace mindspore

Loading…
Cancel
Save