|
|
@@ -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 |