Allow `device_mesh` have multiple dim #38949

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

Merged

SunMarc merged 8 commits into main from fsdp2-tp

Jul 23, 2025

+10 −5

Merged

Allow `device_mesh` have multiple dim #38949

Merge branch 'main' into fsdp2-tp

CircleCI Checks / setup_and_quality succeeded Jul 23, 2025 in 5m 27s

Workflow: setup_and_quality

View CircleCI Workflow

View more details on CircleCI Checks