TensorRT-LLMs/cpp/tensorrt_llm/kernels/cutlass_kernels
Daniel Stokes 109f27265c
[None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126)
Signed-off-by: djns99 <40156487+djns99@users.noreply.github.com>
2025-09-02 21:54:43 -04:00
..
allreduce_gemm fix TMA error with GEMM+AR on TP=2 (#6075) 2025-07-18 10:26:08 +08:00
fp4_gemm [TRTLLM-5366][feat]Add support for sm121 (#5524) 2025-07-08 14:27:00 -07:00
fp8_blockscale_gemm [fix] fix tileN cannot % 16==0 & support sm89 deepgemm bmm (#5531) 2025-07-10 15:16:18 +08:00
fp8_rowwise_gemm feat: reduce unnecessary kernel generation (#5476) 2025-07-04 14:37:49 +08:00
fpA_intB_gemm [None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126) 2025-09-02 21:54:43 -04:00
fused_gated_gemm feat: reduce unnecessary kernel generation (#5476) 2025-07-04 14:37:49 +08:00
include [None][perf] Disable Swap AB when num tokens exceeds N dimension (#7104) 2025-08-28 21:29:55 -04:00
int8_gemm [None][perf] Make finalize fusion part of the tactic selection logic (#6915) 2025-08-21 14:08:03 -07:00
low_latency_gemm feat: reduce unnecessary kernel generation (#5476) 2025-07-04 14:37:49 +08:00
moe_gemm [None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126) 2025-09-02 21:54:43 -04:00
python [None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126) 2025-09-02 21:54:43 -04:00
CMakeLists.txt [None] [feat] Add model gpt-oss (#6645) 2025-08-07 03:04:18 -04:00
cutlass_heuristic.cpp [None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126) 2025-09-02 21:54:43 -04:00
cutlass_heuristic.h [None][perf] Add MOE support for dynamic cluster shapes and custom epilogue schedules (#6126) 2025-09-02 21:54:43 -04:00
cutlass_preprocessors.cpp [TRTLLM-5366][feat]Add support for sm121 (#5524) 2025-07-08 14:27:00 -07:00
cutlass_preprocessors.h Update TensorRT-LLM (#1492) 2024-04-24 14:44:22 +08:00
cutlass_type_conversion.h chore: cutlass cleanup (#3165) 2025-04-01 13:57:38 +08:00