ColossalAI/colossalai/kernel
zbian 6877121377 updated flash attention api 2022-11-15 15:25:39 +08:00
..
cuda_native updated flash attention api 2022-11-15 15:25:39 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
__init__.py