ColossalAI/colossalai/kernel
oahzxl 25952b67d7
[feat] add flash attention (#1762)
2022-10-26 16:15:52 +08:00
..
cuda_native [feat] add flash attention (#1762) 2022-10-26 16:15:52 +08:00
jit [NFC] polish colossalai/kernel/jit/bias_gelu.py code style (#946) 2022-05-17 10:25:06 +08:00
__init__.py fix format (#568) 2022-04-06 11:40:59 +08:00