From 0ebfac471d04338fb029a37702f66424d0407f59 Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Sun, 28 Mar 2021 15:47:29 +0800
Subject: [PATCH] support unknown while subgraph

---
 tests/ut/ge/graph/load/davinci_model_unittest.cc | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/ut/ge/graph/load/davinci_model_unittest.cc b/tests/ut/ge/graph/load/davinci_model_unittest.cc
index f6f6e9ac..3487f8ed 100644
--- a/tests/ut/ge/graph/load/davinci_model_unittest.cc
+++ b/tests/ut/ge/graph/load/davinci_model_unittest.cc
@@ -49,7 +49,7 @@ int32_t MsprofReport(uint32_t moduleId, uint32_t type, void *data, uint32_t len)
   return 0;
 }
 
-/*TEST_F(UtestDavinciModel, init_success) {
+TEST_F(UtestDavinciModel, init_success) {
   DavinciModel model(0, nullptr);
   ComputeGraphPtr graph = make_shared<ComputeGraph>("default");
   ProfilingManager::Instance().is_load_profiling_ = true;
@@ -139,7 +139,7 @@ int32_t MsprofReport(uint32_t moduleId, uint32_t type, void *data, uint32_t len)
   EXPECT_EQ(outputs.size(), 1);
 
   ProfilingManager::Instance().is_load_profiling_ = false;
-}*/
+}
 
 TEST_F(UtestDavinciModel, init_data_op) {
   DavinciModel model(0, nullptr);
@@ -776,7 +776,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_input_dims_normal) {
 }
 
 // test label_set_task Init
-/*TEST_F(UtestDavinciModel, label_task_success) {
+TEST_F(UtestDavinciModel, label_task_success) {
   DavinciModel model(0, nullptr);
   ComputeGraphPtr graph = make_shared<ComputeGraph>("default");
 
@@ -860,7 +860,7 @@ TEST_F(UtestDavinciModel, init_data_aipp_input_dims_normal) {
   EXPECT_EQ(model.input_addrs_list_.size(), 0);
   EXPECT_EQ(model.output_addrs_list_.size(), 0);
   EXPECT_EQ(model.task_list_.size(), 5);
-}*/
+}
 
 TEST_F(UtestDavinciModel, LoadWithQueue_fail_with_diff_args) {
   DavinciModel model(0, nullptr);