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

re-enable CP tests since DTensor scaled dot product flash attention has been fixed #912

Merged
merged 1 commit into from
Mar 3, 2025

Conversation

XilunWu
Copy link
Contributor

@XilunWu XilunWu commented Mar 3, 2025

Stack from ghstack (oldest at bottom):

Summary

This PR undo #898 and re-enables CP tests in CI as pytorch/pytorch#148125 fixed the DTensor sdp flash attention op.

Test

CI

XilunWu added a commit that referenced this pull request Mar 3, 2025
…as been fixed

ghstack-source-id: b2c441d06db4c737e0cffabacf289145ed535826
Pull Request resolved: #912
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Mar 3, 2025
@XilunWu XilunWu requested review from tianyu-l and fegin March 3, 2025 09:09
@fegin fegin merged commit 9036a08 into gh/XilunWu/17/base Mar 3, 2025
6 checks passed
fegin added a commit that referenced this pull request Mar 3, 2025
#921)

…as been fixed (#912)

Stack from [ghstack](/~https://github.com/ezyang/ghstack) (oldest at
bottom):
* __->__ #912

### Summary
This PR undo #898 and
re-enables CP tests in CI as
pytorch/pytorch#148125 fixed the DTensor sdp
flash attention op.

### Test
CI

Co-authored-by: Xilun Wu <12968408+XilunWu@users.noreply.github.com>
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.

3 participants