|
|
@ -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) {
|
|
|
|