Skip to content
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.

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

CP 32 + torch.compile hang #772

Open
tianyu-l opened this issue Jan 3, 2025 · 1 comment
Open

CP 32 + torch.compile hang #772

tianyu-l opened this issue Jan 3, 2025 · 1 comment
Labels
bug Something isn't working release_blocking Issues that are blocking the milestone / release completion

Comments

@tianyu-l
Copy link
Contributor

tianyu-l commented Jan 3, 2025

CP 16 + torch.compile + Float8 works, but CP 32 + torch.compile would hang.

@tianyu-l tianyu-l added bug Something isn't working release_blocking Issues that are blocking the milestone / release completion labels Jan 3, 2025
@tianyu-l tianyu-l added this to the torchtitan v1.0.0 release milestone Jan 3, 2025
@fegin
Copy link
Contributor

fegin commented Jan 8, 2025

@XilunWu Can we add some update here to keep track that it happens only when torch.compile is enabled?

@tianyu-l tianyu-l changed the title CP hangs when degree is more than 8 CP 32 + torch.compile hang Jan 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working release_blocking Issues that are blocking the milestone / release completion
Projects
None yet
Development

No branches or pull requests

2 participants