github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-05 12:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ac178ca5c17ca751ff9df38be81da2b4a005fc0d
ColossalAI/colossalai/kernel/cuda_native/mha/utils.py
flybird1111 25c57b9fb4 [fix] coloattention support flash attention 2 (#4347) 
Improved ColoAttention interface to support flash attention 2. Solved #4322
2023-08-04 13:46:22 +08:00

2.7 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink