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
..
arm	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
common	[Inference]Support FP16/BF16 Flash Attention 2 And Add high_precision Flag To Rotary Embedding (#5461)	2024-03-25 13:40:34 +08:00
cuda	[Inference]Support FP16/BF16 Flash Attention 2 And Add high_precision Flag To Rotary Embedding (#5461)	2024-03-25 13:40:34 +08:00
x86	refactor code	2024-03-11 17:06:57 +08:00
__init__.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
scaled_softmax.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00