mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
34e63d07e6
* feat: Variable-Beam-Width-Search Part2 Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search Part2 Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search Part2, fix CPP tests Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search Part3, simplify CPP tests Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search Part4, move beam_width_array param Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search, fix CI error Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search part2 Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search part2 Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search part2, fix pre-commit Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> * feat: Variable-Beam-Width-Search part2, fix review Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> --------- Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com> Co-authored-by: wili-65535 <wili-65535@user.noreply.github.com> |
||
|---|---|---|
| .. | ||
| baseSamplingLayerTest.cpp | ||
| baseSamplingLayerTest.h | ||
| beamSearchLayerTest.cpp | ||
| CMakeLists.txt | ||
| dynamicDecodeLayerTest.cpp | ||
| dynamicDecodeLayerTest.h | ||
| explicitDraftTokensLayerTest.cpp | ||
| explicitDraftTokensLayerTest.h | ||
| externalDraftTokensLayerTest.cpp | ||
| layerUtilsTest.cpp | ||
| lookaheadAlgorithmTest.cpp | ||
| lookaheadDecodingLayerTest.cpp | ||
| lookaheadPoolManagerTest.cpp | ||
| lookaheadRandomLlmTest.cpp | ||
| randomLlm.cpp | ||
| randomLlm.h | ||
| samplingLayerTest.cpp | ||
| topKSamplingLayerTest.cpp | ||
| topPSamplingLayerTest.cpp | ||