Skip to content

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

Merged
fegin merged 1 commit intogh/XilunWu/17/basefrom
gh/XilunWu/17/head
Mar 3, 2025
Merged

re-enable CP tests since DTensor scaled dot product flash attention has been fixed#912
fegin merged 1 commit intogh/XilunWu/17/basefrom
gh/XilunWu/17/head

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: b2c441d
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 fegin and tianyu-l 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>
MaxiBoether pushed a commit to eth-easl/torchtitan-mixtera that referenced this pull request Apr 17, 2025
pytorch#921)

…as been fixed (pytorch#912)

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

### Summary
This PR undo pytorch#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>
xrsrke pushed a commit to NousResearch/torchtitan that referenced this pull request Feb 13, 2026
pytorch#921)

…as been fixed (pytorch#912)

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

### Summary
This PR undo pytorch#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>
xrsrke pushed a commit to NousResearch/torchtitan that referenced this pull request Feb 25, 2026
pytorch#921)

…as been fixed (pytorch#912)

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

### Summary
This PR undo pytorch#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