TensorRT-LLMs/cpp/tensorrt_llm/kernels/communicationKernels
Shiyu Li eeb56c2848
[None][feat] MNNVLAllreduce Kernel Refactor (#8018)
Signed-off-by: Shiyu Li <timlee0212@outlook.com>
Co-authored-by: Jin Li <59594262+liji-nv@users.noreply.github.com>
2025-11-05 08:49:47 +08:00
..
allReduceFusionKernels.cu [TRTLLM-8129][feat] Allreduce tuning and benchmark script revising (#7870) 2025-11-04 16:42:31 +08:00
allReduceFusionKernels.h [None] [feat] Add model gpt-oss (#6645) 2025-08-07 03:04:18 -04: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
mnnvlAllreduceKernels.cu [None][feat] MNNVLAllreduce Kernel Refactor (#8018) 2025-11-05 08:49:47 +08:00
mnnvlAllreduceKernels.h [None][feat] MNNVLAllreduce Kernel Refactor (#8018) 2025-11-05 08:49:47 +08:00
moeAllReduceFusionKernels.cu [None] [feat] Add model gpt-oss (#6645) 2025-08-07 03:04:18 -04:00
moeAllReduceFusionKernels.h [None] [feat] Add model gpt-oss (#6645) 2025-08-07 03:04:18 -04:00
moeAlltoAllKernels.cu [None][feat] Integrate MnnvlThroughput into TRTLLM MoE. (#8728) 2025-11-04 21:36:29 +08:00
moeAlltoAllKernels.h [None][feat] Integrate MnnvlThroughput into TRTLLM MoE. (#8728) 2025-11-04 21:36:29 +08:00