Cuiqing Li 4b977541a8 [Kernels] added triton-implemented of self attention for colossal-ai (#4241) ... * added softmax kernel * added qkv_kernel * added ops * adding tests * upload tets * fix tests * debugging * debugging tests * debugging * added * fixed errors * added softmax kernel * clean codes * added tests * update tests * update tests * added attention * add * fixed pytest checking * add cuda check * fix cuda version * fix typo		2023-07-18 23:53:38 +08:00
..
cuda_native	[bf16] add bf16 support (#3882)	2023-06-05 15:58:31 +08:00
jit	fix Tensor is not defined (#4129)	2023-07-03 17:10:18 +08:00
triton	[Kernels] added triton-implemented of self attention for colossal-ai (#4241)	2023-07-18 23:53:38 +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