TensorRT-LLMs/cpp/tensorrt_llm/kernels/communicationKernels
Daniel Stokes 3a4851b7c3
feat: Add Mixture of Experts FP8xMXFP4 support (#4750)
Signed-off-by: Daniel Stokes <40156487+djns99@users.noreply.github.com>
2025-06-09 13:25:04 +08:00
..
allReduceFusionKernels.cu feat: Add Mixture of Experts FP8xMXFP4 support (#4750) 2025-06-09 13:25:04 +08:00
allReduceFusionKernels.h Unify two versions of AllReduce custom op (#3032) 2025-04-22 21:58:42 +08:00
allReduceWorkspace.cu chore: bump version to 0.19.0 (#3598) (#3841) 2025-04-29 16:57:22 +08:00
allReduceWorkspace.h feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00
customLowPrecisionAllReduceKernels.cu feat: Low Precision Allreduce for PCIe based GPU (#4344) 2025-05-20 06:53:46 +08:00
customLowPrecisionAllReduceKernels.h feat: Low Precision Allreduce for PCIe based GPU (#4344) 2025-05-20 06:53:46 +08:00
mnnvlTwoShotAllreduceKernels.cu [TRTLLM-4647][fix] Fix the no fusion allreduce hanging (#4594) 2025-06-04 18:26:13 -07:00
mnnvlTwoShotAllreduceKernels.h Adding two-shot allreduce kernel and mnnvl multicasting buffer (#4216) 2025-05-22 03:42:36 +08:00
moeAllReduceFusionKernels.cu feat: Add Mixture of Experts FP8xMXFP4 support (#4750) 2025-06-09 13:25:04 +08:00
moeAllReduceFusionKernels.h feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00