TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/blockScaleMoe
Matthias Jouanneaux a0b6c635b1
[feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063)
Signed-off-by: Matthias Jouanneaux <mjoux@nvidia.com>
Co-authored-by: hlu1 <14827759+hlu1@users.noreply.github.com>
Co-authored-by: Nikita Korobov <14355239+nekorobov@users.noreply.github.com>
2025-06-13 06:00:02 +08:00
..
CMakeLists.txt feat: [Deepseek] Add trtllm-gen MOE FP4 MOE backend (#3387) 2025-04-21 10:01:33 +08:00
DevKernel.cu feat: update DeepSeek FP8 TRT-LLM Gen cubins (#4643) 2025-06-03 14:07:54 -07:00
DevKernel.h [feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063) 2025-06-13 06:00:02 +08:00
IntFastDiv.h [feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063) 2025-06-13 06:00:02 +08:00
RoutingKernel.cu [feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063) 2025-06-13 06:00:02 +08:00
RoutingKernel.h [feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063) 2025-06-13 06:00:02 +08:00
runner.cu [feat] trtllmGen MoE routing: added support for top groups and top K bounds (#4063) 2025-06-13 06:00:02 +08:00
runner.h [TRTLLM-3927] [feat] Finalize + Allreduce + add + rmsnorm fusion (#4756) 2025-06-10 19:55:16 +08:00