fix depthwise dnn bug (#41666)

7b627dd8 · hong · GitHub · 3749198e · 7b627dd8 · 7b627dd8
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/phi/kernels/gpudnn/conv_grad_kernel.cu paddle/phi/kernels/gpudnn/conv_grad_kernel.cu +1 -0

paddle/phi/kernels/gpudnn/conv_kernel.cu paddle/phi/kernels/gpudnn/conv_kernel.cu +1 -0

未找到文件。
--- 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,

--- 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,