Skip to content

Conversation

allenwang28
Copy link
Contributor

We previously would slice on procs=0, but if the proc_mesh is N-d (like hosts=2) then the actor extent/shape is {"procs": 1, "hosts": 2}

This change will reduce any N-d into just a singleton/0th rank on all dimensions

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Oct 7, 2025
@allenwang28 allenwang28 merged commit 67d2c38 into meta-pytorch:main Oct 7, 2025
8 checks passed
@allenwang28 allenwang28 deleted the provisioner_slice branch October 7, 2025 15:19
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.

2 participants