Skip to content

[WIP] Enforce nested traversals for Partition Scheduling#987

Draft
njriasan wants to merge 2 commits intofacebookexperimental:mainfrom
njriasan:njriasan/nested_store_assignment
Draft

[WIP] Enforce nested traversals for Partition Scheduling#987
njriasan wants to merge 2 commits intofacebookexperimental:mainfrom
njriasan:njriasan/nested_store_assignment

Conversation

@njriasan
Copy link
Contributor

The Partition scheduler only identifies top level load/store operations, which breaks our core assumptions about partition annotations in AutoWS. This updates the load/store to do a full walk.

The schedule is likely still suboptimal because similar to the for work the if is being assigned a partition.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Feb 26, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant