diff --git a/lib/models/dcn/src/deform_conv_cuda.cpp b/lib/models/dcn/src/deform_conv_cuda.cpp
index 54dc8e4..9a94f6f 100644
--- a/lib/models/dcn/src/deform_conv_cuda.cpp
+++ b/lib/models/dcn/src/deform_conv_cuda.cpp
@@ -7,6 +7,8 @@
 #include <cmath>
 #include <vector>
 
+#define AT_CHECK TORCH_CHECK
+
 void deformable_im2col(const at::Tensor data_im, const at::Tensor data_offset,
                        const int channels, const int height, const int width,
                        const int ksize_h, const int ksize_w, const int pad_h,
@@ -428,6 +430,7 @@ int deform_conv_backward_parameters_cuda(
   gradOutputBuffer =
       gradOutputBuffer.view({batchSize / im2col_step, nOutputPlane, im2col_step,
                              outputHeight, outputWidth});
+  gradOutputBuffer = gradOutputBuffer.contiguous();
   gradOutputBuffer.copy_(gradOutput);
   gradOutputBuffer =
       gradOutputBuffer.view({batchSize / im2col_step, nOutputPlane,
diff --git a/lib/models/dcn/src/deform_pool_cuda.cpp b/lib/models/dcn/src/deform_pool_cuda.cpp
index 9e0e3ff..6c39d04 100644
--- a/lib/models/dcn/src/deform_pool_cuda.cpp
+++ b/lib/models/dcn/src/deform_pool_cuda.cpp
@@ -11,6 +11,8 @@
 #include <cmath>
 #include <vector>
 
+#define AT_CHECK TORCH_CHECK
+
 void DeformablePSROIPoolForward(
     const at::Tensor data, const at::Tensor bbox, const at::Tensor trans,
     at::Tensor out, at::Tensor top_count, const int batch, const int channels,