|
|
|
#pragma once |
|
#include <torch/types.h> |
|
|
|
namespace detectron2 { |
|
|
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
int deform_conv_forward_cuda( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor offset, |
|
at::Tensor output, |
|
at::Tensor columns, |
|
at::Tensor ones, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
int im2col_step); |
|
|
|
int deform_conv_backward_input_cuda( |
|
at::Tensor input, |
|
at::Tensor offset, |
|
at::Tensor gradOutput, |
|
at::Tensor gradInput, |
|
at::Tensor gradOffset, |
|
at::Tensor weight, |
|
at::Tensor columns, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
int im2col_step); |
|
|
|
int deform_conv_backward_parameters_cuda( |
|
at::Tensor input, |
|
at::Tensor offset, |
|
at::Tensor gradOutput, |
|
at::Tensor gradWeight, |
|
at::Tensor columns, |
|
at::Tensor ones, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
float scale, |
|
int im2col_step); |
|
|
|
void modulated_deform_conv_cuda_forward( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor bias, |
|
at::Tensor ones, |
|
at::Tensor offset, |
|
at::Tensor mask, |
|
at::Tensor output, |
|
at::Tensor columns, |
|
int kernel_h, |
|
int kernel_w, |
|
const int stride_h, |
|
const int stride_w, |
|
const int pad_h, |
|
const int pad_w, |
|
const int dilation_h, |
|
const int dilation_w, |
|
const int group, |
|
const int deformable_group, |
|
const bool with_bias); |
|
|
|
void modulated_deform_conv_cuda_backward( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor bias, |
|
at::Tensor ones, |
|
at::Tensor offset, |
|
at::Tensor mask, |
|
at::Tensor columns, |
|
at::Tensor grad_input, |
|
at::Tensor grad_weight, |
|
at::Tensor grad_bias, |
|
at::Tensor grad_offset, |
|
at::Tensor grad_mask, |
|
at::Tensor grad_output, |
|
int kernel_h, |
|
int kernel_w, |
|
int stride_h, |
|
int stride_w, |
|
int pad_h, |
|
int pad_w, |
|
int dilation_h, |
|
int dilation_w, |
|
int group, |
|
int deformable_group, |
|
const bool with_bias); |
|
|
|
#endif |
|
|
|
inline int deform_conv_forward( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor offset, |
|
at::Tensor output, |
|
at::Tensor columns, |
|
at::Tensor ones, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
int im2col_step) { |
|
if (input.is_cuda()) { |
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
TORCH_CHECK(weight.is_cuda(), "weight tensor is not on GPU!"); |
|
TORCH_CHECK(offset.is_cuda(), "offset tensor is not on GPU!"); |
|
return deform_conv_forward_cuda( |
|
input, |
|
weight, |
|
offset, |
|
output, |
|
columns, |
|
ones, |
|
kW, |
|
kH, |
|
dW, |
|
dH, |
|
padW, |
|
padH, |
|
dilationW, |
|
dilationH, |
|
group, |
|
deformable_group, |
|
im2col_step); |
|
#else |
|
AT_ERROR("Detectron2 is not compiled with GPU support!"); |
|
#endif |
|
} |
|
AT_ERROR("This operator is not implemented on CPU"); |
|
} |
|
|
|
inline int deform_conv_backward_input( |
|
at::Tensor input, |
|
at::Tensor offset, |
|
at::Tensor gradOutput, |
|
at::Tensor gradInput, |
|
at::Tensor gradOffset, |
|
at::Tensor weight, |
|
at::Tensor columns, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
int im2col_step) { |
|
if (gradOutput.is_cuda()) { |
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
TORCH_CHECK(input.is_cuda(), "input tensor is not on GPU!"); |
|
TORCH_CHECK(weight.is_cuda(), "weight tensor is not on GPU!"); |
|
TORCH_CHECK(offset.is_cuda(), "offset tensor is not on GPU!"); |
|
return deform_conv_backward_input_cuda( |
|
input, |
|
offset, |
|
gradOutput, |
|
gradInput, |
|
gradOffset, |
|
weight, |
|
columns, |
|
kW, |
|
kH, |
|
dW, |
|
dH, |
|
padW, |
|
padH, |
|
dilationW, |
|
dilationH, |
|
group, |
|
deformable_group, |
|
im2col_step); |
|
#else |
|
AT_ERROR("Detectron2 is not compiled with GPU support!"); |
|
#endif |
|
} |
|
AT_ERROR("This operator is not implemented on CPU"); |
|
} |
|
|
|
inline int deform_conv_backward_filter( |
|
at::Tensor input, |
|
at::Tensor offset, |
|
at::Tensor gradOutput, |
|
at::Tensor gradWeight, |
|
at::Tensor columns, |
|
at::Tensor ones, |
|
int kW, |
|
int kH, |
|
int dW, |
|
int dH, |
|
int padW, |
|
int padH, |
|
int dilationW, |
|
int dilationH, |
|
int group, |
|
int deformable_group, |
|
float scale, |
|
int im2col_step) { |
|
if (gradOutput.is_cuda()) { |
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
TORCH_CHECK(input.is_cuda(), "input tensor is not on GPU!"); |
|
TORCH_CHECK(offset.is_cuda(), "offset tensor is not on GPU!"); |
|
return deform_conv_backward_parameters_cuda( |
|
input, |
|
offset, |
|
gradOutput, |
|
gradWeight, |
|
columns, |
|
ones, |
|
kW, |
|
kH, |
|
dW, |
|
dH, |
|
padW, |
|
padH, |
|
dilationW, |
|
dilationH, |
|
group, |
|
deformable_group, |
|
scale, |
|
im2col_step); |
|
#else |
|
AT_ERROR("Detectron2 is not compiled with GPU support!"); |
|
#endif |
|
} |
|
AT_ERROR("This operator is not implemented on CPU"); |
|
} |
|
|
|
inline void modulated_deform_conv_forward( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor bias, |
|
at::Tensor ones, |
|
at::Tensor offset, |
|
at::Tensor mask, |
|
at::Tensor output, |
|
at::Tensor columns, |
|
int kernel_h, |
|
int kernel_w, |
|
const int stride_h, |
|
const int stride_w, |
|
const int pad_h, |
|
const int pad_w, |
|
const int dilation_h, |
|
const int dilation_w, |
|
const int group, |
|
const int deformable_group, |
|
const bool with_bias) { |
|
if (input.is_cuda()) { |
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
TORCH_CHECK(weight.is_cuda(), "weight tensor is not on GPU!"); |
|
TORCH_CHECK(bias.is_cuda(), "bias tensor is not on GPU!"); |
|
TORCH_CHECK(offset.is_cuda(), "offset tensor is not on GPU!"); |
|
return modulated_deform_conv_cuda_forward( |
|
input, |
|
weight, |
|
bias, |
|
ones, |
|
offset, |
|
mask, |
|
output, |
|
columns, |
|
kernel_h, |
|
kernel_w, |
|
stride_h, |
|
stride_w, |
|
pad_h, |
|
pad_w, |
|
dilation_h, |
|
dilation_w, |
|
group, |
|
deformable_group, |
|
with_bias); |
|
#else |
|
AT_ERROR("Detectron2 is not compiled with GPU support!"); |
|
#endif |
|
} |
|
AT_ERROR("This operator is not implemented on CPU"); |
|
} |
|
|
|
inline void modulated_deform_conv_backward( |
|
at::Tensor input, |
|
at::Tensor weight, |
|
at::Tensor bias, |
|
at::Tensor ones, |
|
at::Tensor offset, |
|
at::Tensor mask, |
|
at::Tensor columns, |
|
at::Tensor grad_input, |
|
at::Tensor grad_weight, |
|
at::Tensor grad_bias, |
|
at::Tensor grad_offset, |
|
at::Tensor grad_mask, |
|
at::Tensor grad_output, |
|
int kernel_h, |
|
int kernel_w, |
|
int stride_h, |
|
int stride_w, |
|
int pad_h, |
|
int pad_w, |
|
int dilation_h, |
|
int dilation_w, |
|
int group, |
|
int deformable_group, |
|
const bool with_bias) { |
|
if (grad_output.is_cuda()) { |
|
#if defined(WITH_CUDA) || defined(WITH_HIP) |
|
TORCH_CHECK(input.is_cuda(), "input tensor is not on GPU!"); |
|
TORCH_CHECK(weight.is_cuda(), "weight tensor is not on GPU!"); |
|
TORCH_CHECK(bias.is_cuda(), "bias tensor is not on GPU!"); |
|
TORCH_CHECK(offset.is_cuda(), "offset tensor is not on GPU!"); |
|
return modulated_deform_conv_cuda_backward( |
|
input, |
|
weight, |
|
bias, |
|
ones, |
|
offset, |
|
mask, |
|
columns, |
|
grad_input, |
|
grad_weight, |
|
grad_bias, |
|
grad_offset, |
|
grad_mask, |
|
grad_output, |
|
kernel_h, |
|
kernel_w, |
|
stride_h, |
|
stride_w, |
|
pad_h, |
|
pad_w, |
|
dilation_h, |
|
dilation_w, |
|
group, |
|
deformable_group, |
|
with_bias); |
|
#else |
|
AT_ERROR("Detectron2 is not compiled with GPU support!"); |
|
#endif |
|
} |
|
AT_ERROR("This operator is not implemented on CPU"); |
|
} |
|
|
|
} |
|
|
