kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
54f68287fc
TensorRT-LLMs/cpp/tensorrt_llm/kernels/beamSearchKernels
History
wili eba3623a54
Feat: Variable-Beam-Width-Search (VBWS) part4 (#3979) 
* feat/vbws-part4-v1.8: rebase

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

* feat/vbws-part4-v1.9: fix incorrect output when using short output length

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

* v1.9.1: remove useless variables

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

* v1.9.2:fix incorrect output when using short output length

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

* v1.9.3: rebase

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

* v1.9.4: rebase

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

* v1.9.5: remove API change

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

---------

Signed-off-by: wili-65535 <wili-65535@users.noreply.github.com>
Co-authored-by: wili-65535 <wili-65535@users.noreply.github.com>
2025-05-12 22:32:29 +02:00
..
beamSearchKernels4.cu Update TensorRT-LLM (#2532) 2024-12-04 21:16:56 +08:00
beamSearchKernels8.cu Update TensorRT-LLM (#2532) 2024-12-04 21:16:56 +08:00
beamSearchKernels16.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels32.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels64.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels128.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels256.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels512.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernels1024.cu v1.2 (#3082) 2025-03-26 23:31:29 +08:00
beamSearchKernelsTemplate.h Feat: Variable-Beam-Width-Search (VBWS) part4 (#3979) 2025-05-12 22:32:29 +02:00