TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/fmha
Tian Zheng 5efee01da1
[None][feat] Add Skip Softmax MLA kernels for Blackwell and Fix an accuracy bug of NVFP4 KV (#10813)
Signed-off-by: Tian Zheng <29906817+Tom-Zheng@users.noreply.github.com>
2026-01-26 16:46:33 +08:00
..
cubin [None][feat] Add Skip Softmax MLA kernels for Blackwell and Fix an accuracy bug of NVFP4 KV (#10813) 2026-01-26 16:46:33 +08:00
CMakeLists.txt Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
fmhaKernels.h [https://nvbugs/5783509][fix] Fix a hang issue when enabling skip softmax on Blackwell (#10490) 2026-01-16 18:59:54 +08:00
fmhaReduction.cu [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
fmhaReduction.h [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
fmhaRunner.cpp [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
fmhaRunner.h [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
fmhaRunnerParams.h [None][feat] update trtllm-gen to support groupsTokensHeadsQ (#10261) 2026-01-15 02:24:25 -05:00
kernelParams.h [None][feat] update trtllm-gen to support groupsTokensHeadsQ (#10261) 2026-01-15 02:24:25 -05:00
kernelUtils.h [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
prepareCustomMask.cu [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00
prepareCustomMask.h [None][fix] Introduce inline namespace to avoid symbol collision (#9541) 2025-12-12 23:32:15 +08:00