Port deform_conv2d (CUDA) to stable ABI. (#9626)

Co-authored-by: Nicolas Hug <nh.nicolas.hug@gmail.com>

Author: adabeyta

CMakeLists.txt

@@ -119,6 +119,7 @@ set(TORCHVISION_STABLE_SOURCES
   ${TVCPP}/ops/cuda/ps_roi_align_kernel.cu
   ${TVCPP}/ops/deform_conv2d.cpp
   ${TVCPP}/ops/cpu/deform_conv2d_kernel.cpp
+  ${TVCPP}/ops/cuda/deform_conv2d_kernel.cu
   ${TVCPP}/io/image/cuda/decode_jpegs_cuda.cpp
   ${TVCPP}/io/image/cuda/encode_jpegs_cuda.cpp
   ${TVCPP}/io/image/cpu/encode_png.cpp

setup.py

@@ -187,6 +187,7 @@ def get_macros_and_flags():
 )
 STABLE_SOURCES.add(CSRS_DIR / ("ops/hip/roi_pool_kernel.hip" if IS_ROCM else "ops/cuda/roi_pool_kernel.cu"))
 STABLE_SOURCES.add(CSRS_DIR / ("ops/hip/roi_align_kernel.hip" if IS_ROCM else "ops/cuda/roi_align_kernel.cu"))
+STABLE_SOURCES.add(CSRS_DIR / ("ops/hip/deform_conv2d_kernel.hip" if IS_ROCM else "ops/cuda/deform_conv2d_kernel.cu"))
 STABLE_SOURCES.add(CSRS_DIR / ("ops/hip/ps_roi_pool_kernel.hip" if IS_ROCM else "ops/cuda/ps_roi_pool_kernel.cu"))
 STABLE_SOURCES.add(CSRS_DIR / ("ops/hip/ps_roi_align_kernel.hip" if IS_ROCM else "ops/cuda/ps_roi_align_kernel.cu"))
 

torchvision/_autograd_registrations.py

@@ -169,6 +169,8 @@ def _deform_conv2d_setup_context(ctx, inputs, output):
 
 
 def _deform_conv2d_backward(ctx, grad_output):
+    if grad_output.device.type == "cuda":
+        torch._prims_common.alert_not_deterministic("compute_grad_input")
     input, weight, offset, mask, bias = ctx.saved_tensors
     grad_input, grad_weight, grad_offset, grad_mask, grad_bias = torch.ops.torchvision._deform_conv2d_backward(
         grad_output,