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

fix gpu heterogeneous

This commit is contained in:
baihuawei 2020-08-14 14:12:22 +08:00
parent a7157c42eb
commit 517fcc16ee
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/runtime/device/cpu/cpu_kernel_runtime.cc
View File

@ -162,6 +162,7 @@ tensor::TensorPtr CPUKernelRuntime::CreatTensorForOutput(session::KernelGraph *k
tensor->set_device_address(address);
need_sync_outputs->emplace_back(tensor);
} else {
tensor->set_device_address(nullptr);
address->ptr_ = tensor->data_c();
}
address->ref_count_ = INIT_NODE_REF;