TensorRT-LLMs/cpp/tests/unit_tests
wili 54ad95eaa8
Feat: Variable-Beam-Width-Search (VBWS) part3 (#3338)
* feat/Variable-Beam-Width-Search-Part3, v1.0

Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com>

* feat/Variable-Beam-Width-Search-Part3, v1.1

Signed-off-by: wili-65535 <wili-65535@user.noreply.github.com>

* feat/Variable-Beam-Width-Search-Part3, v1.2

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>
2025-04-08 23:51:27 +08:00
..
batch_manager feat: use cudaMalloc to allocate kvCache (#3303) 2025-04-08 10:59:14 +08:00
common chore: remove usernames from comments (#3291) 2025-04-05 13:44:28 +08:00
executor chore: Add output of first token to additional generation outputs (#3205) 2025-04-02 20:14:16 +08:00
kernels feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00
layers Feat: Variable-Beam-Width-Search (VBWS) part3 (#3338) 2025-04-08 23:51:27 +08:00
runtime feat: Introduce UB allocator for pytorch flow (#3257) 2025-04-08 18:39:49 +08:00
thop Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
utils Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
CMakeLists.txt Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00