kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
b8c5e3892b
TensorRT-LLMs/cpp/tensorrt_llm/kernels/communicationKernels
History
Shiyu Li b0d287c9b7
[TRTLLM-4647][fix] Fix the no fusion allreduce hanging (#4594) 
Signed-off-by: Shiyu Li <shili@nvidia.com>
2025-06-04 18:26:13 -07:00
..
allReduceFusionKernels.cu fix potential issues in allreduce fusion kernel and ut (#4226) 2025-05-19 17:38:29 +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 fix potential issues in allreduce fusion kernel and ut (#4226) 2025-05-19 17:38:29 +08:00
moeAllReduceFusionKernels.h feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00