提交 62eab2dc 编写于 作者: Z zhangting2020

add logs

上级 e0bedebe
...@@ -467,6 +467,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdFilterAlgoPerf_t> { ...@@ -467,6 +467,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdFilterAlgoPerf_t> {
algo_t algo; algo_t algo;
if (!exhaustive && !deterministic) { if (!exhaustive && !deterministic) {
#if CUDNN_VERSION >= 7001 #if CUDNN_VERSION >= 7001
VLOG(3) << "=====Not exhaustive=====";
using perf_t = cudnnConvolutionBwdFilterAlgoPerf_t; using perf_t = cudnnConvolutionBwdFilterAlgoPerf_t;
int perf_count; int perf_count;
int best_algo_idx = 0; int best_algo_idx = 0;
...@@ -493,6 +494,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdFilterAlgoPerf_t> { ...@@ -493,6 +494,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdFilterAlgoPerf_t> {
} else if (deterministic) { } else if (deterministic) {
return CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1; return CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1;
} else { } else {
VLOG(3) << "=======exhaustive=======: " << exhaustive;
auto& dev_ctx = auto& dev_ctx =
ctx.template device_context<platform::CUDADeviceContext>(); ctx.template device_context<platform::CUDADeviceContext>();
auto workspace_handle = dev_ctx.cudnn_workspace_handle(); auto workspace_handle = dev_ctx.cudnn_workspace_handle();
......
...@@ -336,6 +336,11 @@ class CUDNNConvGradOpKernel : public framework::OpKernel<T> { ...@@ -336,6 +336,11 @@ class CUDNNConvGradOpKernel : public framework::OpKernel<T> {
int groups = ctx.Attr<int>("groups"); int groups = ctx.Attr<int>("groups");
bool exhaustive_search = bool exhaustive_search =
FLAGS_cudnn_exhaustive_search || ctx.Attr<bool>("exhaustive_search"); FLAGS_cudnn_exhaustive_search || ctx.Attr<bool>("exhaustive_search");
VLOG(3) << "=====exhaustive_search====: " << exhaustive_search;
VLOG(3) << "====FLAGS_cudnn_exhaustive_search====: "
<< FLAGS_cudnn_exhaustive_search;
VLOG(3) << "====Attr: exhaustive_search====: "
<< ctx.Attr<bool>("exhaustive_search");
bool deterministic = FLAGS_cudnn_deterministic; bool deterministic = FLAGS_cudnn_deterministic;
if (exhaustive_search && deterministic) { if (exhaustive_search && deterministic) {
PADDLE_THROW( PADDLE_THROW(
......
...@@ -1582,7 +1582,7 @@ def conv2d(input, ...@@ -1582,7 +1582,7 @@ def conv2d(input,
'use_mkldnn': False, 'use_mkldnn': False,
'fuse_relu_before_depthwise_conv': False, 'fuse_relu_before_depthwise_conv': False,
"padding_algorithm": padding_algorithm, "padding_algorithm": padding_algorithm,
"data_format": data_format "data_format": data_format,
}) })
if data_format == 'NCHW': if data_format == 'NCHW':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册