kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-08 20:21:48 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
c12e67bb66
TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/fmha
History
sunnyqgg 7862b15a65
[TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 
Signed-off-by: qgai <qgai@nvidia.com>
2025-11-17 09:01:53 +08:00
..
cubin [TRTLLM-8816][feat] add optimized trtllm-gen attention kernels on sm103 (#9081) 2025-11-13 12:41:07 +08:00
CMakeLists.txt
fmhaKernels.h [TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 2025-11-17 09:01:53 +08:00
fmhaReduction.cu
fmhaReduction.h
fmhaRunner.cpp
fmhaRunner.h
fmhaRunnerParams.h [TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 2025-11-17 09:01:53 +08:00
kernelParams.h [None][feat] add swapsMmaAb sparseMla kernels (#8913) 2025-11-05 09:32:34 -08:00
kernelUtils.h
prepareCustomMask.cu [TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 2025-11-17 09:01:53 +08:00
prepareCustomMask.h [TRTLLM-8778][feat] Add tree attention support for blackwell arch (#8975) 2025-11-17 09:01:53 +08:00