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.

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

Jump to bottom

[BUG]: all_reduce_fp8 is not compatible with torch.compile #6025

Closed
1 task done
GuangyaoZhang opened this issue Aug 22, 2024 · 0 comments · Fixed by #6040
Closed
1 task done

[BUG]: all_reduce_fp8 is not compatible with torch.compile #6025

GuangyaoZhang opened this issue Aug 22, 2024 · 0 comments · Fixed by #6040
Labels
bug Something isn't working

Comments

@GuangyaoZhang
Copy link
Contributor

GuangyaoZhang commented Aug 22, 2024 •
edited
Loading

Is there an existing issue for this bug?

  • I have searched the existing issues

🐛 Describe the bug

all_reduce_fp8 is not compatible with torch.compile.

img_v3_02e6_e1bf273d-fed3-42a0-9abc-e374ca300a6g

Environment

No response
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
bug Something isn't working
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

[FP8] Unsqueeze scale to make it compatible with torch.compile GuangyaoZhang/ColossalAI
1 participant
@GuangyaoZhang