From 58196f1faf67f5bc575953433b55e9a72e1ac6d3 Mon Sep 17 00:00:00 2001 From: wilfChen Date: Mon, 1 Mar 2021 19:25:01 +0800 Subject: [PATCH] modify gpu backend logger --- .../ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h | 2 +- mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc | 5 ----- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h index 605b4c6f97..425802de08 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h @@ -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; } diff --git a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc index 207893834f..6bdf3b80b4 100644 --- a/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc +++ b/mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc @@ -309,8 +309,6 @@ void GPUKernelRuntime::AllocInplaceNodeMemory(const session::KernelGraph *graph) auto prim = AnfAlgo::GetCNodePrimitive(node); auto index = GetValue(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; } } } @@ -629,7 +627,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; } @@ -937,8 +934,6 @@ bool GPUKernelRuntime::AllocKernelInputDynamicRes(const mindspore::AnfNodePtr &k if (i == input_index) { auto skip_node = AnfAlgo::GetInputNode(utils::cast(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(); } }