Skip to content

[RFC] RaggedIterDomain for nested tensors #5550

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
naoyam wants to merge 24 commits into
base: main
Choose a base branch
from
+868 −0
Open

[RFC] RaggedIterDomain for nested tensors #5550

Changes from 20 commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
071ec53
WIP
naoyam Nov 18, 2025
190b1c1
cleanup
naoyam Nov 18, 2025
f6a03fc
Add select operation and proofreading fixes to RaggedIterDomain design
naoyam Nov 18, 2025
2087623
cleanup
naoyam Nov 18, 2025
e29aa18
update
naoyam Nov 19, 2025
65a29bd
update
naoyam Nov 19, 2025
4857735
cleanup
naoyam Nov 19, 2025
768d271
update
naoyam Nov 20, 2025
ab5413f
update
naoyam Nov 20, 2025
e9207f2
move
naoyam Nov 20, 2025
70e0885
license
naoyam Nov 20, 2025
652ece0
Update RaggedIterDomain design for multi-level nesting and partition/…
naoyam Dec 4, 2025
4788917
Update expert parallelism example with accurate multi-GPU workflow
naoyam Dec 4, 2025
eb10af5
Move multi-level nesting from requirements to future work
naoyam Dec 4, 2025
51178e0
Remove Case 2 (partitioning RaggedIterDomain) from partition semantics
naoyam Dec 4, 2025
d1b9852
Update RaggedIterDomain design: change internal representation to ext…
naoyam Dec 4, 2025
b65e560
Document merge operation semantics for RaggedIterDomain with extent r…
naoyam Dec 5, 2025
d8e9aae
Document runtime binding approach for nested tensors (initial version)
naoyam Dec 5, 2025
ca0c9aa
Add implementation detail for Case 2b merge using ReductionOp
naoyam Dec 5, 2025
e999026
Fix Section 6.1: clarify offset tensor can be multi-dimensional
naoyam Dec 5, 2025
04c3fd6
Clarify load balancing and multiple ragged IDs
naoyam Dec 5, 2025
81e1a62
Drop RaggedIterDomain::offsets
naoyam Dec 5, 2025
ce7886f
PR feedback
naoyam Dec 5, 2025
8cfce69
format
naoyam Dec 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading