m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6614 clean codex

Browse Source 
Merge pull request !6614 from caifubi/dump2
pull/6614/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent feaf4e9aeb 32cd8d8135
commit ebd01dab18
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File

3
mindspore/ccsrc/debug/data_dump/e2e_dump_util.cc
Unescape View File

@ -64,8 +64,11 @@ void E2eDumpUtil::DumpGPUMemToFile(const std::string &file_path, const std::stri
const ShapeVector &int_shapes, const TypeId &type, size_t slot, Debugger *debugger) { const ShapeVector &int_shapes, const TypeId &type, size_t slot, Debugger *debugger) {
#ifdef ENABLE_DEBUGGER #ifdef ENABLE_DEBUGGER
auto format = kOpFormat_DEFAULT; auto format = kOpFormat_DEFAULT;
MS_EXCEPTION_IF_NULL(debugger);
DebugServices *debug_services = debugger->debug_services(); DebugServices *debug_services = debugger->debug_services();
MS_EXCEPTION_IF_NULL(debug_services);
TensorLoader *tensor_loader = debug_services->tensor_loader(); TensorLoader *tensor_loader = debug_services->tensor_loader();
MS_EXCEPTION_IF_NULL(tensor_loader);
auto ret = tensor_loader->DumpTensorToFile(original_kernel_name, trans_flag, file_path, format, int_shapes, type, auto ret = tensor_loader->DumpTensorToFile(original_kernel_name, trans_flag, file_path, format, int_shapes, type,
addr->type_id(), addr->format(), slot); addr->type_id(), addr->format(), slot);
if (!ret) { if (!ret) {

Write Preview
Loading…
Cancel
Save