kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-04 10:11:47 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
baa250d1d6
TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/fmha
History
Tian Zheng cfebfbb505
[https://nvbugs/5783509][fix] Fix a hang issue when enabling skip softmax on Blackwell (#10490) 
Signed-off-by: Tian Zheng <29906817+Tom-Zheng@users.noreply.github.com>
2026-01-16 18:59:54 +08:00
..
cubin [https://nvbugs/5783509][fix] Fix a hang issue when enabling skip softmax on Blackwell (#10490) 2026-01-16 18:59:54 +08:00
CMakeLists.txt
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