TensorRT-LLMs/cpp/tensorrt_llm/kernels/cutlass_kernels/include
Bo Li bf1b958f1a
[TRTLLM-7319][perf] Fuse slicing into MoE. (#6728)
Signed-off-by: Bo Li <22713281+bobboli@users.noreply.github.com>
Signed-off-by: Sergey Klevtsov <sklevtsov@nvidia.com>
Co-authored-by: Sergey Klevtsov <sklevtsov@nvidia.com>
2025-08-25 16:52:30 -04:00
..
allreduce_gemm_runner.h opensource: Opensource MOE MXFP8-MXFP4 implementation (#5222) 2025-06-26 12:18:19 +08:00
common.h [None] [feat] Add model gpt-oss (#6645) 2025-08-07 03:04:18 -04:00
cutlass_kernel_selector.h opensource: Opensource MOE MXFP8-MXFP4 implementation (#5222) 2025-06-26 12:18:19 +08:00
fp4_gemm.h feat: Add w4a8_mxfp4_fp8 quantization recipe. (#4867) 2025-06-16 11:30:57 +08:00
low_latency_gemm.h refactoring: port customized kernels with public cutlass version (#5027) 2025-06-13 16:19:31 +08:00
moe_gemm_kernels.h [TRTLLM-7319][perf] Fuse slicing into MoE. (#6728) 2025-08-25 16:52:30 -04:00
moe_kernels.h [TRTLLM-7319][perf] Fuse slicing into MoE. (#6728) 2025-08-25 16:52:30 -04:00
moe_util_kernels.h [TRTLLM-7319][perf] Fuse slicing into MoE. (#6728) 2025-08-25 16:52:30 -04:00