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
__init__.py