Skip to content

[Test] Port roi_align (CUDA) to stable ABI. - #9573

Draft
adabeyta wants to merge 1 commit into
adabeyta/roi-align-cpu-stablefrom
adabeyta/roi-align-cuda-stable
Draft

[Test] Port roi_align (CUDA) to stable ABI.#9573
adabeyta wants to merge 1 commit into
adabeyta/roi-align-cpu-stablefrom
adabeyta/roi-align-cuda-stable

Conversation

@adabeyta

@adabeyta adabeyta commented Jul 30, 2026

Copy link
Copy Markdown
Collaborator

Ports roi_align CUDA/ROCm kernels to the stable ABI.

Notes

  • Requires [Stable ABI] Move fastAtomicAdd to torch/headeronly. pytorch#191501.
  • 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". The CPU backward remains unchanged .

Stable-ABI audit (torch-abi-audit)

Ran torch-abi-audit on a CUDA build. The counts don't move with this layer since the CUDA kernel's unstable references are all 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=103)
        [STABLE  ] [not-abi3] _C_stable.so     (stable_shim=70, unstable=0)
        [STABLE  ] [not-abi3] image_stable.so  (stable_shim=76, unstable=0)

@pytorch-bot

pytorch-bot Bot commented Jul 30, 2026 •
edited
Loading

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 24 New Failures, 1 Cancelled Job, 1 Unclassified Failure

As of commit c76604f with merge base 10f68db (image):

NEW FAILURES - The following jobs have failed:

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:

CANCELLED JOB - The following job was cancelled. Please retry:

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

@meta-cla meta-cla Bot added the cla signed label Jul 30, 2026
@adabeyta
adabeyta changed the base branch from main to adabeyta/roi-align-cpu-stable July 30, 2026 21:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

cla signed

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@adabeyta