|
|
|
@@ -242,6 +242,11 @@ void DumpKernelOutput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::T |
|
|
|
} |
|
|
|
output.set_original_output_format(GetGeFormat(output_format, output_shape.size())); |
|
|
|
output.set_address(static_cast<uint64_t>(reinterpret_cast<uintptr_t>(args)) + offset); |
|
|
|
// device address data size |
|
|
|
auto address = AnfAlgo::GetOutputAddr(kernel, i); |
|
|
|
MS_EXCEPTION_IF_NULL(address); |
|
|
|
output.set_size(address->GetSize()); |
|
|
|
MS_LOG(INFO) << "[DataDump] output " << i << " address size:" << output.size(); |
|
|
|
MS_EXCEPTION_IF_NULL(task->mutable_output()); |
|
|
|
task->mutable_output()->Add(std::move(output)); |
|
|
|
offset += sizeof(void *); |
|
|
|
@@ -272,6 +277,11 @@ void DumpKernelInput(const CNodePtr &kernel, void *args, NotNull<aicpu::dump::Ta |
|
|
|
input.mutable_shape()->add_dim(dim); |
|
|
|
} |
|
|
|
input.set_address(static_cast<uint64_t>(reinterpret_cast<uintptr_t>(args)) + offset); |
|
|
|
// device address data size |
|
|
|
auto address = AnfAlgo::GetPrevNodeOutputAddr(kernel, i); |
|
|
|
MS_EXCEPTION_IF_NULL(address); |
|
|
|
input.set_size(address->GetSize()); |
|
|
|
MS_LOG(INFO) << "[DataDump] input " << i << " address size:" << input.size(); |
|
|
|
MS_EXCEPTION_IF_NULL(task->mutable_input()); |
|
|
|
task->mutable_input()->Add(std::move(input)); |
|
|
|
offset += sizeof(void *); |
|
|
|
|