fix gpu heterogeneous

pull/4236/head
baihuawei 5 years ago
parent a7157c42eb
commit 517fcc16ee

@ -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