TensorRT-LLMs/cpp/tensorrt_llm
Shunkangz 8ee840159b
Add updateKVCacheTransfer (#2984)
Add kv cache transfer measurement
Signed-off-by: Shunkang <182541032+Shunkangz@users.noreply.github.co>
Co-authored-by: Shunkang <182541032+Shunkangz@users.noreply.github.co>
2025-03-25 21:45:35 +08:00
..
batch_manager Add updateKVCacheTransfer (#2984) 2025-03-25 21:45:35 +08:00
common Add updateKVCacheTransfer (#2984) 2025-03-25 21:45:35 +08:00
cutlass_extensions/include/cutlass_extensions Update (#2978) 2025-03-23 16:39:35 +08:00
executor Update (#2978) 2025-03-23 16:39:35 +08:00
executor_worker Update TensorRT-LLM (#2792) 2025-02-18 21:27:39 +08:00
kernels fix: [MLA] fix the bug with fp8 MLA kernels on Blackwell. (#3008) 2025-03-25 18:03:29 +08:00
layers Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
plugins Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
pybind fix: disable kv cache reuse when minimum window size is reached, instead of maximum window size (#2983) 2025-03-24 22:49:52 +08:00
runtime fix: disable kv cache reuse when minimum window size is reached, instead of maximum window size (#2983) 2025-03-24 22:49:52 +08:00
thop Update (#2978) 2025-03-23 16:39:35 +08:00
CMakeLists.txt Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00