modify tdt
This commit is contained in:
parent
924a34acb8
commit
c718774538
|
@ -194,20 +194,20 @@ bool MsContext::OpenTsd() {
|
|||
}
|
||||
|
||||
MS_LOG(INFO) << "Device id = " << device_id << ", rank size = " << rank_size << ".";
|
||||
|
||||
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
|
||||
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());
|
||||
tsd_ref_++;
|
||||
#endif
|
||||
MS_LOG(INFO) << "Open and init tsd successful, tsd reference = " << tsd_ref_ << ".";
|
||||
return true;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue