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

fix timeout

This commit is contained in:
jiaorui 2022-02-24 16:19:45 +08:00
parent 14421b139b
commit 3a14957ec3
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/graph_scheduler/actor/data_prepare_actor.cc
View File

@ -685,6 +685,7 @@ void DataPrepareActor::PrepareDataForWeightNode(const AnfNodePtr &backend_node,
SET_OPCONTEXT_FAIL_RET_WITH_ERROR_BY_STRATEGY(real_strategy_, (*context), error_info);
}
host_tensor_address = device_tensor;
tensor->set_device_address(device_tensor);
} else {
AnfAlgo::SetOutputAddr(host_tensor_address, 0, backend_node.get());
host_tensor_address->SetNodeIndex(backend_node, 0);