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

!15686 fix graph output address set in the one time memory application scenarios 

From: @limingqi107
Reviewed-by: @wilfchen,@cristoval
Signed-off-by: @cristoval
This commit is contained in:
mindspore-ci-bot 2021-04-26 19:54:33 +08:00 committed by Gitee
parent 0ff1b57a72 179c677fef
commit 409584a138
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -481,7 +481,9 @@ void GPUSession::UpdateOutputTensors(const VectorRef *outputs,
// When the device address of graph cnode output is set in tensor, the graph output need be set new device // When the device address of graph cnode output is set in tensor, the graph output need be set new device
// address, to avoid that the device address context of tensor be rewritten in the next step or next loop. // address, to avoid that the device address context of tensor be rewritten in the next step or next loop.
if (node->isa<CNode>()) { // 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)) {
auto new_address = std::make_shared<device::gpu::GPUDeviceAddress>(nullptr, address->GetSize()); auto new_address = std::make_shared<device::gpu::GPUDeviceAddress>(nullptr, address->GetSize());
AnfAlgo::SetOutputAddr(new_address, output_index, node.get()); AnfAlgo::SetOutputAddr(new_address, output_index, node.get());
if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) { if (context::GraphKernelFlags::GetInstance().IsEnableGraphKernel()) {

1
mindspore/ccsrc/runtime/device/gpu/kernel_info_setter.h
View File

@ -25,7 +25,6 @@
#include "ir/anf.h" #include "ir/anf.h"
#include "ir/dtype.h" #include "ir/dtype.h"
#include "utils/utils.h" #include "utils/utils.h"
#include "frontend/operator/ops.h"
#include "backend/kernel_compiler/kernel.h" #include "backend/kernel_compiler/kernel.h"
#include "backend/session/kernel_graph.h" #include "backend/session/kernel_graph.h"