p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

init plog when opentsd 

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2021-04-09 15:42:34 +08:00
parent a78924ed6f
commit f8248d61b9
2 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mindspore/ccsrc/utils/context/context_extends.cc
View File

@ -19,12 +19,17 @@
#include <string>
#include <memory>
#include <thread>
#include <atomic>
#include "pybind11/pybind11.h"
#include "utils/ms_utils.h"
#include "utils/convert_utils_base.h"
#ifndef NO_DLIB
#include "acl/acl_tdt.h"
#include "runtime/dev.h"
#include "toolchain/plog.h"
#endif
#ifdef ENABLE_GE
#include "transform/graph_ir/df_graph_manager.h"
#endif
namespace py = pybind11;
namespace mindspore {
@ -70,18 +75,21 @@ bool OpenTsd(const std::shared_ptr<MsContext> &ms_context_ptr) {
rank_size = IntToUint(rank_env);
}
int log_ret = DlogReportInitialize();
if (log_ret != 0) {
MS_LOG(WARNING) << "Init slog failed, ret = " << log_ret;
}
MS_LOG(INFO) << "Device id = " << device_id << ", rank size = " << rank_size << ".";
auto ret = rtSetDevice(device_id);
if (ret != RT_ERROR_NONE) {
MS_LOG(EXCEPTION) << "Device " << device_id << " call rtSetDevice failed, ret[" << static_cast<int>(ret) << "]";
return false;
}
ms_context_ptr->increase_param<uint32_t>(MS_CTX_TSD_REF);
#ifdef ENABLE_TDTQUE
acltdtChannelHandle *acl_handle = ms_context_ptr->CreateAclTdtChannelHandle();
if (acl_handle == nullptr) {
MS_LOG(EXCEPTION) << "Get acltdt handle failed";
return false;
}
ms_context_ptr->acl_tdt_print = std::thread(TensorPrint(acl_handle));
#endif
@ -121,6 +129,7 @@ bool CloseTsd(const std::shared_ptr<MsContext> &ms_context_ptr, bool force) {
}
ms_context_ptr->set_param<bool>(MS_CTX_IS_PYNATIVE_GE_INIT, false);
MS_LOG(INFO) << "Call rtDeviceReset, destroy and close tsd successful, ret[" << static_cast<int>(ret) << "]";
(void)DlogReportFinalize();
} else {
MS_LOG(DEBUG) << "Acltdt Dataset client is used, no need to close, tsd reference = "
<< ms_context_ptr->get_param<uint32_t>(MS_CTX_TSD_REF) << ".";

10
mindspore/ccsrc/utils/context/context_extends.h
View File

@ -23,16 +23,6 @@
#include "utils/ms_context.h"
#include "utils/tensorprint_utils.h"
#ifndef NO_DLIB
#include "acl/acl_tdt.h"
#include "tdt/tsd_client.h"
#include "tdt/data_common.h"
#include "runtime/dev.h"
#endif
#ifdef ENABLE_GE
#include "transform/graph_ir/df_graph_manager.h"
#endif
namespace mindspore {
namespace context {
bool OpenTsd(const std::shared_ptr<MsContext> &ms_context_ptr);