xla
1ce54b1a
- support test__flash_attention_impl
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
support test__flash_attention_impl
References
#6870 - [Pallas] Support Flash Attention backward kernels
Author
alanwaketan
Parents
88f93fe7
Loading