Skip to content

Cap remaining block-bucketize launches (#6292) - #6292

Closed
q10 wants to merge 1 commit into
pytorch:mainfrom
q10:export-D116013308
Closed

q10 wants to merge 1 commit into
pytorch:mainfrom
q10:export-D116013308

Conversation

@q10

@q10 q10 commented Sep 11, 2026

Copy link
Copy Markdown
Contributor

Summary:

NOTE: No linked task. Please associate a task with this diff.

Cap the remaining one-dimensional block-bucketize launches on ROCm. Keep the two-dimensional thread-block size as threads_per_block.

Change capped-loop early exits from return to continue so head, middle, and tail rows remain covered.

Reviewed By: jianyuh

Differential Revision: D116013308

@meta-cla meta-cla Bot added the cla signed label Sep 11, 2026
@meta-codesync

meta-codesync Bot commented Sep 11, 2026

Copy link
Copy Markdown
Contributor

@q10 has exported this pull request. If you are a Meta employee, you can view the originating Diff in D116013308.

Summary:

NOTE: No linked task. Please associate a task with this diff.

Cap the remaining one-dimensional block-bucketize launches on ROCm. Keep the two-dimensional thread-block size as `threads_per_block`.

Change capped-loop early exits from `return` to `continue` so head, middle, and tail rows remain covered.

Reviewed By: jianyuh

Differential Revision: D116013308
@meta-codesync meta-codesync Bot changed the title Cap remaining block-bucketize launches Cap remaining block-bucketize launches (#6292) Sep 12, 2026
@q10
q10 force-pushed the export-D116013308 branch from beed2c1 to 36f877b Compare September 12, 2026 01:04
@meta-codesync meta-codesync Bot closed this in 6ceefee Sep 14, 2026
@meta-codesync meta-codesync Bot added the Merged label Sep 14, 2026
@meta-codesync

meta-codesync Bot commented Sep 14, 2026

Copy link
Copy Markdown
Contributor

This pull request has been merged in 6ceefee.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant