ColossalAI/colossalai/kernel
zbian 7bc0afc901 updated flash attention usage 2023-03-20 17:57:04 +08:00
..
cuda_native updated flash attention usage 2023-03-20 17:57:04 +08:00
jit [misc] add reference (#2930) 2023-02-28 18:07:24 +08:00
__init__.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
op_builder [builder] reconfig op_builder for pypi install (#2314) 2023-01-04 16:32:32 +08:00