Skip to content

[CK_TILE] Rename Stream-K grid function#4795

Open
arai713 wants to merge 3 commits intodevelopfrom
ck/arai/ck_tile/grid_size_rename
Open

[CK_TILE] Rename Stream-K grid function#4795
arai713 wants to merge 3 commits intodevelopfrom
ck/arai/ck_tile/grid_size_rename

Conversation

@arai713
Copy link
Contributor

@arai713 arai713 commented Feb 23, 2026

Motivation

This PR introduces a change in the name of the get_grid function in the Stream-K TilePartitioner to avoid confusion with a similarly named method. In the Stream-K TilePartitioner, there is get_grid() which returns num_cu*occupancy and there is grid_size() which returns the grid size used to launch the kernel. In this PR, we change get_grid() to be get_max_active_wgs() to better reflect what the function returns and not confuse it with grid_size().

Technical Details

Initially in the Stream-K TilePartitioner we had get_grid() which returned grid_. We are renaming get_grid() to get_max_active_wgs() and grid_ to max_active_wgs_ internally, while keeping grid_size() the same. The parameter, grid, for the Stream-K TilePartitioner remains the same to maintain consistency with the rest of the Stream-K API.

Test Plan

Validated using the test suite that is already present.

Test Result

All tests passed

Submission Checklist

@arai713 arai713 force-pushed the ck/arai/ck_tile/grid_size_rename branch from 1029482 to 946af83 Compare February 24, 2026 18:32
@arai713 arai713 force-pushed the ck/arai/ck_tile/grid_size_rename branch 3 times, most recently from 7796133 to 5612f49 Compare March 4, 2026 18:46
@arai713 arai713 force-pushed the ck/arai/ck_tile/grid_size_rename branch 5 times, most recently from 5d4df2b to dc14f61 Compare March 10, 2026 16:55
@arai713 arai713 requested review from a team as code owners March 11, 2026 21:42
@math-ci-webhook
Copy link

perfci run on commit 229035a

math-ci run

Copy link
Contributor

@cgmillette cgmillette left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@cgmillette
Copy link
Contributor

Have some build errors in Tile engine

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.

5 participants