TensorRT-LLMs/cpp/tensorrt_llm
DylanChen-NV 1ac0566a93
fix: fix for cp > kvHeadNum (#3002)
* fix for cp > kvHeadNum

Signed-off-by: Dylan Chen <191843203+DylanChen-NV@users.noreply.github.com>

* fix for None kv_head_num

Signed-off-by: Dylan Chen <191843203+DylanChen-NV@users.noreply.github.com>

---------

Signed-off-by: Dylan Chen <191843203+DylanChen-NV@users.noreply.github.com>
2025-03-26 12:39:02 +08:00
..
batch_manager Add updateKVCacheTransfer (#2984) 2025-03-25 21:45:35 +08:00
common fix: fix for cp > kvHeadNum (#3002) 2025-03-26 12:39:02 +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: fix for cp > kvHeadNum (#3002) 2025-03-26 12:39:02 +08:00
layers Update TensorRT-LLM (#2873) 2025-03-11 21:13:42 +08:00
plugins fix: fix for cp > kvHeadNum (#3002) 2025-03-26 12:39:02 +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