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