diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index 904297749d..150fbf914a 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -475,7 +475,7 @@ void GPUSession::UpdateOutputTensors(const VectorRef *outputs, const auto &address = AnfAlgo::GetMutableOutputAddr(node, output_index); // The outputs may have the same tensor, so need skip when the tensor has been set to device address. if ((address == nullptr) || (address->GetPtr() == nullptr)) { - return; + continue; } tensor->set_device_address(address);