Browse Source

gpu fix conv bug

tags/v0.7.0-beta
VectorSL 5 years ago
parent
commit
7d7b81348e
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h
  2. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_filter_gpu_kernel.h View File

@@ -134,7 +134,7 @@ class ConvGradFilterGpuBkwKernel : public GpuKernel {
cudnnTensorDescriptor_t x_desc_real = nullptr; cudnnTensorDescriptor_t x_desc_real = nullptr;
int padA[2]; int padA[2];
int strideA[2] = {stride_[0], stride_[1]}; 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) { if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase || !symmetry_pad) {
pad_height_ = pad_list[0] + pad_list[1]; pad_height_ = pad_list[0] + pad_list[1];
pad_width_ = pad_list[2] + pad_list[3]; pad_width_ = pad_list[2] + pad_list[3];


+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv2d_grad_input_gpu_kernel.h View File

@@ -135,7 +135,7 @@ class ConvGradInputGpuBkwKernel : public GpuKernel {
cudnnTensorDescriptor_t dx_desc_real = nullptr; cudnnTensorDescriptor_t dx_desc_real = nullptr;
int padA[2]; int padA[2];
int strideA[2] = {stride_[0], stride_[1]}; 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) { if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase || !symmetry_pad) {
pad_height_ = pad_list[0] + pad_list[1]; pad_height_ = pad_list[0] + pad_list[1];
pad_width_ = pad_list[2] + pad_list[3]; pad_width_ = pad_list[2] + pad_list[3];


Loading…
Cancel
Save