github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-20 11:02:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3da9993b0d03923755c1fcd6279cc4c7b8d00d1e
ColossalAI/colossalai/kernel/triton/flash_decoding.py
Yuanheng Zhao 3da9993b0d [Kernel/Fix] Revise flash attention triton kernel API and add benchmark (#5301) 
* fix decoding kernel pytest

* revise and add triton context attn benchmark
2024-01-23 17:16:02 +08:00

11 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink