This is an automated email from the ASF dual-hosted git repository.
jxie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 0fbdff8 Fix a minor bug in deformable_im2col.cuh (#12060)
0fbdff8 is described below
commit 0fbdff8bfbe8da6c1f1761abd4db30e7f8876c55
Author: 刘斌 <liubint...@gmail.com>
AuthorDate: Thu Aug 16 07:45:24 2018 +0800
Fix a minor bug in deformable_im2col.cuh (#12060)
Function `deformable_col2im_coord ` called
deformable_col2im_coord_gpu_kernel but check the deformable_col2im_gpu_kernel.
---
src/operator/contrib/nn/deformable_im2col.cuh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/contrib/nn/deformable_im2col.cuh
b/src/operator/contrib/nn/deformable_im2col.cuh
index d874276..5914184 100644
--- a/src/operator/contrib/nn/deformable_im2col.cuh
+++ b/src/operator/contrib/nn/deformable_im2col.cuh
@@ -510,7 +510,7 @@ inline void deformable_col2im_coord(mshadow::Stream<gpu>* s,
num_kernels, data_col, data_im, data_offset, im_shape[1], im_shape[2],
im_shape[3],
kernel_shape[0], kernel_shape[1], pad[0], pad[1], stride[0], stride[1],
dilation[0], dilation[1], channel_per_deformable_group, col_shape[1],
col_shape[2], grad_offset, req);
- MSHADOW_CUDA_POST_KERNEL_CHECK(deformable_col2im_gpu_kernel);
+ MSHADOW_CUDA_POST_KERNEL_CHECK(deformable_col2im_coord_gpu_kernel);
break;
default:
LOG(FATAL) << "col2im_nd_gpu does not support computation with "
