Tri Dao c984208ddb Set block size to 64 x 64 for kvcache to avoid nvcc segfaults		2023-09-17 16:14:58 -07:00
..
src	Set block size to 64 x 64 for kvcache to avoid nvcc segfaults	2023-09-17 16:14:58 -07:00
flash_api.cpp	Implement rotary embedding in flash_attn_with_kvcache	2023-09-16 01:20:16 -07:00