diff --git a/paddle/fluid/operators/conv_cudnn_helper.h b/paddle/fluid/operators/conv_cudnn_helper.h index f9e69de09f49632ca599f26125b473141da83ab7..3f7a87461c7b567bdacc14ce71c7a0a15d9ab057 100644 --- a/paddle/fluid/operators/conv_cudnn_helper.h +++ b/paddle/fluid/operators/conv_cudnn_helper.h @@ -467,6 +467,7 @@ struct SearchAlgorithm { algo_t algo; if (!exhaustive && !deterministic) { #if CUDNN_VERSION >= 7001 + VLOG(3) << "=====Not exhaustive====="; using perf_t = cudnnConvolutionBwdFilterAlgoPerf_t; int perf_count; int best_algo_idx = 0; @@ -493,6 +494,7 @@ struct SearchAlgorithm { } else if (deterministic) { return CUDNN_CONVOLUTION_BWD_FILTER_ALGO_1; } else { + VLOG(3) << "=======exhaustive=======: " << exhaustive; auto& dev_ctx = ctx.template device_context(); auto workspace_handle = dev_ctx.cudnn_workspace_handle(); diff --git a/paddle/fluid/operators/conv_cudnn_op.cu b/paddle/fluid/operators/conv_cudnn_op.cu index 7f705755915924de4ca6ab4c698e46a437bb649c..d8576ffb46da5dd3363b8e3d624e7efa9e9ce0b8 100644 --- a/paddle/fluid/operators/conv_cudnn_op.cu +++ b/paddle/fluid/operators/conv_cudnn_op.cu @@ -336,6 +336,11 @@ class CUDNNConvGradOpKernel : public framework::OpKernel { int groups = ctx.Attr("groups"); bool exhaustive_search = FLAGS_cudnn_exhaustive_search || ctx.Attr("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("exhaustive_search"); bool deterministic = FLAGS_cudnn_deterministic; if (exhaustive_search && deterministic) { PADDLE_THROW( diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 75d11648d20ea0a8c8d959fad38880f3fb62503f..bc9f182d95e3b728fbc0866e1c79f5508d3a04aa 100755 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -1582,7 +1582,7 @@ def conv2d(input, 'use_mkldnn': False, 'fuse_relu_before_depthwise_conv': False, "padding_algorithm": padding_algorithm, - "data_format": data_format + "data_format": data_format, }) if data_format == 'NCHW':