This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-04-26 09:42:27 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
458ae331ad4db52243eebdda551cbc2b65a5c73f
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
