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

!12077 optimize gpu backend logger 

From: @wilfchen
Reviewed-by: @cristoval,@limingqi107
Signed-off-by: @limingqi107
This commit is contained in:
mindspore-ci-bot 2021-03-02 20:20:12 +08:00 committed by Gitee
parent 6dac92aedf 58196f1faf
commit 50542793c8
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h
View File

@ -75,7 +75,7 @@ class LstmGpuKernel : public GpuKernel {
if (!states_init_) {
CHECK_CUDNN_RET_WITH_EXCEPT(
kernel_node_, cudnnSetDropoutDescriptor(dropout_desc_, handle_, dropout_, states_addr, output_size_list_[4], 0),
"set dropout_desc failed");
"set dropout descriptor failed. Possible reasons: the GPU is out of memory.");
states_init_ = true;
}

5
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
View File

@ -333,8 +333,6 @@ void GPUKernelRuntime::AllocInplaceNodeMemory(const session::KernelGraph *graph)
auto prim = AnfAlgo::GetCNodePrimitive(node);
auto index = GetValue<uint32_t>(prim->GetAttr("inplace_output_index"));
AnfAlgo::SetOutputAddr(device_address, index, node.get());
MS_LOG(INFO) << "[inplace optimizer] group id: " << group.first << ", node: " << node->DebugString()
<< ", output_index: " << index;
}
}
}
@ -653,7 +651,6 @@ bool GPUKernelRuntime::LaunchKernelDynamic(const session::KernelGraph *graph, bo
auto kernel_mod = AnfAlgo::GetKernelMod(kernel);
MS_EXCEPTION_IF_NULL(kernel_mod);
if (AnfAlgo::IsInplaceNode(kernel, "skip")) {
MS_LOG(INFO) << "[inplace optimizer] skip node: " << kernel->DebugString();
continue;
}
@ -963,8 +960,6 @@ bool GPUKernelRuntime::AllocKernelInputDynamicRes(const mindspore::AnfNodePtr &k
if (i == input_index) {
auto skip_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(kernel), input_index);
device_address = GetPrevNodeMutableOutputAddr(skip_node, 0, false);
MS_LOG(INFO) << "[inplace optimizer] aggregate: " << kernel->DebugString()
<< ", skip: " << skip_node->DebugString() << ", address: " << device_address->GetMutablePtr();
}
}