From 19be5950c9e70f9cc9bd76b788c39f8a78df3c7b Mon Sep 17 00:00:00 2001 From: OuyangChao Date: Fri, 26 Mar 2021 14:51:57 +0000 Subject: [PATCH] fix bug of KernelDepthwiseConvNHWC --- paddle/fluid/operators/math/depthwise_conv.cu | 2 +- python/paddle/fluid/tests/unittests/test_conv2d_op.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/math/depthwise_conv.cu b/paddle/fluid/operators/math/depthwise_conv.cu index 7439a959d38285..fb3c2f3558d770 100644 --- a/paddle/fluid/operators/math/depthwise_conv.cu +++ b/paddle/fluid/operators/math/depthwise_conv.cu @@ -142,7 +142,7 @@ __device__ __inline__ void KernelDepthwiseConvNHWC( for (int w_in = w_in_start; w_in < w_in_end; w_in += dilate_width) { if (h_in >= h_start && h_in < h_end && w_in >= w_start && w_in < w_end) { int offset = ((batch * input_height + h_in) * input_width + w_in) * - output_channels + + input_channels + c_in; T in_data = input_data[offset]; if (fuse_relu_before_conv) { diff --git a/python/paddle/fluid/tests/unittests/test_conv2d_op.py b/python/paddle/fluid/tests/unittests/test_conv2d_op.py index 29c35d28d4d2e3..453d706f0d74e9 100644 --- a/python/paddle/fluid/tests/unittests/test_conv2d_op.py +++ b/python/paddle/fluid/tests/unittests/test_conv2d_op.py @@ -1235,6 +1235,7 @@ def init_paddings(self): create_test_channel_last_class(TestWithPad_AsyPadding) create_test_channel_last_class(TestWithGroup_AsyPadding) create_test_channel_last_class(TestWith1x1_AsyPadding) +create_test_channel_last_class(TestWithDepthWise7x7_AsyPadding) create_test_channel_last_class(TestWithInput1x1Filter1x1_AsyPadding) create_test_channel_last_class(TestDepthwiseConv_AsyPadding)