Browse Source

!4236 fix gpu heterogeneous bug

Merge pull request !4236 from baihuawei/heter
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b48c1f45f0
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/runtime/device/cpu/cpu_kernel_runtime.cc

+ 1
- 0
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); tensor->set_device_address(address);
need_sync_outputs->emplace_back(tensor); need_sync_outputs->emplace_back(tensor);
} else { } else {
tensor->set_device_address(nullptr);
address->ptr_ = tensor->data_c(); address->ptr_ = tensor->data_c();
} }
address->ref_count_ = INIT_NODE_REF; address->ref_count_ = INIT_NODE_REF;


Loading…
Cancel
Save