Skip to content

[ROCDL] cdna5 tdm_partition: relate only mode-0, pass rest modes through - #1079

Open
xudoyuan wants to merge 1 commit into
pr/enh-cdna5-tiled-tdm-atomfrom
enh-cdna5-tdm-partition-restmode
Open

[ROCDL] cdna5 tdm_partition: relate only mode-0, pass rest modes through#1079
xudoyuan wants to merge 1 commit into
pr/enh-cdna5-tiled-tdm-atomfrom
enh-cdna5-tdm-partition-restmode

Conversation

@xudoyuan

Copy link
Copy Markdown
Collaborator

partitionLayout now cuts the LDS tile and coordinate tile by their mode-0 (the TDM box) and validates only size<0>.

tdm_partition applies the returned mode-0 tiler by-mode (make_tile padded with None per rest mode), so only mode 0 is reshaped and the rest is left alone; the warp slice indexes the nested mode 0.

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

partitionLayout now cuts the LDS tile and coordinate tile by their mode-0
(the TDM box) and validates only size<0>. Trailing "rest" modes -- e.g. a
PIPE double-buffer on the LDS side, k-tiles on the coordinate side -- pass
through untouched and may differ in size.

tdm_partition applies the returned mode-0 tiler by-mode (make_tile padded
with None per rest mode), so only mode 0 is reshaped and the rest is left
alone; the warp slice indexes the nested mode 0.

Example 06 streams K tiles through a single reused LDS box, exercising the
differing rest sizes (LDS rest = 1, coordinate rest = K).

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@xudoyuan
xudoyuan marked this pull request as ready for review August 31, 2026 06:41
@xudoyuan
xudoyuan requested a review from sjfeng1999 August 31, 2026 06:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant