mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-23 04:03:22 +08:00
This PR adds a customized allreduce to TensorRT-LLM. The new allreduce is used for communication on PCIe-based GPUs via low-precision quantization, which can accelerate the PCIe allreduce process. Signed-off-by: Hui Kang <hkang@nvidia.com> Co-authored-by: Hui Kang <hkang@nvidia.com> |
||
|---|---|---|
| .. | ||
| test_allreduce.py | ||
| test_ar_residual_norm.py | ||
| test_embedding.py | ||
| test_linear.py | ||
| test_lowprecision_allreduce.py | ||
| test_star_attention_input.jsonl | ||
| test_star_attention.py | ||
| test_user_buffers.py | ||