Skip to content

fix: experttensorparallel dynamo tracing crash - #4973

Open
devtyagi3909 wants to merge 2 commits into
pytorch:mainfrom
devtyagi3909:fix-expert-tensor-parallel
Open

devtyagi3909 wants to merge 2 commits into
pytorch:mainfrom
devtyagi3909:fix-expert-tensor-parallel

Conversation

@devtyagi3909

Copy link
Copy Markdown

fixes #3926

experttensorparallel was crashing during torch compile because devicemesh['tp'] tries to access a c++ bound attribute in _token_dispatch and _token_combine. fixed by eagerly slicing and caching the meshes in _apply.

@pytorch-bot

pytorch-bot Bot commented Oct 5, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/4973

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Oct 5, 2026

This branch has not been deployed

No deployments
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 Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

torch.compile + DeviceMesh bug: "'DeviceMesh' object has no attribute '_mesh_dim_names'"

1 participant