[Test] Port roi_pool (CUDA) to stable ABI.

[adabeyta]

Ports roi_pool's CUDA/ROCm kernels to the stable ABI.

Notes

• Requires [Stable ABI] Move fastAtomicAdd to torch/headeronly. pytorch#191501, which lands torch::headeronly::fastAtomicAdd (the stable-ABI replacement for the backward's at::native::fastAtomicAdd).
• Determinism layer change. The CUDA backward's at::globalContext().alertNotDeterministic call can't be made from a stable kernel, so it moves to the Python backward as torch._prims_common.alert_not_deterministic, gated to grad_output.device.type == "cuda".

Stable-ABI audit (torch-abi-audit)

Ran torch-abi-audit on a CUDA build. The counts don't move with this layer roi_pool's CUDA kernel references only unstable APIs already shared with the remaining legacy ops.

Package: torchvision
  Torch ABI:   UNSTABLE
  CPython ABI: n/a
  Bundled libs: 3
  -- bundled libs --
    [UNSTABLE] [not-abi3] _C.so            (stable_shim=0,  unstable=104)
    [STABLE  ] [not-abi3] _C_stable.so     (stable_shim=70, unstable=0)
    [STABLE  ] [not-abi3] image_stable.so  (stable_shim=76, unstable=0)

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9582

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 24 New Failures, 1 Pending, 1 Unclassified Failure

As of commit 1a7e398 with merge base 9f1d924 ():

NEW FAILURES - The following jobs have failed:

• Build Aarch64 Linux Wheels / pytorch/vision / build-wheel-py3_10-cuda-aarch6412_6-aarch64 (gh)
  RuntimeError: Error compiling objects for extension
• Build Aarch64 Linux Wheels / pytorch/vision / build-wheel-py3_10-cuda-aarch6413_0-aarch64 (gh)
  RuntimeError: Error compiling objects for extension
• Build Aarch64 Linux Wheels / pytorch/vision / build-wheel-py3_10-cuda-aarch6413_2-aarch64 (gh)
  RuntimeError: Error compiling objects for extension
• Build Aarch64 Linux Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda-aarch6412_6-aarch64 (gh)
  Unable to download artifact(s): Artifact not found
• Build Aarch64 Linux Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda-aarch6413_0-aarch64 (gh)
  Unable to download artifact(s): Artifact not found
• Build Aarch64 Linux Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda-aarch6413_2-aarch64 (gh)
  Unable to download artifact(s): Artifact not found
• Build Linux Wheels / pytorch/vision / build-manywheel-py3_10-cuda12_6 (gh)
  RuntimeError: Error compiling objects for extension
• Build Linux Wheels / pytorch/vision / build-manywheel-py3_10-cuda13_0 (gh)
  RuntimeError: Error compiling objects for extension
• Build Linux Wheels / pytorch/vision / build-manywheel-py3_10-cuda13_2 (gh)
  RuntimeError: Error compiling objects for extension
• Build Linux Wheels / pytorch/vision / build-manywheel-py3_10-rocm7_1 (gh)
  RuntimeError: Error compiling objects for extension
• Build Linux Wheels / pytorch/vision / build-manywheel-py3_10-rocm7_2 (gh)
  RuntimeError: Error compiling objects for extension
• Build Linux Wheels / pytorch/vision / upload / upload-manywheel-py3_10-cuda12_6 (gh)
  Unable to download artifact(s): Artifact not found
• Build Linux Wheels / pytorch/vision / upload / upload-manywheel-py3_10-cuda13_0 (gh)
  Unable to download artifact(s): Artifact not found
• Build Linux Wheels / pytorch/vision / upload / upload-manywheel-py3_10-cuda13_2 (gh)
  Unable to download artifact(s): Artifact not found
• Build Linux Wheels / pytorch/vision / upload / upload-manywheel-py3_10-rocm7_1 (gh)
  Unable to download artifact(s): Artifact not found
• Build Linux Wheels / pytorch/vision / upload / upload-manywheel-py3_10-rocm7_2 (gh)
  Unable to download artifact(s): Artifact not found
• Build Windows Wheels / pytorch/vision / build-wheel-py3_10-cuda12_6 (gh)
  C:\actions-runner\_work\vision\vision\pytorch\vision\torchvision\csrc\ops\cuda\roi_pool_kernel.cu(11): fatal error C1083: Cannot open include file: 'torch/headeronly/cuda/AtomicAdd.h': No such file or directory
• Build Windows Wheels / pytorch/vision / build-wheel-py3_10-cuda13_0 (gh)
  C:\actions-runner\_work\vision\vision\pytorch\vision\torchvision\csrc\ops\cuda\roi_pool_kernel.cu(11): fatal error C1083: Cannot open include file: 'torch/headeronly/cuda/AtomicAdd.h': No such file or directory
• Build Windows Wheels / pytorch/vision / build-wheel-py3_10-cuda13_2 (gh)
  C:\actions-runner\_work\vision\vision\pytorch\vision\torchvision\csrc\ops\cuda\roi_pool_kernel.cu(11): fatal error C1083: Cannot open include file: 'torch/headeronly/cuda/AtomicAdd.h': No such file or directory
• Build Windows Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda12_6 (gh)
  Unable to download artifact(s): Artifact not found
• Build Windows Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda13_0 (gh)
  Unable to download artifact(s): Artifact not found
• Build Windows Wheels / pytorch/vision / upload / upload-wheel-py3_10-cuda13_2 (gh)
  Unable to download artifact(s): Artifact not found
• CMake / linux (linux.g5.4xlarge.nvidia.gpu, cuda, 12.6) / linux-job (gh)
  RuntimeError: Command docker exec -t 46bfa0c0740cbb5873948c7209e6b4ba4403ce6d50c1e76771553f016be10dc8 /exec failed with exit code 1
• Tests / unittests-linux (3.10, linux.g5.4xlarge.nvidia.gpu, cuda, 12.8) / linux-job (gh)
  RuntimeError: Command docker exec -t a4a0bb0f58f16bc1df74772aec40c77c72dea6599c65ff80213986ed9c887d0d /exec failed with exit code 1

UNCLASSIFIED FAILURE - DrCI could not classify the following job because the workflow did not run on the merge base. The failure may be pre-existing on trunk or introduced by this PR:

• .github/workflows/pr-labels.yml (gh) (this job did not run on the merge base, so DrCI cannot tell whether the failure is pre-existing)

This comment was automatically generated by Dr. CI and updates every 15 minutes.