xla
2298cbcb
- Add flash attention as custom op so dynamo can trace it (#6875)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Add flash attention as custom op so dynamo can trace it (#6875)
References
#6884 - [Backport] Add flash attention as custom op so dynamo can trace it (#6875)
Author
JackCaoG
Committer
JackCaoG
Parents
6e4b2ed4
Loading
