Browse Source

sync change of tdt

tags/v0.6.0-beta
hanjun996 5 years ago
parent
commit
ed1a6dbc1c
2 changed files with 10 additions and 3 deletions
  1. +1
    -1
      mindspore/ccsrc/CMakeLists.txt
  2. +9
    -2
      mindspore/ccsrc/utils/context/ms_context.cc

+ 1
- 1
mindspore/ccsrc/CMakeLists.txt View File

@@ -131,7 +131,7 @@ if (ENABLE_GE)
else ()
target_link_libraries(mindspore ge_client)
endif ()
target_link_libraries(mindspore graph tsdclient)
target_link_libraries(mindspore graph tsdclient datatransfer)
endif()

if (ENABLE_D)


+ 9
- 2
mindspore/ccsrc/utils/context/ms_context.cc View File

@@ -177,14 +177,21 @@ bool MsContext::OpenTsd() {
}

MS_LOG(INFO) << "Device id = " << device_id << ", rank size = " << rank_size << ".";

#if (defined(ENABLE_TDTQUE) && defined(ENABLE_GE))
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
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 << ".";
return false;
}
tsd_ref_++;
#ifdef ENABLE_TDTQUE
#if (defined(ENABLE_TDTQUE) && !defined(ENABLE_GE))
int32_t initStatus = tdt::TdtHostInit(device_id);
if (initStatus != TDT_OK_CODE) {
MS_LOG(EXCEPTION) << "Init tsd failed, status = " << initStatus << ".";


Loading…
Cancel
Save