From 7d7b81348e3d6fe4a8efea038cb38ef8e3f8a2a4 Mon Sep 17 00:00:00 2001 From: VectorSL Date: Fri, 21 Aug 2020 17:28:25 +0800 Subject: [PATCH] gpu fix conv bug --- .../kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h | 2 +- .../kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h index 56836291ed..997ff68317 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h @@ -134,7 +134,7 @@ class ConvGradFilterGpuBkwKernel : public GpuKernel { cudnnTensorDescriptor_t x_desc_real = nullptr; int padA[2]; int strideA[2] = {stride_[0], stride_[1]}; - int dilaA[2] = {dilation_[0], dilation_[1]}; + int dilaA[2] = {dilation_[2], dilation_[3]}; if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase || !symmetry_pad) { pad_height_ = pad_list[0] + pad_list[1]; pad_width_ = pad_list[2] + pad_list[3]; diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h index cc123b912f..c6f67ab9a0 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h @@ -135,7 +135,7 @@ class ConvGradInputGpuBkwKernel : public GpuKernel { cudnnTensorDescriptor_t dx_desc_real = nullptr; int padA[2]; int strideA[2] = {stride_[0], stride_[1]}; - int dilaA[2] = {dilation_[0], dilation_[1]}; + int dilaA[2] = {dilation_[2], dilation_[3]}; if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase || !symmetry_pad) { pad_height_ = pad_list[0] + pad_list[1]; pad_width_ = pad_list[2] + pad_list[3];