Browse Source

!12077 optimize gpu backend logger

From: @wilfchen
Reviewed-by: @cristoval,@limingqi107
Signed-off-by: @limingqi107
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
50542793c8
2 changed files with 1 additions and 6 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/nn/lstm_gpu_kernel.h
  2. +0
    -5
      mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc

+ 1
- 1
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;
}



+ 0
- 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();
}
}



Loading…
Cancel
Save