TensorRT-LLMs/cpp/tests/unit_tests/kernels/allReduce
Void 950cadf2bd
add support for smaller hidden_dim (#3609)
Signed-off-by: Yilin Zhang <18275976+yilin-void@users.noreply.github.com>
Co-authored-by: Yukun He <23156053+hyukn@users.noreply.github.com>
2025-04-17 12:00:32 +08:00
..
allReduceFusionTest.cu add support for smaller hidden_dim (#3609) 2025-04-17 12:00:32 +08:00
allReduceKernelTest.cu Update TensorRT-LLM (#2792) 2025-02-18 21:27:39 +08:00
gemmAllReduceTest.cu Update (#2978) 2025-03-23 16:39:35 +08:00
moeAllReduceFusionTest.cu feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00