TensorRT-LLMs/cpp/tensorrt_llm/kernels/communicationKernels
Shiyu Li 6e1aee6fd6
[fix] Performance Optimization for MNNVL TwoShot Kernel (#5934)
Signed-off-by: Shiyu Li <shili@nvidia.com>
Co-authored-by: Zongfei Jing <20381269+zongfeijing@users.noreply.github.com>
2025-07-17 10:49:51 +08:00
..
allReduceFusionKernels.cu perf: better heuristic for allreduce (#5432) 2025-07-01 22:56:06 -04:00
allReduceFusionKernels.h Feat/ds r1 min latency opt round3, add router gemm, fused a gemm, PDL (#4560) 2025-06-14 17:36:22 +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 [fix] Performance Optimization for MNNVL TwoShot Kernel (#5934) 2025-07-17 10:49:51 +08:00
mnnvlTwoShotAllreduceKernels.h Adding two-shot allreduce kernel and mnnvl multicasting buffer (#4216) 2025-05-22 03:42:36 +08:00
moeAllReduceFusionKernels.cu Feat/ds r1 min latency opt round3, add router gemm, fused a gemm, PDL (#4560) 2025-06-14 17:36:22 +08:00
moeAllReduceFusionKernels.h [TRTLLM-3927] [feat] Finalize + Allreduce + add + rmsnorm fusion (#4756) 2025-06-10 19:55:16 +08:00