[autotuner][cute] rename ab_stages "three" symbols that are no longer 3-only - #3151
Draft
calebmkim wants to merge 1 commit into
Draft
[autotuner][cute] rename ab_stages "three" symbols that are no longer 3-only#3151calebmkim wants to merge 1 commit into
calebmkim wants to merge 1 commit into
Conversation
… 3-only The tcgen05 AB-stage SMEM machinery originally only admitted a 3-stage pipeline, so its symbols were named "..._three_...". They are now depth-parametric (ab>=3, e.g. ab=4 for the narrow-N 16-bit cm2 tile and ab=12 for fp8), so the "three" in the name is misleading. Rename the symbols whose behavior spans more than ab=3; keep "three" only where the code genuinely still gates on ab==3. Renamed (parametric / ab>=3): ab_stages_three_fits -> ab_stages_fits ab_stages_three_search_constraints -> ab_stages_search_constraints allow_ab_stages_three_search -> allow_ab_stages_search allow_tcgen05_ab_stages_three_search -> allow_tcgen05_ab_stages_search Tcgen05AbStagesThreeSearchConstraints -> Tcgen05AbStagesSearchConstraints TCGEN05_AB_STAGES_THREE_RESERVED_SMEM_BYTES -> TCGEN05_AB_STAGES_RESERVED_SMEM_BYTES TCGEN05_AB_STAGES_THREE_MIN_DEVICE_SMEM_OPTIN -> TCGEN05_AB_STAGES_MIN_DEVICE_SMEM_OPTIN _tcgen05_ab_stages_three_search_constraints -> _tcgen05_ab_stages_search_constraints _tcgen05_ab_stages_three_fits -> _tcgen05_ab_stages_fits ab_stages_three_dtype_bytes / _device (kwargs) -> ab_stages_dtype_bytes / _device Kept "three" (genuinely ab==3-only): _fix_ab_stages_three_search_config and its _fix_tcgen05_ab_stages_three_search_config wrapper, which still early-return unless tcgen05_ab_stages == 3. Pure rename; no behavior change. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> stack-info: PR: #3151, branch: calebmkim/stack/35
calebmkim
force-pushed
the
calebmkim/stack/35
branch
from
July 24, 2026 17:09
08071ba to
b2dea58
Compare
calebmkim
force-pushed
the
calebmkim/stack/34
branch
from
July 24, 2026 17:09
6dbbe01 to
1128942
Compare
This was referenced Jul 24, 2026
meta-cla
Bot
added
the
CLA Signed
calebmkim
force-pushed
the
calebmkim/stack/35
branch
2 times, most recently
from
July 27, 2026 22:14
3627b90 to
44c0415
Compare
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
1 participant
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.
Stacked PRs:
[autotuner][cute] rename ab_stages "three" symbols that are no longer 3-only
The tcgen05 AB-stage SMEM machinery originally only admitted a 3-stage
pipeline, so its symbols were named "...three...". They are now
depth-parametric (ab>=3, e.g. ab=4 for the narrow-N 16-bit cm2 tile and
ab=12 for fp8), so the "three" in the name is misleading. Rename the
symbols whose behavior spans more than ab=3; keep "three" only where the
code genuinely still gates on ab==3.
Renamed (parametric / ab>=3):
ab_stages_three_fits -> ab_stages_fits
ab_stages_three_search_constraints -> ab_stages_search_constraints
allow_ab_stages_three_search -> allow_ab_stages_search
allow_tcgen05_ab_stages_three_search -> allow_tcgen05_ab_stages_search
Tcgen05AbStagesThreeSearchConstraints -> Tcgen05AbStagesSearchConstraints
TCGEN05_AB_STAGES_THREE_RESERVED_SMEM_BYTES -> TCGEN05_AB_STAGES_RESERVED_SMEM_BYTES
TCGEN05_AB_STAGES_THREE_MIN_DEVICE_SMEM_OPTIN -> TCGEN05_AB_STAGES_MIN_DEVICE_SMEM_OPTIN
_tcgen05_ab_stages_three_search_constraints -> _tcgen05_ab_stages_search_constraints
_tcgen05_ab_stages_three_fits -> _tcgen05_ab_stages_fits
ab_stages_three_dtype_bytes / _device (kwargs) -> ab_stages_dtype_bytes / _device
Kept "three" (genuinely ab==3-only): _fix_ab_stages_three_search_config
and its _fix_tcgen05_ab_stages_three_search_config wrapper, which still
early-return unless tcgen05_ab_stages == 3.
Pure rename; no behavior change.
Co-Authored-By: Claude Opus 4.8 (1M context) noreply@anthropic.com