heyuhhh f07e9977c6 [None] [feat] Use triton kernels for RocketKV prediction module (#8682) ... Signed-off-by: yuhangh <58161490+heyuhhh@users.noreply.github.com>		2025-11-13 18:51:09 -08:00
..
sparse	[None] [feat] Use triton kernels for RocketKV prediction module (#8682)	2025-11-13 18:51:09 -08:00
__init__.py	[TRTLLM-8536][feat] Add the sparse attention framework and one use case--RocketKV support (#8086)	2025-10-14 08:23:16 -07:00
flashinfer.py	[#6507][fix] Fix precision issue due to KV layout mismatch for split/concat kernels (#6917)	2025-11-13 12:14:58 +08:00
interface.py	[TRTLLM-8690][feat] add more tensors to share buffers (#8691)	2025-11-03 21:08:01 -08:00
star_flashinfer.py	[#6507][fix] Fix precision issue due to KV layout mismatch for split/concat kernels (#6917)	2025-11-13 12:14:58 +08:00
trtllm.py	[None] [feat] Use triton kernels for RocketKV prediction module (#8682)	2025-11-13 18:51:09 -08:00
utils.py	[TRTLLM-8535][feat] Support DeepSeek V3.2 with FP8 + BF16 KV cache/NVFP4 + BF16 KV cache (#8405)	2025-10-24 13:40:41 -04:00
vanilla.py	[TRTLLM-8536][feat] Add the sparse attention framework and one use case--RocketKV support (#8086)	2025-10-14 08:23:16 -07:00