From 91d15cdc13a7acf67151afa990b6bdefaec84221 Mon Sep 17 00:00:00 2001
From: zhangxiaokun <zhang.xiaokun@huawei.com>
Date: Wed, 7 Jul 2021 18:43:12 +0800
Subject: [PATCH] Fix UT

---
 .../graph/execute/model_executor_unittest.cc  | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/ut/ge/graph/execute/model_executor_unittest.cc b/tests/ut/ge/graph/execute/model_executor_unittest.cc
index 33643993..d4e0e3a4 100644
--- a/tests/ut/ge/graph/execute/model_executor_unittest.cc
+++ b/tests/ut/ge/graph/execute/model_executor_unittest.cc
@@ -63,7 +63,7 @@ static NodePtr CreateNode(ComputeGraph &graph, const string &name, const string
 
 TEST_F(UtestModelExecutorTest, test_load_graph_sync) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   auto compute_graph = MakeShared<ComputeGraph>("test_graph");
   GeRootModelPtr ge_root_model = MakeShared<GeRootModel>(compute_graph);
@@ -86,7 +86,7 @@ TEST_F(UtestModelExecutorTest, test_load_graph_sync) {
 
 TEST_F(UtestModelExecutorTest, test_load_graph_async) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   Graph graph("test_graph");
   auto compute_graph = MakeShared<ComputeGraph>("test_graph");
@@ -111,7 +111,7 @@ TEST_F(UtestModelExecutorTest, test_load_graph_async) {
 
 TEST_F(UtestModelExecutorTest, test_load_graph_failed) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   Graph graph("test_graph");
   auto compute_graph = MakeShared<ComputeGraph>("test_graph");
@@ -144,7 +144,7 @@ TEST_F(UtestModelExecutorTest, test_check_and_release_memory) {
   }
 
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   GeModelPtr ge_model = make_shared<GeModel>();
   int64_t memory_size = 25 * 1024UL * 1024UL * 1024UL;
@@ -171,7 +171,7 @@ TEST_F(UtestModelExecutorTest, test_check_and_release_memory) {
 
 TEST_F(UtestModelExecutorTest, parse_inputs_dims_data) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   OmeContext context;
   SetLocalOmeContext(context);
@@ -195,7 +195,7 @@ TEST_F(UtestModelExecutorTest, parse_inputs_dims_data) {
 
 TEST_F(UtestModelExecutorTest, parse_inputs_dims_getnext) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   OmeContext context;
   SetLocalOmeContext(context);
@@ -223,7 +223,7 @@ TEST_F(UtestModelExecutorTest, parse_inputs_dims_getnext) {
 
 TEST_F(UtestModelExecutorTest, test_run_thread) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   GraphId graph_id = 1;
   uint64_t session_id = 0;
@@ -281,7 +281,7 @@ static void test_run_graph(ModelExecutor &model_executor) {
 TEST_F(UtestModelExecutorTest, test_run_graph_train) {
   GetThreadLocalContext().SetGlobalOption({{OPTION_GRAPH_RUN_MODE, "1"}});
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
   test_run_graph(model_executor);
   EXPECT_EQ(model_executor.Finalize(), SUCCESS);
 }
@@ -291,14 +291,14 @@ TEST_F(UtestModelExecutorTest, test_run_graph_infer) {
   GetThreadLocalContext().SetSessionOption({});
   GetThreadLocalContext().SetGraphOption({});
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
   test_run_graph(model_executor);
   EXPECT_EQ(model_executor.Finalize(), SUCCESS);
 }
 
 TEST_F(UtestModelExecutorTest, test_run_graph_with_stream) {
   ModelExecutor model_executor;
-  EXPECT_EQ(model_executor.Initialize({}), SUCCESS);
+  EXPECT_EQ(model_executor.Initialize({}, 0), SUCCESS);
 
   GraphId graph_id = 1;
   auto compute_graph = MakeShared<ComputeGraph>("test_graph");