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
Hongxin Liu 554aa9592e
[legacy] move communication and nn to legacy and refactor logger (#4671)
1 year ago
..
cuda_native [example] add llama2 example (#4527) 1 year ago
jit [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
triton [Kernels] added triton-implemented of self attention for colossal-ai (#4241) 1 year 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