github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/kernel/cuda_native/mha
History
flybird1111 38b792aab2
[coloattention] fix import error (#4380) 
fixed an import error
 		2023-08-04 16:28:41 +08:00
..
__init__.py [coloattention] fix import error (#4380) 2023-08-04 16:28:41 +08:00
flash_attn_2.py [fix] coloattention support flash attention 2 (#4347) 2023-08-04 13:46:22 +08:00
mem_eff_attn.py [fix] coloattention support flash attention 2 (#4347) 2023-08-04 13:46:22 +08:00
mha.py [fix] coloattention support flash attention 2 (#4347) 2023-08-04 13:46:22 +08:00
utils.py [fix] coloattention support flash attention 2 (#4347) 2023-08-04 13:46:22 +08:00