TensorRT-LLMs/cpp/tensorrt_llm/kernels/contextFusedMultiHeadAttention
2025-10-21 12:32:13 +08:00
..
cubin [https://nvbugs/5569081][fix] Upgrade fmha_v2. (cherry-pick from https://github.com/NVIDIA/TensorRT-LLM/pull/8364) (#8499) 2025-10-21 12:32:13 +08:00
CMakeLists.txt feat: reduce unnecessary kernel generation (#5476) 2025-07-04 14:37:49 +08:00
fmhaPackedMask.cu
fmhaPackedMask.h
fmhaRunner.cpp [TRTLLM-4629] [feat] Add support of CUDA13 and sm103 devices (#7568) 2025-09-16 09:56:18 +08:00
fmhaRunner.h hopper-style context MLA (#5713) 2025-07-23 14:37:20 +08:00
fused_multihead_attention_common.h [None][feat] Support NVFP4 KV Cache (#6244) 2025-09-01 09:24:52 +08:00
fused_multihead_attention_v2.cpp [TRTLLM-5366][feat]Add support for sm121 (#5524) 2025-07-08 14:27:00 -07:00
fused_multihead_attention_v2.h
tmaDescriptor.h