Port roi_align (CUDA) to stable ABI. - #9573
Merged
Merged
Conversation
🔗 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. This comment was automatically generated by Dr. CI and updates every 15 minutes. |
NicolasHug
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 4, 2026 19:05
c76604f to
bf0a70a
Compare
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 17, 2026 19:48
bf0a70a to
823ee29
Compare
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 17, 2026 20:53
823ee29 to
5443e7f
Compare
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 17, 2026 21:39
5443e7f to
492c79e
Compare
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 17, 2026 21:56
492c79e to
5443e7f
Compare
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
2 times, most recently
from
August 24, 2026 15:25
c1fb40c to
a04cebd
Compare
adabeyta
marked this pull request as ready for review
August 24, 2026 16:29
adabeyta
force-pushed
the
adabeyta/roi-align-cuda-stable
branch
from
August 24, 2026 17:05
a04cebd to
9dd4643
Compare
This was referenced Aug 24, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Ports
roi_alignCUDA/ROCm kernels to the stable ABI.Notes
at::globalContext().alertNotDeterministiccall can't be made from a stable kernel, so it moves to the Python backward astorch._prims_common.alert_not_deterministic, gated tograd_output.device.type == "cuda". The CPU backward remains unchanged .Stable-ABI audit (
torch-abi-audit)Ran
torch-abi-auditon 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.