TensorRT-LLMs/cpp/tests/unit_tests/kernels/allReduce
Yuan Tong 4b6c19737b
feat: support add internal cutlass kernels as subproject (#3658)
Signed-off-by: Yuan Tong <13075180+tongyuantongyu@users.noreply.github.com>
2025-05-06 11:35:07 +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 feat: support add internal cutlass kernels as subproject (#3658) 2025-05-06 11:35:07 +08:00
moeAllReduceFusionTest.cu feat: fix and improve allreduce and fusion kernels (#3064) 2025-04-08 19:33:52 +08:00