Browse Source

!6614 clean codex

Merge pull request !6614 from caifubi/dump2
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ebd01dab18
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/debug/data_dump/e2e_dump_util.cc

+ 3
- 0
mindspore/ccsrc/debug/data_dump/e2e_dump_util.cc 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) {
#ifdef ENABLE_DEBUGGER
auto format = kOpFormat_DEFAULT;
MS_EXCEPTION_IF_NULL(debugger);
DebugServices *debug_services = debugger->debug_services();
MS_EXCEPTION_IF_NULL(debug_services);
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,
addr->type_id(), addr->format(), slot);
if (!ret) {


Loading…
Cancel
Save