From 7b627dd8487c2d06d0b40a1764f781f49078d980 Mon Sep 17 00:00:00 2001
From: hong <43953930+phlrain@users.noreply.github.com>
Date: Tue, 12 Apr 2022 12:59:01 +0800
Subject: [PATCH] fix depthwise dnn bug (#41666)

---
 paddle/phi/kernels/gpudnn/conv_grad_kernel.cu | 1 +
 paddle/phi/kernels/gpudnn/conv_kernel.cu      | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddle/phi/kernels/gpudnn/conv_grad_kernel.cu b/paddle/phi/kernels/gpudnn/conv_grad_kernel.cu
index 3696ab08ea..9d4acb95ea 100644
--- a/paddle/phi/kernels/gpudnn/conv_grad_kernel.cu
+++ b/paddle/phi/kernels/gpudnn/conv_grad_kernel.cu
@@ -641,6 +641,7 @@ void DepthwiseConvCudnnGradKernel(const Context& dev_ctx,
                                   bool use_addto,
                                   int workspace_size_MB,
                                   bool exhaustive_search,
+                                  bool fuse_relu,
                                   DenseTensor* input_grad,
                                   DenseTensor* filter_grad) {
   ConvCudnnGradKernel<T>(dev_ctx,
diff --git a/paddle/phi/kernels/gpudnn/conv_kernel.cu b/paddle/phi/kernels/gpudnn/conv_kernel.cu
index d40cbecaee..3d3ab7b7a4 100644
--- a/paddle/phi/kernels/gpudnn/conv_kernel.cu
+++ b/paddle/phi/kernels/gpudnn/conv_kernel.cu
@@ -429,6 +429,7 @@ void DepthwiseConvCudnnKernel(const Context& dev_ctx,
                               bool use_addto,
                               int workspace_size_MB,
                               bool exhaustive_search,
+                              bool fuse_relu,
                               DenseTensor* out) {
   ConvCudnnKernel<T>(dev_ctx,
                      input,
-- 
GitLab