Skip to content

Account for X/Z blocks in grid-Y caps - #6290

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

Account for X/Z blocks in grid-Y caps#6290
q10 wants to merge 1 commit into
pytorch:mainfrom
q10:export-D116806924

Conversation

@q10

@q10 q10 commented Sep 10, 2026

Copy link
Copy Markdown
Contributor

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

Migrate rowwise quantization launches to cap grid.y with the Y-specific helper. Pass threads_per_block and the unchanged X/Z plane separately; kernels keep X/Z unchanged and grid-stride over rows.

Document that OverflowOnly skips the total-thread cap on CUDA while the helper still enforces the CUDA grid-y driver limit.

Reviewed By: jianyuh

Differential Revision: D116806924

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

Migrate rowwise quantization launches to cap `grid.y` with the Y-specific helper. Pass `threads_per_block` and the unchanged X/Z plane separately; kernels keep X/Z unchanged and grid-stride over rows.

Document that `OverflowOnly` skips the total-thread cap on CUDA while the helper still enforces the CUDA grid-y driver limit.

Reviewed By: jianyuh

Differential Revision: D116806924
@meta-cla meta-cla Bot added the cla signed label Sep 10, 2026
@meta-codesync

meta-codesync Bot commented Sep 10, 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 D116806924.

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