Skip to content

[release/1.6] Fix CUDA indexing, graph execution and output buffers - #24488

Merged
JacobSzwejbka merged 1 commit into
release/1.6from
codex/release-1.6-clean-cuda-index-20261103
Nov 3, 2026
Merged

JacobSzwejbka merged 1 commit into
release/1.6from
codex/release-1.6-clean-cuda-index-20261103

Conversation

@JacobSzwejbka

Copy link
Copy Markdown
Contributor

Supersedes #24320 with a clean release pick from #24299.

Adds CUDA mask-index shims, corrects graph output layout and captured-shape handling, and checks bound output buffers against their upper bounds. Includes the graph retry, host-constant handling, allocation bookkeeping, input-shape checks and wheel shim-check follow-ups already carried by #24320.

The Windows import library preserves the existing release dtype-element-size export while adding the three source PR exports. Buffers smaller than the maximum planned output size and replays with changed input shapes are rejected.

Scope and landing order

First in the verified landing order; #24298 is already merged and remains the baseline.

This branch starts directly at release/1.6 commit dfb78b9ecd933a5fee33ebc94f69ed4b71cbb1e7. Its GitHub diff contains 21 files for this pick; it does not carry duplicate #24298 history or the excluded unmerged QNN work. Dependencies are documented here rather than embedded as other cherry-pick commits.

Verification

All changed Python files compile and pass Ruff F821 checks. All seven replacements merge cleanly in the documented sequence, including simulated squash merges. No QNN files are changed.

Native CUDA, OpenVINO and portable C++ suites were not run locally on this macOS host; PR CI remains required before landing.

Prepared with OpenAI Codex.

Cherry-pick #24299 with the CUDA graph, input-shape and shim-check follow-ups already carried by #24320. Resolve the Windows import library by retaining the release dtype-element-size export alongside the three indexing/get-size exports.

Prepared with OpenAI Codex.
@pytorch-bot

pytorch-bot Bot commented Nov 3, 2026 •

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/24488

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

❌ 7 New Failures

As of commit 8a7c7bb with merge base 4db778e (image):

NEW FAILURES - The following jobs have failed:

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

@JacobSzwejbka
JacobSzwejbka merged commit f79349d into release/1.6 Nov 3, 2026
476 of 485 checks passed
@JacobSzwejbka
JacobSzwejbka deleted the codex/release-1.6-clean-cuda-index-20261103 branch November 3, 2026 17:03
JacobSzwejbka added a commit that referenced this pull request Nov 3, 2026
Supersedes #24324 with a clean release pick from #24314.

Adds validated per-program CUDA weights filenames and lowerings for
antialiased bilinear/bicubic resize. Retains the data-name validation,
resize-test removal and documentation follow-ups already carried by
#24324.

Preserves the release behavior that mismatched CUDA weights fail to
load. The resize rewrite follows the source PR tolerances rather than
promising bit-identical floating-point results.

## Scope and landing order

Land after #24488 in the
verified landing order.

This branch starts directly at release/1.6 commit
`dfb78b9ecd933a5fee33ebc94f69ed4b71cbb1e7`. Its GitHub diff contains 10
files for this pick; it does not carry duplicate #24298 history or the
excluded unmerged QNN work. Dependencies are documented here rather than
embedded as other cherry-pick commits.

## Verification

All changed Python files compile and pass Ruff F821 checks. All seven
replacements merge cleanly in the documented sequence, including
simulated squash merges. No QNN files are changed.

Native CUDA, OpenVINO and portable C++ suites were not run locally on
this macOS host; PR CI remains required before landing.

Prepared with OpenAI Codex.

This branch was successfully deployed

1 active and 1 inactive deployments
upload-benchmark-results — 8a7c7bb4 Deployed Nov 3, 2026 by JacobSzwejbka via upload-benchmark-results #21798
cadence — 8a7c7bb4 Deployed Nov 3, 2026 by JacobSzwejbka via hifi-op-test / hifi4 #36184
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants