Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/131712
Note: Links to docs will display an error until the docs builds have been completed. ❌ 6 New FailuresAs of commit d14484c with merge base f389bca ( NEW FAILURES - The following jobs have failed:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
ghstack-source-id: cdfdb9a Pull Request resolved: pytorch#131712
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
|
@drisspg Now that the CUTLASS upgrade is merged in, might be time to rebase and update flash attention. :) |
|
@Skylion007 Ahh sweet thanks for the ping, I will rebase |
| # Suspect users building from source will need this | ||
| add_definitions(-DFLASHATTENTION_DISABLE_ALIBI) | ||
| # Same for softcapping | ||
| add_definitions(-FLASHATTENTION_DISABLE_SOFTCAP) |
There was a problem hiding this comment.
| add_definitions(-FLASHATTENTION_DISABLE_SOFTCAP) | |
| add_definitions(-DFLASHATTENTION_DISABLE_SOFTCAP) |
|
Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as |
Stack from ghstack (oldest at bottom):