TensorRT-LLMs/cpp/tensorrt_llm/plugins/gptAttentionPlugin
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
..
CMakeLists.txt Initial commit 2023-09-20 00:29:41 -07:00
gptAttentionPlugin.cpp fix: fix for cp > kvHeadNum (#3002) 2025-03-26 12:39:02 +08:00
gptAttentionPlugin.h fix: fix for cp > kvHeadNum (#3002) 2025-03-26 12:39:02 +08:00