|
|
|
@ -131,6 +131,10 @@ void LoadKernelData(Debugger *debugger, const CNodePtr &kernel,
|
|
|
|
|
std::string input_kernel_name = input_kernel->fullname_with_scope();
|
|
|
|
|
auto addr = kernel_inputs[j];
|
|
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(input_kernel, PARAMETER_OUTPUT_INDEX);
|
|
|
|
|
// For example, this happens with the Depend op
|
|
|
|
|
if (type == kMetaTypeNone) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
auto format = kOpFormat_DEFAULT;
|
|
|
|
|
auto gpu_addr = std::make_unique<GPUDeviceAddress>(addr->addr, addr->size, format, type);
|
|
|
|
|
string input_tensor_name = input_kernel_name + ':' + "0";
|
|
|
|
@ -157,6 +161,10 @@ void LoadKernelData(Debugger *debugger, const CNodePtr &kernel,
|
|
|
|
|
for (int j : real_outputs) {
|
|
|
|
|
auto addr = kernel_outputs[j];
|
|
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(kernel, j);
|
|
|
|
|
// For example, this happens with the Depend op
|
|
|
|
|
if (type == kMetaTypeNone) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
auto format = kOpFormat_DEFAULT;
|
|
|
|
|
auto gpu_addr = std::make_unique<GPUDeviceAddress>(addr->addr, addr->size, format, type);
|
|
|
|
|
string tensor_name = kernel_name + ':' + std::to_string(j);
|
|
|
|
|