like20
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!18262 fixed ps data_parallel case result is error 

Merge pull request !18262 from anancds/error
This commit is contained in:
i-robot 2021-06-12 09:45:34 +08:00 committed by Gitee
parent b19330d25f 7d0d8f2a92
commit 7f731faa9b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/session/gpu_session.cc
View File

@ -538,7 +538,7 @@ void GPUSession::UpdateOutputTensors(const VectorRef *outputs,
// address, to avoid that the device address context of tensor be rewritten in the next step or next loop.
// But one time memory application scenarios need to be skipped, because the memory is not allocated next step:
// 1. Non cnode 2. Communication kernel.
if (node->isa<CNode>() && !AnfAlgo::IsCommunicationOp(node)) {
if (node->isa<CNode>() && !AnfAlgo::IsCommunicationOp(node) && !ps::PSContext::instance()->is_ps_mode()) {
auto new_address = std::make_shared<device::gpu::GPUDeviceAddress>(nullptr, address->GetSize());
AnfAlgo::SetOutputAddr(new_address, output_index, node.get());
if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) {