Files
vllm/benchmark/benchmark_attention.py
Woosuk Kwon 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

4.8 KiB