|
|
|
@ -57,9 +57,6 @@ constexpr const char *kOpTypeOpDebug = "Opdebug";
|
|
|
|
|
namespace mindspore {
|
|
|
|
|
namespace device {
|
|
|
|
|
namespace ascend {
|
|
|
|
|
static void DumpKernelOutput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task);
|
|
|
|
|
static void DumpKernelInput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task);
|
|
|
|
|
static void RtLoadDumpData(const aicpu::dump::OpMappingInfo &dump_info, void **ptr);
|
|
|
|
|
|
|
|
|
|
DataDumper::~DataDumper() {
|
|
|
|
|
ReleaseDevMem(&dev_load_mem_);
|
|
|
|
@ -328,7 +325,7 @@ void DataDumper::OpDebugUnregister() {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void RtLoadDumpData(const aicpu::dump::OpMappingInfo &dump_info, void **ptr) {
|
|
|
|
|
void DataDumper::RtLoadDumpData(const aicpu::dump::OpMappingInfo &dump_info, void **ptr) {
|
|
|
|
|
std::string proto_str;
|
|
|
|
|
size_t proto_size = dump_info.ByteSizeLong();
|
|
|
|
|
bool ret = dump_info.SerializeToString(&proto_str);
|
|
|
|
@ -357,7 +354,7 @@ void RtLoadDumpData(const aicpu::dump::OpMappingInfo &dump_info, void **ptr) {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void DumpKernelOutput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task) {
|
|
|
|
|
void DataDumper::DumpKernelOutput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task) {
|
|
|
|
|
if (!DumpJsonParser::GetInstance().OutputNeedDump()) {
|
|
|
|
|
MS_LOG(INFO) << "Skip dump output";
|
|
|
|
|
return;
|
|
|
|
@ -391,7 +388,7 @@ void DumpKernelOutput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::T
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void DumpKernelInput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task) {
|
|
|
|
|
void DataDumper::DumpKernelInput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Task *> task) {
|
|
|
|
|
if (!DumpJsonParser::GetInstance().InputNeedDump()) {
|
|
|
|
|
MS_LOG(INFO) << "Skip dump input";
|
|
|
|
|
return;
|
|
|
|
|