flybird1111 38b792aab2 [coloattention] fix import error (#4380) ... fixed an import error		2023-08-04 16:28:41 +08:00
..
csrc	[bf16] add bf16 support (#3882)	2023-06-05 15:58:31 +08:00
mha	[coloattention] fix import error (#4380)	2023-08-04 16:28:41 +08:00
__init__.py	[fix] coloattention support flash attention 2 (#4347)	2023-08-04 13:46:22 +08:00
layer_norm.py	[kernel] fixed repeated loading of kernels (#2549)	2023-02-03 09:47:13 +08:00
multihead_attention.py	[nfc] fix typo colossalai/cli fx kernel (#3847)	2023-06-02 15:02:45 +08:00
scaled_softmax.py	[fix] coloattention support flash attention 2 (#4347)	2023-08-04 13:46:22 +08:00