ColossalAI/colossalai/kernel
Frank Lee 918bc94b6b
[triton] added copyright information for flash attention (#2835)
* [triton] added copyright information for flash attention

* polish code
2023-02-21 11:25:57 +08:00
..
cuda_native [triton] added copyright information for flash attention (#2835) 2023-02-21 11:25:57 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +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