p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5990 fix parameter in multi-graph 

Merge pull request !5990 from lianliguang/GAN
This commit is contained in:
mindspore-ci-bot 2020-10-13 09:12:14 +08:00 committed by Gitee
parent beb8bf5d65 8d8981a2fe
commit 39bc43e674
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/runtime/device/kernel_runtime.cc
View File

@ -331,7 +331,7 @@ void KernelRuntime::AssignStaticMemoryInput(const session::KernelGraph *graph) {
// if graph output is a weight and doesn't link to any cnode, it's data type will be unknown
if (output_type_id == kTypeUnknown) {
MS_LOG(WARNING) << "It is not suggested to use a lonely weight parameter as the output of graph";
output_type_id = AnfAlgo::GetOutputInferDataType(item, index);
continue;
}
auto tensor_size = CountNodeDeviceMemorySize(item, index);
auto address = CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(item, index), output_type_id);