fix gpu heterogeneous

5 years ago · 517fcc16ee
parent a7157c42eb
commit 517fcc16ee
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ccsrc/runtime/device/cpu/cpu_kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/cpu/cpu_kernel_runtime.cc
@ -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;