m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!11987 fix UT of test_topk_no_split

Browse Source 
From: @shenghong96
Reviewed-by: @xsmq,@zhoufeng54
Signed-off-by: @xsmq
pull/11987/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent 01a0cdf5f0 49144fde37
commit bae1d3b7bb
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File

4
tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
Unescape View File

@ -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(), 2);
EXPECT_EQ(input_names_vec.size(), 1);
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

Write Preview
Loading…
Cancel
Save