You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/kernel
Frank Lee 918bc94b6b
[triton] added copyright information for flash attention (#2835)
2 years ago
..
cuda_native [triton] added copyright information for flash attention (#2835) 2 years ago
jit [kernel] added jit warmup (#1792) 2 years ago
__init__.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
op_builder [builder] reconfig op_builder for pypi install (#2314) 2 years ago