From 49144fde376c1925fd179d79b744a374defe6342 Mon Sep 17 00:00:00 2001
From: shenghong96 <shenghong@huawei.com>
Date: Tue, 2 Feb 2021 16:52:20 +0800
Subject: [PATCH] fix UT of test_topk_no_split

---
 .../ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc  | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc b/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
index 61de3f6f17..619ae00dee 100644
--- a/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
+++ b/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
@@ -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