modify gpu backend logger

pull/12077/head
wilfChen 4 years ago
parent d638a92946
commit 58196f1faf

@ -75,7 +75,7 @@ class LstmGpuKernel : public GpuKernel {
if (!states_init_) { if (!states_init_) {
CHECK_CUDNN_RET_WITH_EXCEPT( CHECK_CUDNN_RET_WITH_EXCEPT(
kernel_node_, cudnnSetDropoutDescriptor(dropout_desc_, handle_, dropout_, states_addr, output_size_list_[4], 0), 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; states_init_ = true;
} }

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

Loading…
Cancel
Save