From a521894dc41253fd5a6bbbeebc053e1ecf93afff Mon Sep 17 00:00:00 2001 From: ahmedelmahy Date: Fri, 12 Mar 2021 11:26:32 +0200 Subject: [PATCH 1/2] added --- lib/models/dcn/src/deform_conv_cuda.cpp | 2 ++ lib/models/dcn/src/deform_pool_cuda.cpp | 2 ++ 2 files changed, 4 insertions(+) diff --git a/lib/models/dcn/src/deform_conv_cuda.cpp b/lib/models/dcn/src/deform_conv_cuda.cpp index 54dc8e4..1365e1e 100644 --- a/lib/models/dcn/src/deform_conv_cuda.cpp +++ b/lib/models/dcn/src/deform_conv_cuda.cpp @@ -7,6 +7,8 @@ #include #include +#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, 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 #include +#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, From cf5831249999f0b307d5aa948ebdcdef981ba68f Mon Sep 17 00:00:00 2001 From: ahmedelmahy Date: Sat, 13 Mar 2021 20:11:33 +0200 Subject: [PATCH 2/2] adding .contiguous --- lib/models/dcn/src/deform_conv_cuda.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/models/dcn/src/deform_conv_cuda.cpp b/lib/models/dcn/src/deform_conv_cuda.cpp index 1365e1e..9a94f6f 100644 --- a/lib/models/dcn/src/deform_conv_cuda.cpp +++ b/lib/models/dcn/src/deform_conv_cuda.cpp @@ -430,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,