kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-06 03:01:50 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
287f6c2e0f
TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/fmha
History
Perkz Zheng c87f1a6b39
[https://nvbugs/5503479][fix] update trtllm-gen kernels to address few bugs (#10089) 
Signed-off-by: Perkz Zheng <67892460+PerkzZheng@users.noreply.github.com>
2025-12-22 04:45:33 -05:00
..
cubin [https://nvbugs/5503479][fix] update trtllm-gen kernels to address few bugs (#10089) 2025-12-22 04:45:33 -05:00
CMakeLists.txt
fmhaKernels.h [https://nvbugs/5503479][fix] update trtllm-gen kernels to address few bugs (#10089) 2025-12-22 04:45:33 -05:00
fmhaReduction.cu
fmhaReduction.h
fmhaRunner.cpp
fmhaRunner.h
fmhaRunnerParams.h [TRTLLM-9805][feat] Skip Softmax Attention. (#9821) 2025-12-21 02:52:42 -05:00
kernelParams.h [TRTLLM-9805][feat] Skip Softmax Attention. (#9821) 2025-12-21 02:52:42 -05:00
kernelUtils.h
prepareCustomMask.cu
prepareCustomMask.h