From 3a14957ec3c0ae82e40c737cc8709be226eeeeef Mon Sep 17 00:00:00 2001 From: jiaorui Date: Thu, 24 Feb 2022 16:19:45 +0800 Subject: [PATCH] fix timeout --- .../ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.cc b/mindspore/ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.cc index 453ddea746a..a6cf21aebc2 100644 --- a/mindspore/ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.cc +++ b/mindspore/ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.cc @@ -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);