Commit Graph
1 Commits
Author SHA1 Message Date
Woosuk KwonandGitHub c267b1a02c Add query stride to multi_query_cached_kv_attention & Add kernel benchmark script (#27)
* Add query stride to multi_query_cached_kv_attention

* Add kernel benchmark script
2023-04-08 13:36:09 -07:00