yuehuayingxueluo 87079cffe8 [Inference]Support FP16/BF16 Flash Attention 2 And Add high_precision Flag To Rotary Embedding (#5461) ... * Support FP16/BF16 Flash Attention 2 * fix bugs in test_kv_cache_memcpy.py * add context_kv_cache_memcpy_kernel.cu * rm typename MT * add tail process * add high_precision * add high_precision to config.py * rm unused code * change the comment for the high_precision parameter * update test_rotary_embdding_unpad.py * fix vector_copy_utils.h * add comment for self.high_precision when using float32		2024-03-25 13:40:34 +08:00
..
inference.cpp	[Inference]Support FP16/BF16 Flash Attention 2 And Add high_precision Flag To Rotary Embedding (#5461)	2024-03-25 13:40:34 +08:00
layer_norm.cpp	fix include path	2024-03-13 11:21:06 +08:00
moe.cpp	refactor code	2024-03-11 17:06:57 +08:00
optimizer.cpp	refactor code	2024-03-11 17:06:57 +08:00
scaled_masked_softmax.cpp	refactor code	2024-03-11 17:06:57 +08:00
scaled_upper_triang_masked_softmax.cpp	refactor code	2024-03-11 17:06:57 +08:00