diff --git a/cpp/tensorrt_llm/executor/decodingConfig.cpp b/cpp/tensorrt_llm/executor/decodingConfig.cpp index 5d052846b6c9..a9506438e482 100644 --- a/cpp/tensorrt_llm/executor/decodingConfig.cpp +++ b/cpp/tensorrt_llm/executor/decodingConfig.cpp @@ -164,6 +164,10 @@ EagleConfig::EagleConfig(std::optional eagleChoices, bool greedySa TLLM_CHECK_WITH_INFO(eagleChoices.has_value() == false, "When dynamic tree is enabled (for Eagle-2), eagle choices should not be set."); } + if (mDynamicTreeMaxTopK.has_value()) + { + TLLM_CHECK_WITH_INFO(mDynamicTreeMaxTopK.value() > 0, "dynamicTreeMaxTopK must be positive"); + } } bool EagleConfig::operator==(EagleConfig const& other) const