diff --git a/paddle/fluid/operators/conv_fusion_op.cu b/paddle/fluid/operators/conv_fusion_op.cu index 17edccb4ea9eee27481fa3488b2094bacabd77ed..d84050fec8fae57738b6b6a607be125cf15addba 100644 --- a/paddle/fluid/operators/conv_fusion_op.cu +++ b/paddle/fluid/operators/conv_fusion_op.cu @@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ +#include #include "paddle/fluid/framework/op_registry.h" -#include "paddle/fluid/operators/conv_cudnn_helper.h" #include "paddle/fluid/operators/conv_cudnn_op_cache.h" #include "paddle/fluid/operators/conv_op.h" #include "paddle/fluid/operators/math/padding.h" @@ -208,7 +208,8 @@ class CUDNNConvFusionOpKernel : public framework::OpKernel { workspace_size_limit, &algo)); VLOG(3) << "cuDNN forward algo " << algo; } else { - auto search_func = [&]() { + std::function search_func = + [&]() -> cudnnConvolutionFwdAlgo_t { int returned_algo_count; std::array fwd_perf_stat;