From d52270029b1220e043dcf7c5172cf31dbc056cfa Mon Sep 17 00:00:00 2001
From: hanjun996 <hanjun12@huawei.com>
Date: Wed, 29 Jul 2020 19:10:06 +0800
Subject: [PATCH] modify tdt

---
 mindspore/ccsrc/utils/context/ms_context.cc | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/mindspore/ccsrc/utils/context/ms_context.cc b/mindspore/ccsrc/utils/context/ms_context.cc
index 4a9be65869..720c77e23e 100644
--- a/mindspore/ccsrc/utils/context/ms_context.cc
+++ b/mindspore/ccsrc/utils/context/ms_context.cc
@@ -192,13 +192,6 @@ bool MsContext::OpenTsd() {
   }
 
   MS_LOG(INFO) << "Device id = " << device_id << ", rank size = " << rank_size << ".";
-#ifdef ENABLE_TDTQUE
-  int32_t initStatus = tdt::TdtHostInit(device_id);
-  if (initStatus != TDT_OK_CODE) {
-    MS_LOG(EXCEPTION) << "Init tsd failed, status = " << initStatus << ".";
-    return false;
-  }
-#endif
   TDT_StatusT status = tdt::TsdClient::GetInstance()->Open(device_id, rank_size);
   if (status != TDT_OK) {
     MS_LOG(EXCEPTION) << "Device " << device_id << " is occupied, open tsd failed, status = " << status << ".";
@@ -206,6 +199,11 @@ bool MsContext::OpenTsd() {
   }
   tsd_ref_++;
 #ifdef ENABLE_TDTQUE
+  int32_t initStatus = tdt::TdtHostInit(device_id);
+  if (initStatus != TDT_OK_CODE) {
+    MS_LOG(EXCEPTION) << "Init tsd failed, status = " << initStatus << ".";
+    return false;
+  }
   tdt_print_ = std::thread(TensorPrint());
 #endif
   MS_LOG(INFO) << "Open and init tsd successful, tsd reference = " << tsd_ref_ << ".";